1 files changed, 26 insertions, 5 deletions
diff --git a/main/Test/Service.hs b/main/Test/Service.hs
index 1018e0d..c0be07d 100644
--- a/main/Test/Service.hs
+++ b/main/Test/Service.hs
@@ -1,20 +1,26 @@
 module Test.Service (
     TestMessage(..),
     TestMessageAttributes(..),
+
+    openTestStreams,
 ) where
 
+import Control.Monad
 import Control.Monad.Reader
 
 import Data.ByteString.Lazy.Char8 qualified as BL
 
 import Erebos.Network
+import Erebos.Object
 import Erebos.Service
-import Erebos.Storage
+import Erebos.Service.Stream
+import Erebos.Storable
 
 data TestMessage = TestMessage (Stored Object)
 
 data TestMessageAttributes = TestMessageAttributes
-    { testMessageReceived :: String -> String -> String -> ServiceHandler TestMessage ()
+    { testMessageReceived :: Object -> String -> String -> String -> ServiceHandler TestMessage ()
+    , testStreamsReceived :: [ StreamReader ] -> ServiceHandler TestMessage ()
     }
 
 instance Storable TestMessage where
@@ -25,12 +31,27 @@ instance Service TestMessage where
     serviceID _ = mkServiceID "cb46b92c-9203-4694-8370-8742d8ac9dc8"
 
     type ServiceAttributes TestMessage = TestMessageAttributes
-    defaultServiceAttributes _ = TestMessageAttributes (\_ _ _ -> return ())
+    defaultServiceAttributes _ = TestMessageAttributes
+        { testMessageReceived = \_ _ _ _ -> return ()
+        , testStreamsReceived = \_ -> return ()
+        }
 
     serviceHandler smsg = do
         let TestMessage sobj = fromStored smsg
-        case map BL.unpack $ BL.words $ BL.takeWhile (/='\n') $ serializeObject $ fromStored sobj of
+            obj = fromStored sobj
+        case map BL.unpack $ BL.words $ BL.takeWhile (/='\n') $ serializeObject obj of
             [otype, len] -> do
                 cb <- asks $ testMessageReceived . svcAttributes
-                cb otype len (show $ refDigest $ storedRef sobj)
+                cb obj otype len (show $ refDigest $ storedRef sobj)
             _ -> return ()
+
+        streams <- receivedStreams
+        when (not $ null streams) $ do
+            cb <- asks $ testStreamsReceived . svcAttributes
+            cb streams
+
+
+openTestStreams :: Int -> ServiceHandler TestMessage [ StreamWriter ]
+openTestStreams count = do
+    replyPacket . TestMessage =<< mstore (Rec [])
+    replicateM count openStream