Separate module for output handling

1 files changed, 8 insertions, 2 deletions

diff --git a/src/Test.hs b/src/Test.hs
index 8136afb..83ffac5 100644
--- a/src/Test.hs
+++ b/src/Test.hs
@@ -2,8 +2,8 @@ module Test (
     Test(..),
     TestStep(..),
 
-    ProcName(..), unpackProcName,
-    NodeName(..), unpackNodeName,
+    ProcName(..), textProcName, unpackProcName,
+    NodeName(..), textNodeName, unpackNodeName,
 ) where
 
 import Data.Text (Text)
@@ -24,11 +24,17 @@ data TestStep = Spawn ProcName NodeName
 newtype ProcName = ProcName Text
     deriving (Eq, Ord)
 
+textProcName :: ProcName -> Text
+textProcName (ProcName name) = name
+
 unpackProcName :: ProcName -> String
 unpackProcName (ProcName tname) = T.unpack tname
 
 newtype NodeName = NodeName Text
     deriving (Eq, Ord)
 
+textNodeName :: NodeName -> Text
+textNodeName (NodeName name) = name
+
 unpackNodeName :: NodeName -> String
 unpackNodeName (NodeName tname) = T.unpack tname