1 files changed, 20 insertions, 1 deletions
diff --git a/main/Test/Service.hs b/main/Test/Service.hs
index 8c58dee..c0be07d 100644
--- a/main/Test/Service.hs
+++ b/main/Test/Service.hs
@@ -1,8 +1,11 @@
 module Test.Service (
     TestMessage(..),
     TestMessageAttributes(..),
+
+    openTestStreams,
 ) where
 
+import Control.Monad
 import Control.Monad.Reader
 
 import Data.ByteString.Lazy.Char8 qualified as BL
@@ -10,12 +13,14 @@ import Data.ByteString.Lazy.Char8 qualified as BL
 import Erebos.Network
 import Erebos.Object
 import Erebos.Service
+import Erebos.Service.Stream
 import Erebos.Storable
 
 data TestMessage = TestMessage (Stored Object)
 
 data TestMessageAttributes = TestMessageAttributes
     { testMessageReceived :: Object -> String -> String -> String -> ServiceHandler TestMessage ()
+    , testStreamsReceived :: [ StreamReader ] -> ServiceHandler TestMessage ()
     }
 
 instance Storable TestMessage where
@@ -26,7 +31,10 @@ 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
@@ -36,3 +44,14 @@ instance Service TestMessage where
                 cb <- asks $ testMessageReceived . svcAttributes
                 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