server-test: prop_unidirectional_Sim use NonFailingBearerInfoScript

This simulation is not designed to handle errors.
IntersectMBO · Oct 14, 2021 · 6c4ff64 · 6c4ff64
1 parent d4fe55e
commit 6c4ff64
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ouroboros-network-framework/test/Test/Ouroboros/Network/Server2.hs b/ouroboros-network-framework/test/Test/Ouroboros/Network/Server2.hs
@@ -823,11 +823,13 @@ unidirectionalExperiment timeouts snocket socket clientAndServerData = do
                 (property True)
                 $ zip rs (expectedResult clientAndServerData clientAndServerData)
 
-prop_unidirectional_Sim :: AbsBearerInfo -> ClientAndServerData Int -> Property
-prop_unidirectional_Sim absBi clientAndServerData =
+prop_unidirectional_Sim :: NonFailingBearerInfoScript
+                        -> ClientAndServerData Int
+                        -> Property
+prop_unidirectional_Sim (NonFailingBearerInfoScript script) clientAndServerData =
   simulatedPropertyWithTimeout 7200 $
     withSnocket nullTracer
-                (Script (toBearerInfo absBi :| [noAttenuation])) $ \snock ->
+                (toBearerInfo <$> script) $ \snock ->
       bracket (Snocket.open snock Snocket.TestFamily)
               (Snocket.close snock) $ \fd -> do
         Snocket.bind   snock fd serverAddr