git.iko.soy/distributed-process
on 13a3746 help sign in
.travis.yml+6−11
since splitsnapshots
unifiedsplit
@@ -6,11 +6,11 @@ matrix:
6 include:
6 include:
7 - env: ARGS="--stack-yaml stack-ghc-7.10.3.yaml"
7 - env: ARGS="--stack-yaml stack-ghc-7.10.3.yaml"
8 addons: {apt: {packages: [libgmp-dev]}}
8 addons: {apt: {packages: [libgmp-dev]}}
9+
9− - env: ARGS="--stack-yaml stack-ghc-8.2.2.yaml"
10+ - env: ARGS="--stack-yaml stack-ghc-8.0.2.yaml"
10− addons: {apt: {packages: [libgmp-dev]}}
11− - env: ARGS="--stack-yaml stack-ghc-8.4.4.yaml"
12 addons: {apt: {packages: [libgmp-dev]}}
11 addons: {apt: {packages: [libgmp-dev]}}
13− - env: ARGS="--stack-yaml stack-nightly.yaml --resolver nightly"
12+
13+ - env: ARGS=
14 addons: {apt: {packages: [libgmp-dev]}}
14 addons: {apt: {packages: [libgmp-dev]}}
15
15
16cache:
16cache:
@@ -34,10 +34,5 @@ script:
34- stack ${ARGS} test $ARG='--plain -t "!Flaky"' ${TEST_PACKAGE}TestCHInTCP
34- stack ${ARGS} test $ARG='--plain -t "!Flaky"' ${TEST_PACKAGE}TestCHInTCP
35- stack ${ARGS} test $ARG='--plain -t "!SpawnReconnect"' ${TEST_PACKAGE}TestClosure
35- stack ${ARGS} test $ARG='--plain -t "!SpawnReconnect"' ${TEST_PACKAGE}TestClosure
36- stack ${ARGS} test $ARG='--plain' ${TEST_PACKAGE}TestStats
36- stack ${ARGS} test $ARG='--plain' ${TEST_PACKAGE}TestStats
37−- stack ${ARGS} test $ARG='--plain' ${TEST_PACKAGE}TestMxInMemory
37+- stack ${ARGS} test $ARG='--plain' ${TEST_PACKAGE}TestMx
38−- stack ${ARGS} test $ARG='--plain' ${TEST_PACKAGE}TestMxInTCP
38+- stack ${ARGS} test $ARG='--plain' ${TEST_PACKAGE}TestTracing
39−- stack ${ARGS} test $ARG='--plain' ${TEST_PACKAGE}TestTracingInMemory
40−
41−notifications:
42− slack:
43− secure: LbqbNjK1KiHNiQo/i/Za4vkMD9l9mhmN+PWYFBQkNvGFWffSRayRKFrTqt4znV2p7h4dY1XZFoC7T5RemBVQLq2ppZAUWxkITeu1OUlnmVLDfJLeVYGTR/fn90nP6Y5ITE7T3A07nKmWaRbKpIgFBAOjzgqaM0csscDx3z0WW18=
benchmarks/Channels.hs+5−6
@@ -4,7 +4,7 @@ import Control.Monad
4import Control.Applicative
4import Control.Applicative
5import Control.Distributed.Process
5import Control.Distributed.Process
6import Control.Distributed.Process.Node
6import Control.Distributed.Process.Node
7−import Network.Transport.TCP (createTransport, defaultTCPAddr, defaultTCPParameters)
7+import Network.Transport.TCP (createTransport, defaultTCPParameters)
8import Data.Binary (encode, decode)
8import Data.Binary (encode, decode)
9import qualified Data.ByteString.Lazy as BSL
9import qualified Data.ByteString.Lazy as BSL
10
10
@@ -36,8 +36,7 @@ initialProcess "CLIENT" = do
36main :: IO ()
36main :: IO ()
37main = do
37main = do
38 [role, host, port] <- getArgs
38 [role, host, port] <- getArgs
39+ Right transport <- createTransport
39− trans <- createTransport (defaultTCPAddr host port) defaultTCPParameters
40+ host port (\sn -> (host, port)) defaultTCPParameters
40− case trans of
41− Right transport -> do node <- newLocalNode transport initRemoteTable
41+ node <- newLocalNode transport initRemoteTable
42− runProcess node $ initialProcess role
42+ runProcess node $ initialProcess role
43− Left other -> error $ show other
benchmarks/Latency.hs+5−7
@@ -3,7 +3,7 @@ import Control.Monad
3import Control.Applicative
3import Control.Applicative
4import Control.Distributed.Process
4import Control.Distributed.Process
5import Control.Distributed.Process.Node
5import Control.Distributed.Process.Node
6−import Network.Transport.TCP (createTransport, defaultTCPAddr, defaultTCPParameters)
6+import Network.Transport.TCP (createTransport, defaultTCPParameters)
7import Data.Binary (encode, decode)
7import Data.Binary (encode, decode)
8import qualified Data.ByteString.Lazy as BSL
8import qualified Data.ByteString.Lazy as BSL
9
9
@@ -31,9 +31,7 @@ initialProcess "CLIENT" = do
31main :: IO ()
31main :: IO ()
32main = do
32main = do
33 [role, host, port] <- getArgs
33 [role, host, port] <- getArgs
34− trans <- createTransport
34+ Right transport <- createTransport
35− (defaultTCPAddr host port) defaultTCPParameters
35+ host port (\sn -> (host, sn)) defaultTCPParameters
36− case trans of
37− Right transport -> do node <- newLocalNode transport initRemoteTable
36+ node <- newLocalNode transport initRemoteTable
38− runProcess node $ initialProcess role
37+ runProcess node $ initialProcess role
39− Left other -> error $ show other
benchmarks/ProcessRing.hs+2−2
@@ -12,7 +12,7 @@ import Control.Monad
12import Control.Distributed.Process hiding (catch)
12import Control.Distributed.Process hiding (catch)
13import Control.Distributed.Process.Node
13import Control.Distributed.Process.Node
14import Control.Exception (catch, SomeException)
14import Control.Exception (catch, SomeException)
15−import Network.Transport.TCP (createTransport, defaultTCPAddr, defaultTCPParameters)
15+import Network.Transport.TCP (createTransport, defaultTCPParameters)
16import System.Environment
16import System.Environment
17import System.Console.GetOpt
17import System.Console.GetOpt
18
18
@@ -111,7 +111,7 @@ main = do
111 (opt, _) <- parseArgv argv
111 (opt, _) <- parseArgv argv
112 putStrLn $ "options: " ++ (show opt)
112 putStrLn $ "options: " ++ (show opt)
113 Right transport <- createTransport
113 Right transport <- createTransport
114− (defaultTCPAddr "127.0.0.1" "8090" ) defaultTCPParameters
114+ "127.0.0.1" "8090" (\sn -> ("127.0.0.1", sn)) defaultTCPParameters
115 node <- newLocalNode transport initRemoteTable
115 node <- newLocalNode transport initRemoteTable
116 catch (void $ runProcess node $ initialProcess opt)
116 catch (void $ runProcess node $ initialProcess opt)
117 (\(e :: SomeException) -> putStrLn $ "ERROR: " ++ (show e))
117 (\(e :: SomeException) -> putStrLn $ "ERROR: " ++ (show e))
benchmarks/Spawns.hs+5−6
@@ -7,7 +7,7 @@ import Control.Monad
7import Control.Applicative
7import Control.Applicative
8import Control.Distributed.Process
8import Control.Distributed.Process
9import Control.Distributed.Process.Node
9import Control.Distributed.Process.Node
10−import Network.Transport.TCP (createTransport, defaultTCPAddr, defaultTCPParameters)
10+import Network.Transport.TCP (createTransport, defaultTCPParameters)
11import Data.Binary (encode, decode)
11import Data.Binary (encode, decode)
12import qualified Data.ByteString.Lazy as BSL
12import qualified Data.ByteString.Lazy as BSL
13
13
@@ -42,8 +42,7 @@ initialProcess "CLIENT" = do
42main :: IO ()
42main :: IO ()
43main = do
43main = do
44 [role, host, port] <- getArgs
44 [role, host, port] <- getArgs
45+ Right transport <- createTransport
45− trans <- createTransport (defaultTCPAddr host port) defaultTCPParameters
46+ host port (\sn -> (host, sn)) defaultTCPParameters
46− case trans of
47− Right transport -> do node <- newLocalNode transport initRemoteTable
47+ node <- newLocalNode transport initRemoteTable
48− runProcess node $ initialProcess role
48+ runProcess node $ initialProcess role
49− Left other -> error $ show other
benchmarks/Throughput.hs+5−6
@@ -6,7 +6,7 @@ import Control.Monad
6import Control.Applicative
6import Control.Applicative
7import Control.Distributed.Process
7import Control.Distributed.Process
8import Control.Distributed.Process.Node
8import Control.Distributed.Process.Node
9−import Network.Transport.TCP (createTransport, defaultTCPParameters, defaultTCPAddr)
9+import Network.Transport.TCP (createTransport, defaultTCPParameters)
10import Data.Binary
10import Data.Binary
11import qualified Data.ByteString.Lazy as BSL
11import qualified Data.ByteString.Lazy as BSL
12import Data.Typeable
12import Data.Typeable
@@ -67,8 +67,7 @@ initialProcess "CLIENT" = do
67main :: IO ()
67main :: IO ()
68main = do
68main = do
69 [role, host, port] <- getArgs
69 [role, host, port] <- getArgs
70+ Right transport <- createTransport
70− trans <- createTransport (defaultTCPAddr host port) defaultTCPParameters
71+ host port (\sn -> (host, sn)) defaultTCPParameters
71− case trans of
72− Right transport -> do node <- newLocalNode transport initRemoteTable
72+ node <- newLocalNode transport initRemoteTable
73− runProcess node $ initialProcess role
73+ runProcess node $ initialProcess role
74− Left other -> error $ show other
distributed-process-tests/distributed-process-tests.cabal+13−27
@@ -6,7 +6,7 @@ description: Tests and test suite for Cloud Haskell libraries.
6license: BSD3
6license: BSD3
7license-file: LICENSE
7license-file: LICENSE
8Author: Duncan Coutts, Nicolas Wu, Edsko de Vries
8Author: Duncan Coutts, Nicolas Wu, Edsko de Vries
9−Maintainer: Tim Watson <watson.timothy@gmail.com>
9+Maintainer: Facundo Domínguez <facundo.dominguez@tweag.io>
10copyright: Well-Typed LLP
10copyright: Well-Typed LLP
11category: Control, Cloud Haskell
11category: Control, Cloud Haskell
12build-type: Simple
12build-type: Simple
@@ -31,10 +31,9 @@ library
31 bytestring >= 0.9 && < 0.12,
31 bytestring >= 0.9 && < 0.12,
32 distributed-process >= 0.6.0 && < 0.8,
32 distributed-process >= 0.6.0 && < 0.8,
33 distributed-static,
33 distributed-static,
34− exceptions >= 0.5,
35 HUnit >= 1.2 && < 1.7,
34 HUnit >= 1.2 && < 1.7,
36 network-transport >= 0.4.1.0 && < 0.6,
35 network-transport >= 0.4.1.0 && < 0.6,
37− network >= 2.5 && < 2.9,
36+ network >= 2.5 && < 2.7,
38 random >= 1.0 && < 1.2,
37 random >= 1.0 && < 1.2,
39 rematch >= 0.1.2.1,
38 rematch >= 0.1.2.1,
40 setenv >= 0.1.1.3,
39 setenv >= 0.1.1.3,
@@ -61,7 +60,7 @@ Test-Suite TestCHInMemory
61 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.CH
60 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.CH
62 Build-Depends: base >= 4.4 && < 5,
61 Build-Depends: base >= 4.4 && < 5,
63 distributed-process-tests,
62 distributed-process-tests,
64− network >= 2.3 && < 2.9,
63+ network >= 2.3 && < 2.7,
65 network-transport >= 0.4.1.0 && < 0.6,
64 network-transport >= 0.4.1.0 && < 0.6,
66 network-transport-inmemory >= 0.5,
65 network-transport-inmemory >= 0.5,
67 test-framework >= 0.6 && < 0.9
66 test-framework >= 0.6 && < 0.9
@@ -76,7 +75,7 @@ Test-Suite TestCHInTCP
76 if flag(tcp)
75 if flag(tcp)
77 Build-Depends: base >= 4.4 && < 5,
76 Build-Depends: base >= 4.4 && < 5,
78 distributed-process-tests,
77 distributed-process-tests,
79− network >= 2.3 && < 2.9,
78+ network >= 2.3 && < 2.7,
80 network-transport >= 0.4.1.0 && < 0.6,
79 network-transport >= 0.4.1.0 && < 0.6,
81 network-transport-tcp >= 0.5 && < 0.7,
80 network-transport-tcp >= 0.5 && < 0.7,
82 test-framework >= 0.6 && < 0.9
81 test-framework >= 0.6 && < 0.9
@@ -93,7 +92,7 @@ Test-Suite TestClosure
93 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Closure
92 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Closure
94 Build-Depends: base >= 4.4 && < 5,
93 Build-Depends: base >= 4.4 && < 5,
95 distributed-process-tests,
94 distributed-process-tests,
96− network >= 2.3 && < 2.9,
95+ network >= 2.3 && < 2.7,
97 network-transport >= 0.4.1.0 && < 0.6,
96 network-transport >= 0.4.1.0 && < 0.6,
98 network-transport-inmemory >= 0.5,
97 network-transport-inmemory >= 0.5,
99 test-framework >= 0.6 && < 0.9
98 test-framework >= 0.6 && < 0.9
@@ -107,7 +106,7 @@ Test-Suite TestStats
107 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Stats
106 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Stats
108 Build-Depends: base >= 4.4 && < 5,
107 Build-Depends: base >= 4.4 && < 5,
109 distributed-process-tests,
108 distributed-process-tests,
110− network >= 2.3 && < 2.9,
109+ network >= 2.3 && < 2.7,
111 network-transport >= 0.4.1.0 && < 0.6,
110 network-transport >= 0.4.1.0 && < 0.6,
112 network-transport-inmemory >= 0.5,
111 network-transport-inmemory >= 0.5,
113 test-framework >= 0.6 && < 0.9
112 test-framework >= 0.6 && < 0.9
@@ -115,44 +114,31 @@ Test-Suite TestStats
115 ghc-options: -Wall -eventlog -threaded -rtsopts -with-rtsopts=-N -fno-warn-unused-do-bind
114 ghc-options: -Wall -eventlog -threaded -rtsopts -with-rtsopts=-N -fno-warn-unused-do-bind
116 HS-Source-Dirs: tests
115 HS-Source-Dirs: tests
117
116
117+
118−Test-Suite TestMxInMemory
118+Test-Suite TestMx
119 Type: exitcode-stdio-1.0
119 Type: exitcode-stdio-1.0
120 Main-Is: runInMemory.hs
120 Main-Is: runInMemory.hs
121 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Mx
121 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Mx
122 Build-Depends: base >= 4.4 && < 5,
122 Build-Depends: base >= 4.4 && < 5,
123 distributed-process-tests,
123 distributed-process-tests,
124− network >= 2.3 && < 2.9,
124+ network >= 2.3 && < 2.7,
125 network-transport >= 0.4.1.0 && < 0.6,
125 network-transport >= 0.4.1.0 && < 0.6,
126 network-transport-inmemory >= 0.5,
126 network-transport-inmemory >= 0.5,
127 test-framework >= 0.6 && < 0.9
127 test-framework >= 0.6 && < 0.9
128 Extensions: CPP
128 Extensions: CPP
129− ghc-options: -threaded -rtsopts -with-rtsopts=-N -fno-warn-unused-do-bind
129+ ghc-options: -Wall -threaded -rtsopts -with-rtsopts=-N -fno-warn-unused-do-bind
130 HS-Source-Dirs: tests
130 HS-Source-Dirs: tests
131
131
132−Test-Suite TestTracingInMemory
132+Test-Suite TestTracing
133 Type: exitcode-stdio-1.0
133 Type: exitcode-stdio-1.0
134 Main-Is: runInMemory.hs
134 Main-Is: runInMemory.hs
135 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Tracing
135 CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Tracing
136 Build-Depends: base >= 4.4 && < 5,
136 Build-Depends: base >= 4.4 && < 5,
137 distributed-process-tests,
137 distributed-process-tests,
138− network >= 2.3 && < 2.9,
138+ network >= 2.3 && < 2.7,
139 network-transport >= 0.4.1.0 && < 0.6,
139 network-transport >= 0.4.1.0 && < 0.6,
140 network-transport-inmemory >= 0.5,
140 network-transport-inmemory >= 0.5,
141 test-framework >= 0.6 && < 0.9
141 test-framework >= 0.6 && < 0.9
142 Extensions: CPP
142 Extensions: CPP
143− ghc-options: -eventlog -threaded -rtsopts -with-rtsopts=-N -fno-warn-unused-do-bind
143+ ghc-options: -Wall -eventlog -threaded -rtsopts -with-rtsopts=-N -fno-warn-unused-do-bind
144− HS-Source-Dirs: tests
145−
146−Test-Suite TestMxInTCP
147− Type: exitcode-stdio-1.0
148− Main-Is: runInMemory.hs
149− CPP-Options: -DTEST_SUITE_MODULE=Control.Distributed.Process.Tests.Mx
150− Build-Depends: base >= 4.4 && < 5,
151− distributed-process-tests,
152− network >= 2.3 && < 2.9,
153− network-transport >= 0.4.1.0 && < 0.6,
154− network-transport-inmemory >= 0.5,
155− test-framework >= 0.6 && < 0.9
156− Extensions: CPP
157− ghc-options: -Wall -threaded -rtsopts -with-rtsopts=-N -fno-warn-unused-do-bind
158 HS-Source-Dirs: tests
144 HS-Source-Dirs: tests
distributed-process-tests/src/Control/Distributed/Process/Tests/CH.hs+234−466
@@ -25,32 +25,24 @@ import Control.Concurrent.MVar
25 )
25 )
26import Control.Monad (replicateM_, replicateM, forever, void, unless, join)
26import Control.Monad (replicateM_, replicateM, forever, void, unless, join)
27import Control.Exception (SomeException, throwIO, ErrorCall(..))
27import Control.Exception (SomeException, throwIO, ErrorCall(..))
28−import Control.Monad.Catch (try, catch, finally, mask, onException)
28+import qualified Control.Exception as Ex (catch)
29−import Control.Applicative ((<|>))
29+import Control.Applicative ((<$>), (<*>), pure, (<|>))
30import qualified Network.Transport as NT (closeEndPoint, EndPointAddress)
30import qualified Network.Transport as NT (closeEndPoint, EndPointAddress)
31−import Control.Distributed.Process hiding
31+import Control.Distributed.Process
32− ( try
33− , catch
34− , finally
35− , mask
36− , onException
37− )
38import Control.Distributed.Process.Internal.Types
32import Control.Distributed.Process.Internal.Types
33+ ( NodeId(nodeAddress)
39− ( LocalNode(localEndPoint)
34+ , LocalNode(localEndPoint)
40 , ProcessExitException(..)
35 , ProcessExitException(..)
41 , nullProcessId
36 , nullProcessId
42 , createUnencodedMessage
37 , createUnencodedMessage
43 )
38 )
44import Control.Distributed.Process.Node
39import Control.Distributed.Process.Node
45−import Control.Distributed.Process.Debug
46−import Control.Distributed.Process.Management.Internal.Types
47−import Control.Distributed.Process.Tests.Internal.Utils (shouldBe, pause)
48import Control.Distributed.Process.Serializable (Serializable)
40import Control.Distributed.Process.Serializable (Serializable)
49−import Data.Maybe (isNothing, isJust)
41+
50−import Test.HUnit (Assertion, assertBool, assertFailure)
42+import Test.HUnit (Assertion, assertFailure)
51import Test.Framework (Test, testGroup)
43import Test.Framework (Test, testGroup)
52import Test.Framework.Providers.HUnit (testCase)
44import Test.Framework.Providers.HUnit (testCase)
53−import Control.Rematch hiding (match, isNothing, isJust)
45+import Control.Rematch hiding (match)
54import Control.Rematch.Run (Match(..))
46import Control.Rematch.Run (Match(..))
55
47
56newtype Ping = Ping ProcessId
48newtype Ping = Ping ProcessId
@@ -73,7 +65,7 @@ expectThat a matcher = case res of
73forkTry :: IO () -> IO ThreadId
65forkTry :: IO () -> IO ThreadId
74forkTry p = do
66forkTry p = do
75 tid <- myThreadId
67 tid <- myThreadId
76− forkIO $ catch p (\e -> throwTo tid (e :: SomeException))
68+ forkIO $ Ex.catch p (\e -> throwTo tid (e :: SomeException))
77
69
78-- | The ping server from the paper
70-- | The ping server from the paper
79ping :: Process ()
71ping :: Process ()
@@ -83,46 +75,12 @@ ping = do
83 send partner (Ping self)
75 send partner (Ping self)
84 ping
76 ping
85
77
86−verifyClient :: String -> MVar Bool -> IO ()
87−verifyClient s b = takeMVar b >>= assertBool s
88−
89−expectPing :: MVar Bool -> Process ()
90−expectPing mv = expect >>= liftIO . putMVar mv . checkPing
91− where
92− checkPing (Ping _) = True
93−
94-- | Quick and dirty synchronous version of whereisRemoteAsync
78-- | Quick and dirty synchronous version of whereisRemoteAsync
95whereisRemote :: NodeId -> String -> Process (Maybe ProcessId)
79whereisRemote :: NodeId -> String -> Process (Maybe ProcessId)
96whereisRemote nid string = do
80whereisRemote nid string = do
97 whereisRemoteAsync nid string
81 whereisRemoteAsync nid string
98− receiveWait [
99− match (\(WhereIsReply _ mPid) -> return mPid)
82+ WhereIsReply _ mPid <- expect
100− ]
101−
102−verifyWhereIsRemote :: NodeId -> String -> Process ProcessId
103−verifyWhereIsRemote n s = whereisRemote n s >>= maybe (die "remote name not found") return
104−
105−waitForExit :: ProcessId -> Process DiedReason
106−waitForExit pid = monitor pid >>= waitForDown
107−
108−waitForDown :: MonitorRef -> Process DiedReason
109−waitForDown ref =
110− receiveWait [ matchIf (\(ProcessMonitorNotification ref' _ _) -> ref == ref')
111− (\(ProcessMonitorNotification _ _ dr) -> return dr) ]
112−
113−syncBreakConnection :: (NT.EndPointAddress -> NT.EndPointAddress -> IO ()) -> LocalNode -> LocalNode -> IO ()
114−syncBreakConnection breakConnection nid0 nid1 = do
115− m <- newEmptyMVar
116− _ <- forkProcess nid1 $ getSelfPid >>= liftIO . putMVar m
117− runProcess nid0 $ do
118− them <- liftIO $ takeMVar m
119− pinger <- spawnLocal $ forever $ send them ()
120− _ <- monitorNode (localNodeId nid1)
121− liftIO $ breakConnection (nodeAddress $ localNodeId nid0)
122− (nodeAddress $ localNodeId nid1)
123− NodeMonitorNotification _ _ _ <- expect
124− kill pinger "finished"
125− return ()
83+ return mPid
126
84
127data Add = Add ProcessId Double Double deriving (Typeable)
85data Add = Add ProcessId Double Double deriving (Typeable)
128data Divide = Divide ProcessId Double Double deriving (Typeable)
86data Divide = Divide ProcessId Double Double deriving (Typeable)
@@ -197,18 +155,12 @@ monitorTestProcess theirAddr mOrL un reason monitorSetup done =
197 unmonitor ref
155 unmonitor ref
198 liftIO $ putMVar done ()
156 liftIO $ putMVar done ()
199 (False, ref) -> do
157 (False, ref) -> do
200− receiveWait [
201− match (\(ProcessMonitorNotification ref' pid reason') -> do
158+ ProcessMonitorNotification ref' pid reason' <- expect
202− liftIO $ do
203− assertBool "Bad Monitor Signal"
204− (Just ref' == ref && pid == theirAddr &&
159+ True <- return $ Just ref' == ref && pid == theirAddr && mOrL && reason == reason'
205− mOrL && reason == reason')
206− putMVar done ())
160+ liftIO $ putMVar done ()
207− ]
208 )
161 )
209 (\(ProcessLinkException pid reason') -> do
162 (\(ProcessLinkException pid reason') -> do
210− (liftIO $ assertBool "link exception unmatched" $
211− pid == theirAddr && not mOrL && not un && reason == reason')
163+ True <- return $ pid == theirAddr && not mOrL && not un && reason == reason'
212 liftIO $ putMVar done ()
164 liftIO $ putMVar done ()
213 )
165 )
214
166
@@ -239,10 +191,8 @@ testPing TestTransport{..} = do
239 pid <- getSelfPid
191 pid <- getSelfPid
240 replicateM_ numPings $ do
192 replicateM_ numPings $ do
241 send pingServer (Pong pid)
193 send pingServer (Pong pid)
242− p <- expectTimeout 3000000
194+ Ping _ <- expect
243− case p of
244− Just (Ping _) -> return ()
195+ return ()
245− Nothing -> die "Failed to receive Ping"
246
196
247 putMVar clientDone ()
197 putMVar clientDone ()
248
198
@@ -329,7 +279,7 @@ testMonitorLocalDeadProcess TestTransport{..} mOrL un = do
329 runProcess localNode $ do
279 runProcess localNode $ do
330 monitor theirAddr
280 monitor theirAddr
331 -- wait for the process to die
281 -- wait for the process to die
332− expect :: Process ProcessMonitorNotification
282+ ProcessMonitorNotification _ _ _ <- expect
333 monitorTestProcess theirAddr mOrL un DiedUnknownId Nothing done
283 monitorTestProcess theirAddr mOrL un DiedUnknownId Nothing done
334
284
335 takeMVar done
285 takeMVar done
@@ -403,17 +353,14 @@ testMath TestTransport{..} = do
403 runProcess localNode $ do
353 runProcess localNode $ do
404 pid <- getSelfPid
354 pid <- getSelfPid
405 send mathServer (Add pid 1 2)
355 send mathServer (Add pid 1 2)
406− three <- expect :: Process Double
356+ 3 <- expect :: Process Double
407 send mathServer (Divide pid 8 2)
357 send mathServer (Divide pid 8 2)
408− four <- expect :: Process Double
358+ 4 <- expect :: Process Double
409 send mathServer (Divide pid 8 0)
359 send mathServer (Divide pid 8 0)
410− divByZ <- expect
360+ DivByZero <- expect
411− liftIO $ putMVar clientDone (three, four, divByZ)
361+ liftIO $ putMVar clientDone ()
412
362
413− res <- takeMVar clientDone
363+ takeMVar clientDone
414− case res of
415− (3, 4, DivByZero) -> return ()
416− _ -> error $ "Something went horribly wrong"
417
364
418-- | Send first message (i.e. connect) to an already terminated process
365-- | Send first message (i.e. connect) to an already terminated process
419-- (without monitoring); then send another message to a second process on
366-- (without monitoring); then send another message to a second process on
@@ -441,9 +388,11 @@ testSendToTerminated TestTransport{..} = do
441 pid <- getSelfPid
388 pid <- getSelfPid
442 send server1 "Hi"
389 send server1 "Hi"
443 send server2 (Pong pid)
390 send server2 (Pong pid)
391+ Ping pid' <- expect
392+ True <- return $ pid' == server2
444− expectPing clientDone
393+ liftIO $ putMVar clientDone ()
445
394
446− verifyClient "Expected Ping from server" clientDone
395+ takeMVar clientDone
447
396
448-- | Test (non-zero) timeout
397-- | Test (non-zero) timeout
449testTimeout :: TestTransport -> Assertion
398testTimeout :: TestTransport -> Assertion
@@ -452,10 +401,10 @@ testTimeout TestTransport{..} = do
452 done <- newEmptyMVar
401 done <- newEmptyMVar
453
402
454 runProcess localNode $ do
403 runProcess localNode $ do
455− res <- receiveTimeout 1000000 [match (\Add{} -> return ())]
404+ Nothing <- receiveTimeout 1000000 [match (\(Add _ _ _) -> return ())]
456− liftIO $ putMVar done $ res == Nothing
405+ liftIO $ putMVar done ()
457
406
458− verifyClient "Expected receiveTimeout to timeout..." done
407+ takeMVar done
459
408
460-- | Test zero timeout
409-- | Test zero timeout
461testTimeout0 :: TestTransport -> Assertion
410testTimeout0 :: TestTransport -> Assertion
@@ -483,9 +432,10 @@ testTimeout0 TestTransport{..} = do
483 -- is not interested in, and then a single message that it wants
432 -- is not interested in, and then a single message that it wants
484 replicateM_ 10000 $ send server "Irrelevant message"
433 replicateM_ 10000 $ send server "Irrelevant message"
485 send server (Pong pid)
434 send server (Pong pid)
435+ Ping _ <- expect
486− expectPing clientDone
436+ liftIO $ putMVar clientDone ()
487
437
488− verifyClient "Expected Ping from server" clientDone
438+ takeMVar clientDone
489
439
490-- | Test typed channels
440-- | Test typed channels
491testTypedChannels :: TestTransport -> Assertion
441testTypedChannels :: TestTransport -> Assertion
@@ -508,10 +458,10 @@ testTypedChannels TestTransport{..} = do
508 runProcess localNode $ do
458 runProcess localNode $ do
509 (clientSendPort, rport) <- newChan
459 (clientSendPort, rport) <- newChan
510 sendChan serverSendPort (clientSendPort, 5)
460 sendChan serverSendPort (clientSendPort, 5)
511− ch <- receiveChan rport
461+ False <- receiveChan rport
512− liftIO $ putMVar clientDone $ ch == False
462+ liftIO $ putMVar clientDone ()
513
463
514− verifyClient "Expected channel to send 'False'" clientDone
464+ takeMVar clientDone
515
465
516-- | Test merging receive ports
466-- | Test merging receive ports
517testMergeChannels :: TestTransport -> Assertion
467testMergeChannels :: TestTransport -> Assertion
@@ -534,8 +484,9 @@ testMergeChannels TestTransport{..} = do
534 rs <- mapM charChannel "abc"
484 rs <- mapM charChannel "abc"
535 m <- mergePorts biased rs
485 m <- mergePorts biased rs
536 xs <- replicateM 9 $ receiveChan m
486 xs <- replicateM 9 $ receiveChan m
537− liftIO $ putMVar done $ xs == expected
487+ True <- return $ xs == expected
538− verifyClient "Expected single layer merge to match expected ordering" done
488+ liftIO $ putMVar done ()
489+ takeMVar done
539
490
540 -- Two layers of merging
491 -- Two layers of merging
541 testNested :: LocalNode -> Bool -> Bool -> String -> IO ()
492 testNested :: LocalNode -> Bool -> Bool -> String -> IO ()
@@ -546,8 +497,9 @@ testMergeChannels TestTransport{..} = do
546 ms <- mapM (mergePorts biasedInner) rss
497 ms <- mapM (mergePorts biasedInner) rss
547 m <- mergePorts biasedOuter ms
498 m <- mergePorts biasedOuter ms
548 xs <- replicateM (9 * 3) $ receiveChan m
499 xs <- replicateM (9 * 3) $ receiveChan m
549− liftIO $ putMVar done $ xs == expected
500+ True <- return $ xs == expected
550− verifyClient "Expected nested channels to match expeted ordering" done
501+ liftIO $ putMVar done ()
502+ takeMVar done
551
503
552 -- Test that if no messages are (immediately) available, the scheduler makes no difference
504 -- Test that if no messages are (immediately) available, the scheduler makes no difference
553 testBlocked :: LocalNode -> Bool -> IO ()
505 testBlocked :: LocalNode -> Bool -> IO ()
@@ -556,45 +508,43 @@ testMergeChannels TestTransport{..} = do
556 done <- newEmptyMVar
508 done <- newEmptyMVar
557
509
558 forkProcess localNode $ do
510 forkProcess localNode $ do
559− ss <- liftIO $ mapM readMVar vs
511+ [sa, sb, sc] <- liftIO $ mapM readMVar vs
560− case ss of
561− [sa, sb, sc] ->
562− mapM_ ((>> pause 10000) . uncurry sendChan)
512+ mapM_ ((>> liftIO (threadDelay 10000)) . uncurry sendChan)
563− [ -- a, b, c
513+ [ -- a, b, c
564− (sa, 'a')
514+ (sa, 'a')
565− , (sb, 'b')
515+ , (sb, 'b')
566− , (sc, 'c')
516+ , (sc, 'c')
567− -- a, c, b
517+ -- a, c, b
568− , (sa, 'a')
518+ , (sa, 'a')
569− , (sc, 'c')
519+ , (sc, 'c')
570− , (sb, 'b')
520+ , (sb, 'b')
571− -- b, a, c
521+ -- b, a, c
572− , (sb, 'b')
522+ , (sb, 'b')
573− , (sa, 'a')
523+ , (sa, 'a')
574− , (sc, 'c')
524+ , (sc, 'c')
575− -- b, c, a
525+ -- b, c, a
576− , (sb, 'b')
526+ , (sb, 'b')
577− , (sc, 'c')
527+ , (sc, 'c')
578− , (sa, 'a')
528+ , (sa, 'a')
579− -- c, a, b
529+ -- c, a, b
580− , (sc, 'c')
530+ , (sc, 'c')
581− , (sa, 'a')
531+ , (sa, 'a')
582− , (sb, 'b')
532+ , (sb, 'b')
583− -- c, b, a
533+ -- c, b, a
584− , (sc, 'c')
534+ , (sc, 'c')
585− , (sb, 'b')
535+ , (sb, 'b')
586− , (sa, 'a')
536+ , (sa, 'a')
587− ]
537+ ]
588− _ -> die "Something went horribly wrong"
589
538
590 forkProcess localNode $ do
539 forkProcess localNode $ do
591 (ss, rs) <- unzip <$> replicateM 3 newChan
540 (ss, rs) <- unzip <$> replicateM 3 newChan
592 liftIO $ mapM_ (uncurry putMVar) $ zip vs ss
541 liftIO $ mapM_ (uncurry putMVar) $ zip vs ss
593 m <- mergePorts biased rs
542 m <- mergePorts biased rs
594 xs <- replicateM (6 * 3) $ receiveChan m
543 xs <- replicateM (6 * 3) $ receiveChan m
595− liftIO $ putMVar done $ xs == "abcacbbacbcacabcba"
544+ True <- return $ xs == "abcacbbacbcacabcba"
545+ liftIO $ putMVar done ()
596
546
597− verifyClient "Expected merged ports to match expected ordering" done
547+ takeMVar done
598
548
599 mergePorts :: Serializable a => Bool -> [ReceivePort a] -> Process (ReceivePort a)
549 mergePorts :: Serializable a => Bool -> [ReceivePort a] -> Process (ReceivePort a)
600 mergePorts True = mergePortsBiased
550 mergePorts True = mergePortsBiased
@@ -612,9 +562,8 @@ testTerminate TestTransport{..} = do
612 localNode <- newLocalNode testTransport initRemoteTable
562 localNode <- newLocalNode testTransport initRemoteTable
613 runProcess localNode $ do
563 runProcess localNode $ do
614 e <- try terminate :: Process (Either ProcessTerminationException ())
564 e <- try terminate :: Process (Either ProcessTerminationException ())
615− if either show show e == show ProcessTerminationException
565+ True <- return $ either show show e == show ProcessTerminationException
616− then return ()
566+ return ()
617− else die "Unexpected result from terminate"
618
567
619testMonitorNode :: TestTransport -> Assertion
568testMonitorNode :: TestTransport -> Assertion
620testMonitorNode TestTransport{..} = do
569testMonitorNode TestTransport{..} = do
@@ -625,12 +574,11 @@ testMonitorNode TestTransport{..} = do
625
574
626 runProcess node2 $ do
575 runProcess node2 $ do
627 ref <- monitorNode (localNodeId node1)
576 ref <- monitorNode (localNodeId node1)
628− receiveWait [
629− match (\(NodeMonitorNotification ref' nid DiedDisconnect) ->
577+ NodeMonitorNotification ref' nid DiedDisconnect <- expect
630− return $ ref == ref' && nid == localNodeId node1)
578+ True <- return $ ref == ref' && nid == localNodeId node1
631− ] >>= liftIO . putMVar done
579+ liftIO $ putMVar done ()
632
580
633− verifyClient "Expected NodeMonitorNotification with matching ref & nodeId" done
581+ takeMVar done
634
582
635testMonitorLiveNode :: TestTransport -> Assertion
583testMonitorLiveNode :: TestTransport -> Assertion
636testMonitorLiveNode TestTransport{..} = do
584testMonitorLiveNode TestTransport{..} = do
@@ -645,16 +593,15 @@ testMonitorLiveNode TestTransport{..} = do
645 liftIO $ putMVar ready ()
593 liftIO $ putMVar ready ()
646 liftIO $ takeMVar readyr
594 liftIO $ takeMVar readyr
647 send p ()
595 send p ()
648− receiveWait [
649− match (\(NodeMonitorNotification ref' nid _) ->
596+ NodeMonitorNotification ref' nid _ <- expect
650− (return $ ref == ref' && nid == localNodeId node1))
597+ True <- return $ ref == ref' && nid == localNodeId node1
651− ] >>= liftIO . putMVar done
598+ liftIO $ putMVar done ()
652
599
653 takeMVar ready
600 takeMVar ready
654 closeLocalNode node1
601 closeLocalNode node1
655 putMVar readyr ()
602 putMVar readyr ()
656
603
657− verifyClient "Expected NodeMonitorNotification for LIVE node" done
604+ takeMVar done
658
605
659testMonitorChannel :: TestTransport -> Assertion
606testMonitorChannel :: TestTransport -> Assertion
660testMonitorChannel TestTransport{..} = do
607testMonitorChannel TestTransport{..} = do
@@ -664,20 +611,18 @@ testMonitorChannel TestTransport{..} = do
664 pid <- forkProcess node1 $ do
611 pid <- forkProcess node1 $ do
665 sport <- expect :: Process (SendPort ())
612 sport <- expect :: Process (SendPort ())
666 ref <- monitorPort sport
613 ref <- monitorPort sport
667− receiveWait [
614+ PortMonitorNotification ref' port' reason <- expect
668− -- reason might be DiedUnknownId if the receive port is GCed before the
615+ -- reason might be DiedUnknownId if the receive port is GCed before the
669− -- monitor is established (TODO: not sure that this is reasonable)
616+ -- monitor is established (TODO: not sure that this is reasonable)
670− match (\(PortMonitorNotification ref' port' reason) ->
671− return $ ref' == ref && port' == sendPortId sport &&
672− (reason == DiedNormal || reason == DiedUnknownId))
617+ return $ ref' == ref && port' == sendPortId sport && (reason == DiedNormal || reason == DiedUnknownId)
673− ] >>= liftIO . putMVar gotNotification
618+ liftIO $ putMVar gotNotification ()
674
619
675 runProcess node2 $ do
620 runProcess node2 $ do
676 (sport, _) <- newChan :: Process (SendPort (), ReceivePort ())
621 (sport, _) <- newChan :: Process (SendPort (), ReceivePort ())
677 send pid sport
622 send pid sport
678 liftIO $ threadDelay 100000
623 liftIO $ threadDelay 100000
679
624
680− verifyClient "Expected PortMonitorNotification" gotNotification
625+ takeMVar gotNotification
681
626
682testRegistry :: TestTransport -> Assertion
627testRegistry :: TestTransport -> Assertion
683testRegistry TestTransport{..} = do
628testRegistry TestTransport{..} = do
@@ -689,28 +634,20 @@ testRegistry TestTransport{..} = do
689
634
690 runProcess node $ do
635 runProcess node $ do
691 register "ping" pingServer
636 register "ping" pingServer
692− whereis "ping" >>= liftIO . assertBool "Unexpected ping" . (== Just pingServer)
637+ Just pid <- whereis "ping"
638+ True <- return $ pingServer == pid
693 us <- getSelfPid
639 us <- getSelfPid
694 nsend "ping" (Pong us)
640 nsend "ping" (Pong us)
695− receiveWait [
641+ Ping pid' <- expect
696− matchIf (\(Ping pid') -> pingServer == pid') (const $ return ())
642+ True <- return $ pingServer == pid'
697− ]
643+ Left (ProcessRegistrationException "dead" Nothing) <- try $ register "dead" deadProcess
698− checkRegException "dead" Nothing deadProcess
644+ Left (ProcessRegistrationException "ping" (Just x)) <- try $ register "ping" deadProcess
699− checkRegException "ping" (Just pingServer) deadProcess
645+ True <- return $ x == pingServer
700− try (unregister "dead") >>= checkReg "dead" Nothing
646+ Left (ProcessRegistrationException "dead" Nothing) <- try $ unregister "dead"
701 liftIO $ putMVar done ()
647 liftIO $ putMVar done ()
702
648
703 takeMVar done
649 takeMVar done
704
650
705− where
706− checkRegException name pid dead =
707− try (register name dead) >>= checkReg name pid
708−
709− checkReg name pid res =
710− case res of
711− Left (ProcessRegistrationException name pid) -> return ()
712− _ -> die $ "Unexpected Registration" ++ show res
713−
714testRegistryRemoteProcess :: TestTransport -> Assertion
651testRegistryRemoteProcess :: TestTransport -> Assertion
715testRegistryRemoteProcess TestTransport{..} = do
652testRegistryRemoteProcess TestTransport{..} = do
716 node1 <- newLocalNode testTransport initRemoteTable
653 node1 <- newLocalNode testTransport initRemoteTable
@@ -721,13 +658,13 @@ testRegistryRemoteProcess TestTransport{..} = do
721
658
722 runProcess node2 $ do
659 runProcess node2 $ do
723 register "ping" pingServer
660 register "ping" pingServer
724− whereis "ping" >>= liftIO . assertBool "Unexpected ping" . (== Just pingServer)
661+ Just pid <- whereis "ping"
662+ True <- return $ pingServer == pid
725 us <- getSelfPid
663 us <- getSelfPid
726 nsend "ping" (Pong us)
664 nsend "ping" (Pong us)
727− receiveWait [
665+ Ping pid' <- expect
728− matchIf (\(Ping pid') -> pingServer == pid')
666+ True <- return $ pingServer == pid'
729− (const $ liftIO $ putMVar done ())
667+ liftIO $ putMVar done ()
730− ]
731
668
732 takeMVar done
669 takeMVar done
733
670
@@ -735,6 +672,7 @@ testRemoteRegistry :: TestTransport -> Assertion
735testRemoteRegistry TestTransport{..} = do
672testRemoteRegistry TestTransport{..} = do
736 node1 <- newLocalNode testTransport initRemoteTable
673 node1 <- newLocalNode testTransport initRemoteTable
737 node2 <- newLocalNode testTransport initRemoteTable
674 node2 <- newLocalNode testTransport initRemoteTable
675+ done <- newEmptyMVar
738
676
739 pingServer <- forkProcess node1 ping
677 pingServer <- forkProcess node1 ping
740 deadProcess <- forkProcess node1 (return ())
678 deadProcess <- forkProcess node1 (return ())
@@ -747,34 +685,32 @@ testRemoteRegistry TestTransport{..} = do
747 "ping" == label' && pid == pingServer)
685 "ping" == label' && pid == pingServer)
748 (\(RegisterReply _ _ _) -> return ()) ]
686 (\(RegisterReply _ _ _) -> return ()) ]
749
687
750− pid <- verifyWhereIsRemote nid1 "ping"
688+ Just pid <- whereisRemote nid1 "ping"
751− liftIO $ assertBool "Expected pindServer to match pid" $ pingServer == pid
689+ True <- return $ pingServer == pid
752−
753 us <- getSelfPid
690 us <- getSelfPid
754 nsendRemote nid1 "ping" (Pong us)
691 nsendRemote nid1 "ping" (Pong us)
755− receiveWait [
692+ Ping pid' <- expect
756− match (\(Ping pid') -> return $ pingServer == pid')
693+ True <- return $ pingServer == pid'
757− ] >>= liftIO . assertBool "Expected Ping with ping server's ProcessId"
758
694
759 -- test that if process was not registered Nothing is returned
695 -- test that if process was not registered Nothing is returned
760 -- in owner field.
696 -- in owner field.
761 registerRemoteAsync nid1 "dead" deadProcess
697 registerRemoteAsync nid1 "dead" deadProcess
698+ receiveWait [
762− receiveWait [ matchIf (\(RegisterReply label' f mPid) -> "dead" == label')
699+ matchIf (\(RegisterReply label' False Nothing) -> "dead" == label')
763− (\(RegisterReply _ f mPid) -> return (not f && isNothing mPid))
700+ (\(RegisterReply _ _ _) -> return ()) ]
764− ] >>= liftIO . assertBool "Expected False Nothing in RegisterReply"
765−
766 registerRemoteAsync nid1 "ping" deadProcess
701 registerRemoteAsync nid1 "ping" deadProcess
767 receiveWait [
702 receiveWait [
768− matchIf (\(RegisterReply label' False mPid) ->
703+ matchIf (\(RegisterReply label' False (Just pid)) ->
769− "ping" == label' && isJust mPid)
704+ "ping" == label' && pid == pingServer)
770− (\(RegisterReply _ f (Just pid'')) -> return (not f && pid'' == pingServer))
705+ (\(RegisterReply _ _ _) -> return ()) ]
771− ] >>= liftIO . assertBool "Expected False and (Just alreadyRegisteredPid) in RegisterReply"
772−
773 unregisterRemoteAsync nid1 "dead"
706 unregisterRemoteAsync nid1 "dead"
774 receiveWait [
707 receiveWait [
775− matchIf (\(RegisterReply label' _ _) -> "dead" == label')
708+ matchIf (\(RegisterReply label' False Nothing) ->
709+ "dead" == label' && pid == pingServer)
776− (\(RegisterReply _ f mPid) -> return (not f && isNothing mPid))
710+ (\(RegisterReply _ _ _) -> return ()) ]
777− ] >>= liftIO . assertBool "Expected False and Nothing in RegisterReply"
711+ liftIO $ putMVar done ()
712+
713+ takeMVar done
778
714
779testRemoteRegistryRemoteProcess :: TestTransport -> Assertion
715testRemoteRegistryRemoteProcess :: TestTransport -> Assertion
780testRemoteRegistryRemoteProcess TestTransport{..} = do
716testRemoteRegistryRemoteProcess TestTransport{..} = do
@@ -790,15 +726,15 @@ testRemoteRegistryRemoteProcess TestTransport{..} = do
790 receiveWait [
726 receiveWait [
791 matchIf (\(RegisterReply label' _ _) -> "ping" == label')
727 matchIf (\(RegisterReply label' _ _) -> "ping" == label')
792 (\(RegisterReply _ _ _) -> return ()) ]
728 (\(RegisterReply _ _ _) -> return ()) ]
793− pid <- verifyWhereIsRemote nid1 "ping"
729+ Just pid <- whereisRemote nid1 "ping"
794− liftIO $ assertBool "Expected pingServer to match remote name" $ pingServer == pid
730+ True <- return $ pingServer == pid
795 us <- getSelfPid
731 us <- getSelfPid
796 nsendRemote nid1 "ping" (Pong us)
732 nsendRemote nid1 "ping" (Pong us)
797− receiveWait [
733+ Ping pid' <- expect
798− match (\(Ping pid') -> return $ pingServer == pid')
734+ True <- return $ pingServer == pid'
799− ] >>= liftIO . putMVar done
735+ liftIO $ putMVar done ()
800
736
801− verifyClient "Expected Ping with ping server's ProcessId" done
737+ takeMVar done
802
738
803testSpawnLocal :: TestTransport -> Assertion
739testSpawnLocal :: TestTransport -> Assertion
804testSpawnLocal TestTransport{..} = do
740testSpawnLocal TestTransport{..} = do
@@ -812,14 +748,15 @@ testSpawnLocal TestTransport{..} = do
812 sport <- expect
748 sport <- expect
813 sendChan sport (1234 :: Int)
749 sendChan sport (1234 :: Int)
814
750
815− sport <- spawnChannelLocal $
751+ sport <- spawnChannelLocal $ \rport -> do
816− \rport -> (receiveChan rport :: Process Int) >>= send us
752+ (1234 :: Int) <- receiveChan rport
753+ send us ()
817
754
818 send pid sport
755 send pid sport
756+ () <- expect
819− expect >>= liftIO . putMVar done
757+ liftIO $ putMVar done ()
820
758
821− res <- takeMVar done
759+ takeMVar done
822− assertBool "Expected 1234 :: Int" $ res == (1234 :: Int)
823
760
824testSpawnAsyncStrictness :: TestTransport -> Assertion
761testSpawnAsyncStrictness :: TestTransport -> Assertion
825testSpawnAsyncStrictness TestTransport{..} = do
762testSpawnAsyncStrictness TestTransport{..} = do
@@ -848,9 +785,8 @@ testReconnect TestTransport{..} = do
848 liftIO $ putMVar processA us
785 liftIO $ putMVar processA us
849 msg1 <- expect
786 msg1 <- expect
850 msg2 <- expect
787 msg2 <- expect
851− liftIO $ do
852− assertBool "messages did not match" $ msg1 == "message 1" && msg2 == "message 3"
788+ True <- return $ msg1 == "message 1" && msg2 == "message 3"
853− putMVar sendTestOk ()
789+ liftIO $ putMVar sendTestOk ()
854
790
855 forkProcess node2 $ do
791 forkProcess node2 $ do
856 {-
792 {-
@@ -883,7 +819,7 @@ testReconnect TestTransport{..} = do
883 matchIf (\(RegisterReply label' _ _) -> "a" == label')
819 matchIf (\(RegisterReply label' _ _) -> "a" == label')
884 (\(RegisterReply _ _ _) -> return ()) ]
820 (\(RegisterReply _ _ _) -> return ()) ]
885
821
886− _ <- whereisRemote nid1 "a"
822+ Just _ <- whereisRemote nid1 "a"
887
823
888
824
889 -- Simulate network failure
825 -- Simulate network failure
@@ -902,10 +838,9 @@ testReconnect TestTransport{..} = do
902 (\(RegisterReply _ _ _) -> return ()) ]
838 (\(RegisterReply _ _ _) -> return ()) ]
903
839
904 -- Check
840 -- Check
905− mPid <- whereisRemote nid1 "a" -- this will fail because the name is removed when the node is disconnected
841+ Nothing <- whereisRemote nid1 "a" -- this will fail because the name is removed when the node is disconnected
906− liftIO $ assertBool "Expected remote name to be lost" $ mPid == Nothing
907− verifyWhereIsRemote nid1 "b" -- this will suceed because the value is set after thereconnect
842+ Just _ <- whereisRemote nid1 "b" -- this will suceed because the value is set after thereconnect
908− verifyWhereIsRemote nid1 "c"
843+ Just _ <- whereisRemote nid1 "c"
909
844
910 liftIO $ putMVar registerTestOk ()
845 liftIO $ putMVar registerTestOk ()
911
846
@@ -946,19 +881,19 @@ testUSend usendPrim TestTransport{..} numMessages = do
946 isSorted _ = True
881 isSorted _ = True
947 -- The list can't be null since there are no failures after sending
882 -- The list can't be null since there are no failures after sending
948 -- the last message.
883 -- the last message.
949− liftIO $ putMVar usendTestOk $ isSorted msgs && not (null msgs)
884+ True <- return $ isSorted msgs && not (null msgs)
885+ liftIO $ putMVar usendTestOk ()
950
886
951 forkProcess node2 $ do
887 forkProcess node2 $ do
952 them <- liftIO $ readMVar processA
888 them <- liftIO $ readMVar processA
953 getSelfPid >>= send them
889 getSelfPid >>= send them
954− expect :: Process ()
890+ () <- expect
955 forM_ [1..numMessages] $ \i -> do
891 forM_ [1..numMessages] $ \i -> do
956 liftIO $ testBreakConnection (nodeAddress nid1) (nodeAddress nid2)
892 liftIO $ testBreakConnection (nodeAddress nid1) (nodeAddress nid2)
957 usendPrim them i
893 usendPrim them i
958 liftIO (threadDelay 30000)
894 liftIO (threadDelay 30000)
959
895
960− res <- takeMVar usendTestOk
896+ takeMVar usendTestOk
961− assertBool "Unexpected failure after sending last msg" res
962
897
963-- | Test 'matchAny'. This repeats the 'testMath' but with a proxy server
898-- | Test 'matchAny'. This repeats the 'testMath' but with a proxy server
964-- in between
899-- in between
@@ -984,17 +919,14 @@ testMatchAny TestTransport{..} = do
984 runProcess localNode $ do
919 runProcess localNode $ do
985 pid <- getSelfPid
920 pid <- getSelfPid
986 send mathServer (Add pid 1 2)
921 send mathServer (Add pid 1 2)
987− three <- expect :: Process Double
922+ 3 <- expect :: Process Double
988 send mathServer (Divide pid 8 2)
923 send mathServer (Divide pid 8 2)
989− four <- expect :: Process Double
924+ 4 <- expect :: Process Double
990 send mathServer (Divide pid 8 0)
925 send mathServer (Divide pid 8 0)
991− divByZ <- expect
926+ DivByZero <- expect
992− liftIO $ putMVar clientDone (three, four, divByZ)
927+ liftIO $ putMVar clientDone ()
993
928
994− res <- takeMVar clientDone
929+ takeMVar clientDone
995− case res of
996− (3, 4, DivByZero) -> return ()
997− _ -> error "Unexpected result"
998
930
999-- | Test 'matchAny'. This repeats the 'testMath' but with a proxy server
931-- | Test 'matchAny'. This repeats the 'testMath' but with a proxy server
1000-- in between, however we block 'Divide' requests ....
932-- in between, however we block 'Divide' requests ....
@@ -1021,13 +953,12 @@ testMatchAnyHandle TestTransport{..} = do
1021 runProcess localNode $ do
953 runProcess localNode $ do
1022 pid <- getSelfPid
954 pid <- getSelfPid
1023 send mathServer (Add pid 1 2)
955 send mathServer (Add pid 1 2)
1024− three <- expect :: Process Double
956+ 3 <- expect :: Process Double
1025 send mathServer (Divide pid 8 2)
957 send mathServer (Divide pid 8 2)
1026− res <- (expectTimeout 100000) :: Process (Maybe Double)
958+ Nothing <- (expectTimeout 100000) :: Process (Maybe Double)
1027− liftIO $ putMVar clientDone $ three == 3 && res == Nothing
959+ liftIO $ putMVar clientDone ()
1028−
1029− verifyClient "Expected Nothing (i.e. timeout)" clientDone
1030
960
961+ takeMVar clientDone
1031 where maybeForward :: ProcessId -> Message -> Process (Maybe ())
962 where maybeForward :: ProcessId -> Message -> Process (Maybe ())
1032 maybeForward s msg =
963 maybeForward s msg =
1033 handleMessage msg (\m@(Add _ _ _) -> send s m)
964 handleMessage msg (\m@(Add _ _ _) -> send s m)
@@ -1055,10 +986,8 @@ testMatchAnyNoHandle TestTransport{..} = do
1055 Just _ -> die "NONSENSE")
986 Just _ -> die "NONSENSE")
1056 ]
987 ]
1057 -- we *must* have removed the message from our mailbox though!!!
988 -- we *must* have removed the message from our mailbox though!!!
1058− res <- receiveTimeout 100000 [ match (\(Add _ _ _) -> return ()) ]
989+ Nothing <- receiveTimeout 100000 [ match (\(Add _ _ _) -> return ()) ]
1059− liftIO $ do
1060− assertBool "Expected timeout!" $ res == Nothing
1061− putMVar serverDone ()
990+ liftIO $ putMVar serverDone ()
1062 putMVar addr server
991 putMVar addr server
1063
992
1064 -- Client
993 -- Client
@@ -1101,21 +1030,14 @@ testMatchAnyIf TestTransport{..} = do
1101 runProcess localNode $ do
1030 runProcess localNode $ do
1102 pid <- getSelfPid
1031 pid <- getSelfPid
1103 send server (pid, "foo")
1032 send server (pid, "foo")
1104− foo <- expect
1033+ "foo" <- expect
1105− -- provoking what would be the wrong ordering is informative here...
1106−
1107− send server (pid, "bar")
1108− bar <- (expectTimeout 100000) :: Process (Maybe String) -- was Double o_O !?
1109−
1110 send server (pid, "baz")
1034 send server (pid, "baz")
1111− baz <- expect
1035+ "baz" <- expect
1112−
1036+ send server (pid, "bar")
1037+ Nothing <- (expectTimeout 100000) :: Process (Maybe Double)
1113− liftIO $ putMVar clientDone (foo, bar, baz)
1038+ liftIO $ putMVar clientDone ()
1114−
1115− res <- takeMVar clientDone
1116− let res' = res == ("foo", Nothing, "baz")
1117− assertBool "Expected timeout due to type mismatch" res'
1118
1039
1040+ takeMVar clientDone
1119 where tryHandleMessage :: Message -> Process (Maybe ())
1041 where tryHandleMessage :: Message -> Process (Maybe ())
1120 tryHandleMessage msg =
1042 tryHandleMessage msg =
1121 handleMessage msg (\(pid :: ProcessId, (m :: String))
1043 handleMessage msg (\(pid :: ProcessId, (m :: String))
@@ -1148,13 +1070,12 @@ testMatchMessageWithUnwrap TestTransport{..} = do
1148 runProcess localNode $ do
1070 runProcess localNode $ do
1149 pid <- getSelfPid
1071 pid <- getSelfPid
1150 send server (pid, wrapMessage ("foo" :: String))
1072 send server (pid, wrapMessage ("foo" :: String))
1151− foo <- expect
1073+ "foo" <- expect
1152 send server (pid, wrapMessage ("baz" :: String))
1074 send server (pid, wrapMessage ("baz" :: String))
1153− baz <- expect
1075+ "baz" <- expect
1154− liftIO $ putMVar clientDone (foo, baz)
1076+ liftIO $ putMVar clientDone ()
1155
1077
1156− res <- takeMVar clientDone
1078+ takeMVar clientDone
1157− assertBool "Unexpected unwrapped results" $ res == ("foo", "baz")
1158
1079
1159-- Test 'receiveChanTimeout'
1080-- Test 'receiveChanTimeout'
1160testReceiveChanTimeout :: TestTransport -> Assertion
1081testReceiveChanTimeout :: TestTransport -> Assertion
@@ -1172,19 +1093,18 @@ testReceiveChanTimeout TestTransport{..} = do
1172
1093
1173 -- Wait for a message with a delay. No message arrives, we should get
1094 -- Wait for a message with a delay. No message arrives, we should get
1174 -- Nothing after the delay.
1095 -- Nothing after the delay.
1175− receiveChanTimeout 100000 rp >>= maybe (return ()) (const $ die "Expected Timeout")
1096+ Nothing <- receiveChanTimeout 100000 rp
1176
1097
1177 -- Let the sender know that it can send a message.
1098 -- Let the sender know that it can send a message.
1178 liftIO $ putMVar mvSender ()
1099 liftIO $ putMVar mvSender ()
1179
1100
1180 -- Wait for a message with a delay again. Now a message arrives after
1101 -- Wait for a message with a delay again. Now a message arrives after
1181 -- 0.1 seconds
1102 -- 0.1 seconds
1182− res <- receiveChanTimeout 20000000 rp >>= maybe (die "Timeout") return
1103+ Just True <- receiveChanTimeout 20000000 rp
1183− liftIO $ assertBool "Expected result to be 'True'" res
1184
1104
1185 -- Wait for a message with zero timeout: non-blocking check. No message is
1105 -- Wait for a message with zero timeout: non-blocking check. No message is
1186 -- available, we get Nothing
1106 -- available, we get Nothing
1187− receiveChanTimeout 0 rp >>= maybe (return ()) (const $ die "Expected Timeout")
1107+ Nothing <- receiveChanTimeout 0 rp
1188
1108
1189 -- Let the sender know that it can send a message.
1109 -- Let the sender know that it can send a message.
1190 liftIO $ putMVar mvSender ()
1110 liftIO $ putMVar mvSender ()
@@ -1194,7 +1114,7 @@ testReceiveChanTimeout TestTransport{..} = do
1194 liftIO $ threadDelay 100000
1114 liftIO $ threadDelay 100000
1195 mb <- receiveChanTimeout 0 rp
1115 mb <- receiveChanTimeout 0 rp
1196 case mb of
1116 case mb of
1197− Just b -> do liftIO $ assertBool "Unexpected Message" $ not b
1117+ Just b -> do False <- return b; return ()
1198 _ -> loop
1118 _ -> loop
1199
1119
1200 liftIO $ putMVar done ()
1120 liftIO $ putMVar done ()
@@ -1231,8 +1151,8 @@ testReceiveChanFeatures TestTransport{..} = do
1231
1151
1232 rp1 <- mergePortsBiased [even <$> rpInt, rpBool]
1152 rp1 <- mergePortsBiased [even <$> rpInt, rpBool]
1233
1153
1234− receiveChan rp1 >>= liftIO . assertBool "Expected True"
1154+ True <- receiveChan rp1
1235− receiveChan rp1 >>= liftIO . assertBool "Expected False" . not
1155+ False <- receiveChan rp1
1236
1156
1237 -- Test Applicative instance
1157 -- Test Applicative instance
1238
1158
@@ -1241,7 +1161,7 @@ testReceiveChanFeatures TestTransport{..} = do
1241
1161
1242 let rp2 = pure (+) <*> rpInt <*> rpInt
1162 let rp2 = pure (+) <*> rpInt <*> rpInt
1243
1163
1244− receiveChan rp2 >>= liftIO . assertBool "Expected 7" . (== 7)
1164+ 7 <- receiveChan rp2
1245
1165
1246 -- Test Alternative instance
1166 -- Test Alternative instance
1247
1167
@@ -1250,8 +1170,8 @@ testReceiveChanFeatures TestTransport{..} = do
1250
1170
1251 let rp3 = (even <$> rpInt) <|> rpBool
1171 let rp3 = (even <$> rpInt) <|> rpBool
1252
1172
1253− receiveChan rp3 >>= liftIO . assertBool "Expected False" . not
1173+ False <- receiveChan rp3
1254− receiveChan rp3 >>= liftIO . assertBool "Expected True"
1174+ True <- receiveChan rp3
1255
1175
1256 -- Test Monad instance
1176 -- Test Monad instance
1257
1177
@@ -1265,86 +1185,17 @@ testReceiveChanFeatures TestTransport{..} = do
1265 then rpInt
1185 then rpInt
1266 else return 7
1186 else return 7
1267
1187
1268− receiveChan rp4 >>= liftIO . assertBool "Expected 5" . (== 5)
1188+ 5 <- receiveChan rp4
1269− receiveChan rp4 >>= liftIO . assertBool "Expected 7" . (== 7)
1189+ 7 <- receiveChan rp4
1270
1190
1271 liftIO $ putMVar done ()
1191 liftIO $ putMVar done ()
1272
1192
1273 takeMVar done
1193 takeMVar done
1274
1194
1275−testChanLifecycle :: TestTransport -> Assertion
1276−testChanLifecycle TestTransport{..} = let delay = 3000000 in do
1277− result <- newEmptyMVar
1278− tchMV <- newEmptyMVar
1279− localNode <- newLocalNode testTransport initRemoteTable
1280− runProcess localNode $ do
1281−
1282− pid <- spawnLocal $ do tCh <- newChan :: Process (SendPort (), ReceivePort ())
1283− liftIO $ putMVar tchMV tCh
1284− expect :: Process ()
1285− let (sp, _) = tCh
1286− sendChan sp ()
1287− expect :: Process ()
1288−
1289− mRefPid <- monitor pid
1290−
1291− cPid <- spawnLocal $ do
1292− (sp', rp) <- liftIO $ takeMVar tchMV
1293− -- say "picked up our test channel"
1294− send pid ()
1295− -- say "told pid to continue"
1296− res <- receiveChanTimeout delay rp
1297− case res of
1298− Nothing -> say "initial chan () missing!" >> (liftIO $ putMVar result False)
1299− Just () -> do mr <- monitor pid
1300− pause 10000
1301− -- say "sending pid a second () will cause it to exit"
1302− send pid ()
1303−
1304− -- say "make sure we see a DOWN notification for pid having stopped"
1305− receiveWait [ match (\(_ :: ProcessMonitorNotification) -> return ()) ]
1306−
1307− -- now that pid has died, the send port should be useless...
1308− liftIO $ putMVar tchMV (sp', rp)
1309−
1310− -- let's verify that we do not see the message from our
1311− -- parent process on the channel, once pid has died...
1312− recv <- receiveChanTimeout delay rp
1313− -- say $ "finished waiting for second (), writing result" ++ (show recv)
1314− liftIO $ putMVar result $ isNothing recv
1315−
1316− mRefCPid <- monitor cPid
1317−
1318− receiveWait
1319− [ matchIf (\(ProcessMonitorNotification r _ _) -> r == mRefPid)
1320− (\_ -> return ())
1321− ]
1322−
1323− -- say "seen first pid die..."
1324−
1325− (sendPort, _) <- liftIO $ takeMVar tchMV
1326− sendChan sendPort ()
1327− -- say "sent () after owning pid died"
1328−
1329− -- let cPid know we've written to the channel...
1330− send cPid ()
1331−
1332− receiveWait
1333− [ matchIf (\(ProcessMonitorNotification r _ _) -> r == mRefCPid)
1334− (\_ -> return ())
1335− ]
1336−
1337− -- say "seen both pids die now..."
1338−
1339− -- and wait on the result back in IO land...
1340− testRes <- takeMVar result
1341− -- runProcess localNode $ say "got result..."
1342− assertBool "Expected sending on the channel to fail, but received data!" testRes
1343−
1344−
1345testKillLocal :: TestTransport -> Assertion
1195testKillLocal :: TestTransport -> Assertion
1346testKillLocal TestTransport{..} = do
1196testKillLocal TestTransport{..} = do
1347 localNode <- newLocalNode testTransport initRemoteTable
1197 localNode <- newLocalNode testTransport initRemoteTable
1198+ done <- newEmptyMVar
1348
1199
1349 pid <- forkProcess localNode $ do
1200 pid <- forkProcess localNode $ do
1350 liftIO $ threadDelay 1000000
1201 liftIO $ threadDelay 1000000
@@ -1353,18 +1204,17 @@ testKillLocal TestTransport{..} = do
1353 ref <- monitor pid
1204 ref <- monitor pid
1354 us <- getSelfPid
1205 us <- getSelfPid
1355 kill pid "TestKill"
1206 kill pid "TestKill"
1356− mn <- expect
1207+ ProcessMonitorNotification ref' pid' (DiedException ex) <- expect
1357− case mn of
1358− ProcessMonitorNotification ref' pid' (DiedException ex) ->
1359− case ref == ref' && pid == pid' && ex == "killed-by=" ++ show us ++ ",reason=TestKill" of
1208+ True <- return $ ref == ref' && pid == pid' && ex == "killed-by=" ++ show us ++ ",reason=TestKill"
1360− True -> return ()
1209+ liftIO $ putMVar done ()
1361− False -> die "Invalid ProcessMonitorNotification received"
1210+
1362− _ -> die "Something went horribly wrong"
1211+ takeMVar done
1363
1212
1364testKillRemote :: TestTransport -> Assertion
1213testKillRemote :: TestTransport -> Assertion
1365testKillRemote TestTransport{..} = do
1214testKillRemote TestTransport{..} = do
1366 node1 <- newLocalNode testTransport initRemoteTable
1215 node1 <- newLocalNode testTransport initRemoteTable
1367 node2 <- newLocalNode testTransport initRemoteTable
1216 node2 <- newLocalNode testTransport initRemoteTable
1217+ done <- newEmptyMVar
1368
1218
1369 pid <- forkProcess node1 $ do
1219 pid <- forkProcess node1 $ do
1370 liftIO $ threadDelay 1000000
1220 liftIO $ threadDelay 1000000
@@ -1373,17 +1223,11 @@ testKillRemote TestTransport{..} = do
1373 ref <- monitor pid
1223 ref <- monitor pid
1374 us <- getSelfPid
1224 us <- getSelfPid
1375 kill pid "TestKill"
1225 kill pid "TestKill"
1376− mn <- expect
1377− case mn of
1378− ProcessMonitorNotification ref' pid' (DiedException reason) ->
1226+ ProcessMonitorNotification ref' pid' (DiedException reason) <- expect
1379− case (ref == ref', pid == pid', reason == "killed-by=" ++ show us ++ ",reason=TestKill") of
1227+ True <- return $ ref == ref' && pid == pid' && reason == "killed-by=" ++ show us ++ ",reason=TestKill"
1380− (True, True, True) -> return ()
1228+ liftIO $ putMVar done ()
1381− (a, b, c) -> do
1229+
1382− let a' = if a then "" else "Invalid ref"
1230+ takeMVar done
1383− let b' = if b then "" else "Invalid pid"
1384− let c' = if c then "" else "Invalid message"
1385− die $ unwords [a', b', c']
1386− _ -> die "Received unexpected message"
1387
1231
1388testCatchesExit :: TestTransport -> Assertion
1232testCatchesExit :: TestTransport -> Assertion
1389testCatchesExit TestTransport{..} = do
1233testCatchesExit TestTransport{..} = do
@@ -1409,10 +1253,7 @@ testHandleMessageIf TestTransport{..} = do
1409 self <- getSelfPid
1253 self <- getSelfPid
1410 send self (5 :: Integer, 10 :: Integer)
1254 send self (5 :: Integer, 10 :: Integer)
1411 msg <- receiveWait [ matchMessage return ]
1255 msg <- receiveWait [ matchMessage return ]
1412− handleMessageIf msg
1256+ Nothing <- handleMessageIf msg (\() -> True) (\() -> die $ "whoops")
1413− (\() -> True)
1414− (\() -> die "whoops") >>= maybe (return ())
1415− (const $ die "Expected Mismatch")
1416 handleMessageIf msg (\(x :: Integer, y :: Integer) -> x == 5 && y == 10)
1257 handleMessageIf msg (\(x :: Integer, y :: Integer) -> x == 5 && y == 10)
1417 (\input -> liftIO $ putMVar done input)
1258 (\input -> liftIO $ putMVar done input)
1418 return ()
1259 return ()
@@ -1457,11 +1298,8 @@ testDie TestTransport{..} = do
1457 (die ("foobar", 123 :: Int))
1298 (die ("foobar", 123 :: Int))
1458 `catchExit` \_from reason -> do
1299 `catchExit` \_from reason -> do
1459 -- TODO: should verify that 'from' has the right value
1300 -- TODO: should verify that 'from' has the right value
1460− let res = reason == ("foobar", 123 :: Int)
1301+ True <- return $ reason == ("foobar", 123 :: Int)
1461 liftIO $ putMVar done ()
1302 liftIO $ putMVar done ()
1462− if res
1463− then return ()
1464− else die "Something went horribly wrong"
1465
1303
1466 takeMVar done
1304 takeMVar done
1467
1305
@@ -1475,11 +1313,8 @@ testPrettyExit TestTransport{..} = do
1475 `catch` \ex@(ProcessExitException from _) ->
1313 `catch` \ex@(ProcessExitException from _) ->
1476 let expected = "exit-from=" ++ (show from)
1314 let expected = "exit-from=" ++ (show from)
1477 in do
1315 in do
1478− let res = (show ex) == expected
1316+ True <- return $ (show ex) == expected
1479 liftIO $ putMVar done ()
1317 liftIO $ putMVar done ()
1480− if res
1481− then return ()
1482− else die "Something went horribly wrong"
1483
1318
1484 takeMVar done
1319 takeMVar done
1485
1320
@@ -1495,26 +1330,17 @@ testExitLocal TestTransport{..} = do
1495 pid <- forkProcess localNode $ do
1330 pid <- forkProcess localNode $ do
1496 (liftIO (putMVar handlerSetUp ()) >> expect) `catchExit` \_from reason -> do
1331 (liftIO (putMVar handlerSetUp ()) >> expect) `catchExit` \_from reason -> do
1497 -- TODO: should verify that 'from' has the right value
1332 -- TODO: should verify that 'from' has the right value
1498− res <- return $ reason == "TestExit"
1333+ True <- return $ reason == "TestExit"
1499 liftIO $ putMVar supervisedDone ()
1334 liftIO $ putMVar supervisedDone ()
1500− if res
1501− then return ()
1502− else die "Something went horribly wrong"
1503
1335
1504 runProcess localNode $ do
1336 runProcess localNode $ do
1505 liftIO $ takeMVar handlerSetUp
1337 liftIO $ takeMVar handlerSetUp
1506 ref <- monitor pid
1338 ref <- monitor pid
1507 exit pid "TestExit"
1339 exit pid "TestExit"
1508 -- This time the client catches the exception, so it dies normally
1340 -- This time the client catches the exception, so it dies normally
1509− mn <- expect
1510− case mn of
1511− ProcessMonitorNotification ref' pid' DiedNormal -> do
1341+ ProcessMonitorNotification ref' pid' DiedNormal <- expect
1512− let res = ref == ref' && pid == pid'
1342+ True <- return $ ref == ref' && pid == pid'
1513− liftIO $ putMVar supervisorDone ()
1343+ liftIO $ putMVar supervisorDone ()
1514− if res
1515− then return ()
1516− else die "Something went horribly wrong"
1517− _ -> die "Something went horribly wrong"
1518
1344
1519 takeMVar supervisedDone
1345 takeMVar supervisedDone
1520 takeMVar supervisorDone
1346 takeMVar supervisorDone
@@ -1530,98 +1356,19 @@ testExitRemote TestTransport{..} = do
1530 (receiveWait [] :: Process ()) -- block forever
1356 (receiveWait [] :: Process ()) -- block forever
1531 `catchExit` \_from reason -> do
1357 `catchExit` \_from reason -> do
1532 -- TODO: should verify that 'from' has the right value
1358 -- TODO: should verify that 'from' has the right value
1533− res <- return $ reason == "TestExit"
1359+ True <- return $ reason == "TestExit"
1534 liftIO $ putMVar supervisedDone ()
1360 liftIO $ putMVar supervisedDone ()
1535− if res
1536− then return ()
1537− else die "Something went horribly wrong"
1538
1361
1539 runProcess node2 $ do
1362 runProcess node2 $ do
1540 ref <- monitor pid
1363 ref <- monitor pid
1541 exit pid "TestExit"
1364 exit pid "TestExit"
1542− mn <- expect
1543− case mn of
1544− ProcessMonitorNotification ref' pid' DiedNormal -> do
1365+ ProcessMonitorNotification ref' pid' DiedNormal <- expect
1545− res' <- return $ ref == ref' && pid == pid'
1366+ True <- return $ ref == ref' && pid == pid'
1546− liftIO $ putMVar supervisorDone ()
1367+ liftIO $ putMVar supervisorDone ()
1547− if res'
1548− then return ()
1549− else die "Something went horribly wrong"
1550− _ -> die "Something went horribly wrong"
1551
1368
1552 takeMVar supervisedDone
1369 takeMVar supervisedDone
1553 takeMVar supervisorDone
1370 takeMVar supervisorDone
1554
1371
1555−testRegistryMonitoring :: TestTransport -> Assertion
1556−testRegistryMonitoring TestTransport{..} = do
1557− node1 <- newLocalNode testTransport initRemoteTable
1558− node2 <- newLocalNode testTransport initRemoteTable
1559−
1560− let nid = localNodeId node2
1561− pid <- forkProcess node1 $ do
1562− self <- getSelfPid
1563− runUntilRegistered nid self
1564− say $ (show self) ++ " registered as " ++ regName
1565− expect :: Process ()
1566− say $ (show self) ++ " exiting normally"
1567−
1568− runProcess node2 $ do
1569− register regName pid
1570− say $ regName ++ " registered to " ++ show pid
1571− res <- whereis regName
1572− send pid ()
1573− say $ " sent finish signal to " ++ show pid
1574− us <- getSelfPid
1575− liftIO $ assertBool "expected (Just pid)" $ res == (Just pid)
1576−
1577−
1578− -- This delay isn't essential!
1579− -- The test case passes perfectly fine without it (feel free to comment out
1580− -- and see), however waiting a few seconds here, makes it much more likely
1581− -- that in delayUntilMaybeUnregistered we will hit the match case right
1582− -- away, and thus not be subjected to a 20 second delay. The value of 4
1583− -- seconds appears to work optimally on osx and across several linux distros
1584− -- running in virtual machines (which is essentially what we do in CI)
1585− void $ receiveTimeout 4000000 [ matchAny return ]
1586−
1587− -- This delay doesn't serve much purpose in the happy path, however if some
1588− -- future patch breaks the cooperative behaviour of node controllers viz
1589− -- remote process registration and notification taking place via ncEffectDied,
1590− -- there would be the possibility of a race in the test case should we attempt
1591− -- to evaluate `whereis regName` on node2 right away. In case the name is still
1592− -- erroneously registered, observing the 20 second delay (or lack of), could at
1593− -- least give a hint that something is wrong, and we give up our time slice
1594− -- so that there's a higher change the registrations have been cleaned up
1595− -- in either case.
1596− runProcess node2 $ delayUntilMaybeUnregistered nid pid
1597−
1598− regHere <- newEmptyMVar
1599− runProcess node2 $ whereis regName >>= liftIO . putMVar regHere
1600− res <- takeMVar regHere
1601− case res of
1602− Nothing -> return ()
1603− Just pid -> assertBool ("expected Nothing, but got " ++ show pid) False
1604−
1605− where
1606− runUntilRegistered nid us = do
1607− whereisRemoteAsync nid regName
1608− receiveWait [
1609− matchIf (\(WhereIsReply n (Just p)) -> n == regName && p == us)
1610− (const $ return ())
1611− ]
1612−
1613− delayUntilMaybeUnregistered nid p = do
1614− whereisRemoteAsync nid regName
1615− res <- receiveTimeout 20000000 {- 20 sec delay -} [
1616− matchIf (\(WhereIsReply n p) -> n == regName && isNothing p)
1617− (const $ return ())
1618− ]
1619− case res of
1620− Just () -> return ()
1621− Nothing -> delayUntilMaybeUnregistered nid p
1622−
1623− regName = "testRegisterRemote"
1624−
1625testUnsafeSend :: TestTransport -> Assertion
1372testUnsafeSend :: TestTransport -> Assertion
1626testUnsafeSend TestTransport{..} = do
1373testUnsafeSend TestTransport{..} = do
1627 serverAddr <- newEmptyMVar
1374 serverAddr <- newEmptyMVar
@@ -1637,7 +1384,8 @@ testUnsafeSend TestTransport{..} = do
1637 void $ forkProcess localNode $ do
1384 void $ forkProcess localNode $ do
1638 serverPid <- liftIO $ takeMVar serverAddr
1385 serverPid <- liftIO $ takeMVar serverAddr
1639 getSelfPid >>= unsafeSend serverPid
1386 getSelfPid >>= unsafeSend serverPid
1387+ () <- expect
1640− expect >>= liftIO . putMVar clientDone
1388+ liftIO $ putMVar clientDone ()
1641
1389
1642 takeMVar clientDone
1390 takeMVar clientDone
1643
1391
@@ -1656,7 +1404,8 @@ testUnsafeUSend TestTransport{..} = do
1656 void $ forkProcess localNode $ do
1404 void $ forkProcess localNode $ do
1657 serverPid <- liftIO $ takeMVar serverAddr
1405 serverPid <- liftIO $ takeMVar serverAddr
1658 getSelfPid >>= unsafeUSend serverPid
1406 getSelfPid >>= unsafeUSend serverPid
1407+ () <- expect
1659− expect >>= liftIO . putMVar clientDone
1408+ liftIO $ putMVar clientDone ()
1660
1409
1661 takeMVar clientDone
1410 takeMVar clientDone
1662
1411
@@ -1667,7 +1416,8 @@ testUnsafeNSend TestTransport{..} = do
1667 localNode <- newLocalNode testTransport initRemoteTable
1416 localNode <- newLocalNode testTransport initRemoteTable
1668
1417
1669 pid <- forkProcess localNode $ do
1418 pid <- forkProcess localNode $ do
1419+ () <- expect
1670− expect >>= liftIO . putMVar clientDone
1420+ liftIO $ putMVar clientDone ()
1671
1421
1672 void $ runProcess localNode $ do
1422 void $ runProcess localNode $ do
1673 register "foobar" pid
1423 register "foobar" pid
@@ -1685,7 +1435,8 @@ testUnsafeNSendRemote TestTransport{..} = do
1685 _ <- forkProcess localNode1 $ do
1435 _ <- forkProcess localNode1 $ do
1686 getSelfPid >>= register "foobar"
1436 getSelfPid >>= register "foobar"
1687 liftIO $ putMVar clientDone ()
1437 liftIO $ putMVar clientDone ()
1438+ () <- expect
1688− expect >>= liftIO . putMVar clientDone
1439+ liftIO $ putMVar clientDone ()
1689
1440
1690 takeMVar clientDone
1441 takeMVar clientDone
1691 void $ runProcess localNode2 $ do
1442 void $ runProcess localNode2 $ do
@@ -1709,7 +1460,7 @@ testUnsafeSendChan TestTransport{..} = do
1709 serverPid <- liftIO $ takeMVar serverAddr
1460 serverPid <- liftIO $ takeMVar serverAddr
1710 (sp, rp) <- newChan
1461 (sp, rp) <- newChan
1711 unsafeSend serverPid sp
1462 unsafeSend serverPid sp
1712− receiveChan rp :: Process ()
1463+ () <- receiveChan rp
1713 liftIO $ putMVar clientDone ()
1464 liftIO $ putMVar clientDone ()
1714
1465
1715 takeMVar clientDone
1466 takeMVar clientDone
@@ -1723,8 +1474,8 @@ testCallLocal TestTransport{..} = do
1723 runProcess node $ do
1474 runProcess node $ do
1724 r <- (/=) <$> getSelfPid <*> callLocal getSelfPid
1475 r <- (/=) <$> getSelfPid <*> callLocal getSelfPid
1725 liftIO $ putMVar result r
1476 liftIO $ putMVar result r
1726−
1727− takeMVar result >>= assertBool "Expected 'True'"
1477+ True <- takeMVar result
1478+ return ()
1728
1479
1729 -- Testing that when callLocal is interrupted, the worker is interrupted.
1480 -- Testing that when callLocal is interrupted, the worker is interrupted.
1730 ibox <- newIORef False
1481 ibox <- newIORef False
@@ -1741,9 +1492,10 @@ testCallLocal TestTransport{..} = do
1741 (send keeper ())
1492 (send keeper ())
1742 caller <- expect
1493 caller <- expect
1743 exit caller "test"
1494 exit caller "test"
1744− expect :: Process ()
1495+ () <- expect
1745−
1496+ return ()
1746− readIORef ibox >>= assertBool "Expected 'True'"
1497+ True <- readIORef ibox
1498+ return ()
1747
1499
1748 -- Testing that when the worker raises an exception, the exception is propagated to the parent.
1500 -- Testing that when the worker raises an exception, the exception is propagated to the parent.
1749 ibox2 <- newIORef False
1501 ibox2 <- newIORef False
@@ -1752,8 +1504,8 @@ testCallLocal TestTransport{..} = do
1752 liftIO $ writeIORef ibox2 $ case r of
1504 liftIO $ writeIORef ibox2 $ case r of
1753 Left (ErrorCall "e") -> True
1505 Left (ErrorCall "e") -> True
1754 _ -> False
1506 _ -> False
1755−
1756− readIORef ibox >>= assertBool "Expected 'True'"
1507+ True <- readIORef ibox
1508+ return ()
1757
1509
1758 -- Test that caller waits for the worker in correct situation
1510 -- Test that caller waits for the worker in correct situation
1759 ibox3 <- newIORef False
1511 ibox3 <- newIORef False
@@ -1770,8 +1522,8 @@ testCallLocal TestTransport{..} = do
1770 liftIO $ putMVar result3 =<< readIORef ibox3
1522 liftIO $ putMVar result3 =<< readIORef ibox3
1771 worker <- expect
1523 worker <- expect
1772 send worker ()
1524 send worker ()
1773−
1774− takeMVar result3 >>= assertBool "Expected 'True'"
1525+ True <- takeMVar result3
1526+ return ()
1775
1527
1776 -- Test that caller waits for the worker in case when caller gets an exception
1528 -- Test that caller waits for the worker in case when caller gets an exception
1777 ibox4 <- newIORef False
1529 ibox4 <- newIORef False
@@ -1787,10 +1539,14 @@ testCallLocal TestTransport{..} = do
1787 `finally` (liftIO $ putMVar result4 =<< readIORef ibox4)
1539 `finally` (liftIO $ putMVar result4 =<< readIORef ibox4)
1788 caller <- expect
1540 caller <- expect
1789 exit caller "hi!"
1541 exit caller "hi!"
1790−
1791− takeMVar result4 >>= assertBool "Expected 'True'"
1542+ True <- takeMVar result4
1543+ return ()
1792 -- XXX: Testing that when mask_ $ callLocal p runs p in masked state.
1544 -- XXX: Testing that when mask_ $ callLocal p runs p in masked state.
1793
1545
1546+
1547+
1548+
1549+
1794tests :: TestTransport -> IO [Test]
1550tests :: TestTransport -> IO [Test]
1795tests testtrans = return [
1551tests testtrans = return [
1796 testGroup "Basic features" [
1552 testGroup "Basic features" [
@@ -1814,7 +1570,6 @@ tests testtrans = return [
1814 , testCase "MatchMessageUnwrap" (testMatchMessageWithUnwrap testtrans)
1570 , testCase "MatchMessageUnwrap" (testMatchMessageWithUnwrap testtrans)
1815 , testCase "ReceiveChanTimeout" (testReceiveChanTimeout testtrans)
1571 , testCase "ReceiveChanTimeout" (testReceiveChanTimeout testtrans)
1816 , testCase "ReceiveChanFeatures" (testReceiveChanFeatures testtrans)
1572 , testCase "ReceiveChanFeatures" (testReceiveChanFeatures testtrans)
1817− , testCase "ChanLifecycle" (testChanLifecycle testtrans)
1818 , testCase "KillLocal" (testKillLocal testtrans)
1573 , testCase "KillLocal" (testKillLocal testtrans)
1819 , testCase "KillRemote" (testKillRemote testtrans)
1574 , testCase "KillRemote" (testKillRemote testtrans)
1820 , testCase "Die" (testDie testtrans)
1575 , testCase "Die" (testDie testtrans)
@@ -1824,7 +1579,6 @@ tests testtrans = return [
1824 , testCase "MaskRestoreScope" (testMaskRestoreScope testtrans)
1579 , testCase "MaskRestoreScope" (testMaskRestoreScope testtrans)
1825 , testCase "ExitLocal" (testExitLocal testtrans)
1580 , testCase "ExitLocal" (testExitLocal testtrans)
1826 , testCase "ExitRemote" (testExitRemote testtrans)
1581 , testCase "ExitRemote" (testExitRemote testtrans)
1827− , testCase "RegistryMonitoring" (testRegistryMonitoring testtrans)
1828 , testCase "TextCallLocal" (testCallLocal testtrans)
1582 , testCase "TextCallLocal" (testCallLocal testtrans)
1829 -- Unsafe Primitives
1583 -- Unsafe Primitives
1830 , testCase "TestUnsafeSend" (testUnsafeSend testtrans)
1584 , testCase "TestUnsafeSend" (testUnsafeSend testtrans)
@@ -1877,3 +1631,17 @@ tests testtrans = return [
1877 , testCase "MonitorUnreachable" (testMonitorUnreachable testtrans True False)
1631 , testCase "MonitorUnreachable" (testMonitorUnreachable testtrans True False)
1878 ]
1632 ]
1879 ]
1633 ]
1634+
1635+syncBreakConnection :: (NT.EndPointAddress -> NT.EndPointAddress -> IO ()) -> LocalNode -> LocalNode -> IO ()
1636+syncBreakConnection breakConnection nid0 nid1 = do
1637+ m <- newEmptyMVar
1638+ _ <- forkProcess nid1 $ getSelfPid >>= liftIO . putMVar m
1639+ runProcess nid0 $ do
1640+ them <- liftIO $ takeMVar m
1641+ pinger <- spawnLocal $ forever $ send them ()
1642+ _ <- monitorNode (localNodeId nid1)
1643+ liftIO $ breakConnection (nodeAddress $ localNodeId nid0)
1644+ (nodeAddress $ localNodeId nid1)
1645+ NodeMonitorNotification _ _ _ <- expect
1646+ kill pinger "finished"
1647+ return ()
distributed-process-tests/src/Control/Distributed/Process/Tests/Closure.hs+45−84
@@ -169,11 +169,8 @@ testUnclosure TestTransport{..} rtable = do
169 node <- newLocalNode testTransport rtable
169 node <- newLocalNode testTransport rtable
170 done <- newEmptyMVar
170 done <- newEmptyMVar
171 forkProcess node $ do
171 forkProcess node $ do
172− i <- join . unClosure $ factorialClosure 5
172+ 120 <- join . unClosure $ factorialClosure 5
173 liftIO $ putMVar done ()
173 liftIO $ putMVar done ()
174− if i == 720
175− then return ()
176− else error "Something went horribly wrong"
177 takeMVar done
174 takeMVar done
178
175
179testBind :: TestTransport -> RemoteTable -> Assertion
176testBind :: TestTransport -> RemoteTable -> Assertion
@@ -183,11 +180,8 @@ testBind TestTransport{..} rtable = do
183 runProcess node $ do
180 runProcess node $ do
184 us <- getSelfPid
181 us <- getSelfPid
185 join . unClosure $ sendFac 6 us
182 join . unClosure $ sendFac 6 us
186− (i :: Int) <- expect
183+ (720 :: Int) <- expect
187 liftIO $ putMVar done ()
184 liftIO $ putMVar done ()
188− if i == 720
189− then return ()
190− else error "Something went horribly wrong"
191 takeMVar done
185 takeMVar done
192
186
193testSendPureClosure :: TestTransport -> RemoteTable -> Assertion
187testSendPureClosure :: TestTransport -> RemoteTable -> Assertion
@@ -200,7 +194,7 @@ testSendPureClosure TestTransport{..} rtable = do
200 addr <- forkProcess node $ do
194 addr <- forkProcess node $ do
201 cl <- expect
195 cl <- expect
202 fn <- unClosure cl :: Process (Int -> Int)
196 fn <- unClosure cl :: Process (Int -> Int)
203− (_ :: Int) <- return $ fn 6
197+ 13 <- return $ fn 6
204 liftIO $ putMVar serverDone ()
198 liftIO $ putMVar serverDone ()
205 putMVar serverAddr addr
199 putMVar serverAddr addr
206
200
@@ -224,11 +218,8 @@ testSendIOClosure TestTransport{..} rtable = do
224 liftIO $ do
218 liftIO $ do
225 someMVar <- newEmptyMVar
219 someMVar <- newEmptyMVar
226 io someMVar
220 io someMVar
227− i <- readMVar someMVar
221+ 5 <- readMVar someMVar
228 putMVar serverDone ()
222 putMVar serverDone ()
229− if i == 5
230− then return ()
231− else error "Something went horribly wrong"
232 putMVar serverAddr addr
223 putMVar serverAddr addr
233
224
234 forkIO $ do
225 forkIO $ do
@@ -257,10 +248,8 @@ testSendProcClosure TestTransport{..} rtable = do
257 runProcess node $ do
248 runProcess node $ do
258 pid <- getSelfPid
249 pid <- getSelfPid
259 send theirAddr (cpSend $(mkStatic 'sdictInt) pid)
250 send theirAddr (cpSend $(mkStatic 'sdictInt) pid)
260− i <- expect :: Process Int
251+ 5 <- expect :: Process Int
261− if i == 5
262− then liftIO $ putMVar clientDone ()
252+ liftIO $ putMVar clientDone ()
263− else error "Something went horribly wrong"
264
253
265 takeMVar clientDone
254 takeMVar clientDone
266
255
@@ -280,9 +269,8 @@ testSpawn TestTransport{..} rtable = do
280 pid <- getSelfPid
269 pid <- getSelfPid
281 pid' <- spawn nid (sendPidClosure pid)
270 pid' <- spawn nid (sendPidClosure pid)
282 pid'' <- expect
271 pid'' <- expect
283− if pid' == pid''
272+ True <- return $ pid' == pid''
284− then liftIO $ putMVar clientDone ()
273+ liftIO $ putMVar clientDone ()
285− else error "Something went horribly wrong"
286
274
287 takeMVar clientDone
275 takeMVar clientDone
288
276
@@ -306,9 +294,8 @@ testSpawnRace TestTransport{..} rtable = do
306 spawnLocal $ spawn (localNodeId node2) (sendPidClosure pid) >>= send pid
294 spawnLocal $ spawn (localNodeId node2) (sendPidClosure pid) >>= send pid
307 pid' <- expect :: Process ProcessId
295 pid' <- expect :: Process ProcessId
308 pid'' <- expect :: Process ProcessId
296 pid'' <- expect :: Process ProcessId
309− if pid' == pid''
297+ True <- return $ pid' == pid''
310− then return ()
298+ return ()
311− else error "Something went horribly wrong"
312
299
313 where
300 where
314
301
@@ -345,10 +332,8 @@ testCall TestTransport{..} rtable = do
345 node <- newLocalNode testTransport rtable
332 node <- newLocalNode testTransport rtable
346 nid <- readMVar serverNodeAddr
333 nid <- readMVar serverNodeAddr
347 runProcess node $ do
334 runProcess node $ do
348− (a :: Int) <- call $(mkStatic 'sdictInt) nid (factorialClosure 5)
335+ (120 :: Int) <- call $(mkStatic 'sdictInt) nid (factorialClosure 5)
349− if a == 120
350− then liftIO $ putMVar clientDone ()
336+ liftIO $ putMVar clientDone ()
351− else error "something went horribly wrong"
352
337
353 takeMVar clientDone
338 takeMVar clientDone
354
339
@@ -365,10 +350,8 @@ testCallBind TestTransport{..} rtable = do
365 node <- newLocalNode testTransport rtable
350 node <- newLocalNode testTransport rtable
366 nid <- readMVar serverNodeAddr
351 nid <- readMVar serverNodeAddr
367 runProcess node $ do
352 runProcess node $ do
368− (a :: Int) <- call $(mkStatic 'sdictInt) nid (factorial' 5)
353+ (120 :: Int) <- call $(mkStatic 'sdictInt) nid (factorial' 5)
369− if a == 120
370− then liftIO $ putMVar clientDone ()
354+ liftIO $ putMVar clientDone ()
371− else error "Something went horribly wrong"
372
355
373 takeMVar clientDone
356 takeMVar clientDone
374
357
@@ -379,11 +362,9 @@ testSeq TestTransport{..} rtable = do
379 runProcess node $ do
362 runProcess node $ do
380 us <- getSelfPid
363 us <- getSelfPid
381 join . unClosure $ sendFac 5 us `seqCP` sendFac 6 us
364 join . unClosure $ sendFac 5 us `seqCP` sendFac 6 us
382− a :: Int <- expect
365+ 120 :: Int <- expect
383− b :: Int <- expect
366+ 720 :: Int <- expect
384− if a == 120 && b == 720
385− then liftIO $ putMVar done ()
367+ liftIO $ putMVar done ()
386− else error "Something went horribly wrong"
387 takeMVar done
368 takeMVar done
388
369
389-- Test 'spawnSupervised'
370-- Test 'spawnSupervised'
@@ -412,33 +393,25 @@ testSpawnSupervised TestTransport{..} rtable = do
412 throw supervisorDeath
393 throw supervisorDeath
413
394
414 forkProcess node2 $ do
395 forkProcess node2 $ do
415− res <- liftIO $ mapM readMVar [superPid, childPid]
396+ [super, child] <- liftIO $ mapM readMVar [superPid, childPid]
416− case res of
417− [super, child] -> do
418− ref <- monitor child
397+ ref <- monitor child
419− self <- getSelfPid
398+ self <- getSelfPid
420− let waitForMOrL = do
399+ let waitForMOrL = do
421− liftIO $ threadDelay 10000
400+ liftIO $ threadDelay 10000
422− mpinfo <- getProcessInfo child
401+ mpinfo <- getProcessInfo child
423− case mpinfo of
402+ case mpinfo of
424− Nothing -> waitForMOrL
403+ Nothing -> waitForMOrL
425− Just pinfo ->
404+ Just pinfo ->
426− unless (isJust $ lookup self (infoMonitors pinfo)) waitForMOrL
405+ unless (isJust $ lookup self (infoMonitors pinfo)) waitForMOrL
427− waitForMOrL
406+ waitForMOrL
428− liftIO $ putMVar linkUp ()
407+ liftIO $ putMVar linkUp ()
429− -- because monitor message was sent before message to process
408+ -- because monitor message was sent before message to process
430− -- we hope that it will be processed before
409+ -- we hope that it will be processed before
431− res' <- expect
410+ ProcessMonitorNotification ref' pid' (DiedException e) <- expect
432− case res' of
411+ True <- return $ ref' == ref
433− (ProcessMonitorNotification ref' pid' (DiedException e)) ->
434− if (ref' == ref && pid' == child &&
412+ && pid' == child
435− e == show (ProcessLinkException super
436− (DiedException (show supervisorDeath))))
413+ && e == show (ProcessLinkException super (DiedException (show supervisorDeath)))
437− then liftIO $ putMVar thirdProcessDone ()
414+ liftIO $ putMVar thirdProcessDone ()
438− else error "Something went horribly wrong"
439− _ -> error "Something went horribly wrong"
440−
441− _ -> die $ "Something went horribly wrong"
442
415
443 takeMVar thirdProcessDone
416 takeMVar thirdProcessDone
444 where
417 where
@@ -453,10 +426,8 @@ testSpawnInvalid TestTransport{..} rtable = do
453 (pid, ref) <- spawnMonitor (localNodeId node) (closure (staticLabel "ThisDoesNotExist") empty)
426 (pid, ref) <- spawnMonitor (localNodeId node) (closure (staticLabel "ThisDoesNotExist") empty)
454 ProcessMonitorNotification ref' pid' _reason <- expect
427 ProcessMonitorNotification ref' pid' _reason <- expect
455 -- Depending on the exact interleaving, reason might be NoProc or the exception thrown by the absence of the static closure
428 -- Depending on the exact interleaving, reason might be NoProc or the exception thrown by the absence of the static closure
456− res <- return $ ref' == ref && pid == pid'
429+ True <- return $ ref' == ref && pid == pid'
457− if res == True
458− then liftIO $ putMVar done ()
430+ liftIO $ putMVar done ()
459− else error "Something went horribly wrong"
460 takeMVar done
431 takeMVar done
461
432
462testClosureExpect :: TestTransport -> RemoteTable -> Assertion
433testClosureExpect :: TestTransport -> RemoteTable -> Assertion
@@ -468,10 +439,8 @@ testClosureExpect TestTransport{..} rtable = do
468 us <- getSelfPid
439 us <- getSelfPid
469 them <- spawn nodeId $ cpExpect $(mkStatic 'sdictInt) `bindCP` cpSend $(mkStatic 'sdictInt) us
440 them <- spawn nodeId $ cpExpect $(mkStatic 'sdictInt) `bindCP` cpSend $(mkStatic 'sdictInt) us
470 send them (1234 :: Int)
441 send them (1234 :: Int)
471− (res :: Int) <- expect
442+ (1234 :: Int) <- expect
472− if res == 1234
473− then liftIO $ putMVar done ()
443+ liftIO $ putMVar done ()
474− else error "Something went horribly wrong"
475 takeMVar done
444 takeMVar done
476
445
477testSpawnChannel :: TestTransport -> RemoteTable -> Assertion
446testSpawnChannel :: TestTransport -> RemoteTable -> Assertion
@@ -496,10 +465,8 @@ testTDict TestTransport{..} rtable = do
496 done <- newEmptyMVar
465 done <- newEmptyMVar
497 [node1, node2] <- replicateM 2 $ newLocalNode testTransport rtable
466 [node1, node2] <- replicateM 2 $ newLocalNode testTransport rtable
498 forkProcess node1 $ do
467 forkProcess node1 $ do
499− res <- call $(functionTDict 'isPrime) (localNodeId node2) ($(mkClosure 'isPrime) (79 :: Integer))
468+ True <- call $(functionTDict 'isPrime) (localNodeId node2) ($(mkClosure 'isPrime) (79 :: Integer))
500− if res == True
501− then liftIO $ putMVar done ()
469+ liftIO $ putMVar done ()
502− else error "Something went horribly wrong..."
503 takeMVar done
470 takeMVar done
504
471
505testFib :: TestTransport -> RemoteTable -> Assertion
472testFib :: TestTransport -> RemoteTable -> Assertion
@@ -510,11 +477,8 @@ testFib TestTransport{..} rtable = do
510 forkProcess (head nodes) $ do
477 forkProcess (head nodes) $ do
511 (sport, rport) <- newChan
478 (sport, rport) <- newChan
512 spawnLocal $ dfib (map localNodeId nodes, sport, 10)
479 spawnLocal $ dfib (map localNodeId nodes, sport, 10)
513− ff <- receiveChan rport :: Process Integer
480+ 55 <- receiveChan rport :: Process Integer
514 liftIO $ putMVar done ()
481 liftIO $ putMVar done ()
515− if ff /= 55
516− then die $ "Something went horribly wrong"
517− else return ()
518
482
519 takeMVar done
483 takeMVar done
520
484
@@ -522,7 +486,7 @@ testSpawnReconnect :: TestTransport -> RemoteTable -> Assertion
522testSpawnReconnect testtrans@TestTransport{..} rtable = do
486testSpawnReconnect testtrans@TestTransport{..} rtable = do
523 [node1, node2] <- replicateM 2 $ newLocalNode testTransport rtable
487 [node1, node2] <- replicateM 2 $ newLocalNode testTransport rtable
524 let nid1 = localNodeId node1
488 let nid1 = localNodeId node1
525− -- nid2 = localNodeId node2
489+ nid2 = localNodeId node2
526 done <- newEmptyMVar
490 done <- newEmptyMVar
527 iv <- newMVar (0 :: Int)
491 iv <- newMVar (0 :: Int)
528
492
@@ -539,12 +503,9 @@ testSpawnReconnect testtrans@TestTransport{..} rtable = do
539 liftIO $ threadDelay 100000
503 liftIO $ threadDelay 100000
540
504
541 count <- liftIO $ takeMVar iv
505 count <- liftIO $ takeMVar iv
542− res <- return $ count == 2 || count == 3 -- It depends on which message we get first in 'spawn'
506+ True <- return $ count == 2 || count == 3 -- It depends on which message we get first in 'spawn'
543
507
544 liftIO $ putMVar done ()
508 liftIO $ putMVar done ()
545− if res /= True
546− then error "Something went horribly wrong"
547− else return ()
548
509
549 takeMVar done
510 takeMVar done
550
511
distributed-process-tests/src/Control/Distributed/Process/Tests/Internal/Utils.hs+5−16
@@ -20,7 +20,6 @@ module Control.Distributed.Process.Tests.Internal.Utils
20 -- ping !
20 -- ping !
21 , Ping(Ping)
21 , Ping(Ping)
22 , ping
22 , ping
23− , pause
24 , shouldBe
23 , shouldBe
25 , shouldMatch
24 , shouldMatch
26 , shouldContain
25 , shouldContain
@@ -74,16 +73,15 @@ import Control.Concurrent
74import Control.Concurrent.MVar
73import Control.Concurrent.MVar
75 ( putMVar
74 ( putMVar
76 )
75 )
77−import Control.Distributed.Process hiding (finally, catch)
76+import Control.Distributed.Process
78import Control.Distributed.Process.Node
77import Control.Distributed.Process.Node
79import Control.Distributed.Process.Serializable()
78import Control.Distributed.Process.Serializable()
80
79
81import Control.Exception (AsyncException(ThreadKilled), SomeException)
80import Control.Exception (AsyncException(ThreadKilled), SomeException)
82−import Control.Monad (forever, void)
81+import Control.Monad (forever)
83−import Control.Monad.Catch (finally, catch)
84import Control.Monad.STM (atomically)
82import Control.Monad.STM (atomically)
85import Control.Rematch hiding (match)
83import Control.Rematch hiding (match)
86−import Control.Rematch.Run
84+import Control.Rematch.Run
87import Data.Binary
85import Data.Binary
88import Data.Typeable (Typeable)
86import Data.Typeable (Typeable)
89
87
@@ -91,7 +89,6 @@ import Test.HUnit (Assertion, assertFailure)
91import Test.HUnit.Base (assertBool)
89import Test.HUnit.Base (assertBool)
92
90
93import GHC.Generics
91import GHC.Generics
94−import System.Timeout (timeout)
95
92
96-- | A mutable cell containing a test result.
93-- | A mutable cell containing a test result.
97type TestResult a = MVar a
94type TestResult a = MVar a
@@ -110,18 +107,10 @@ data TestProcessControl = Stop | Go | Report ProcessId
110
107
111instance Binary TestProcessControl where
108instance Binary TestProcessControl where
112
109
113−data Private = Private
114− deriving (Typeable, Generic)
115−instance Binary Private where
116−
117-- | Does exactly what it says on the tin, doing so in the @Process@ monad.
110-- | Does exactly what it says on the tin, doing so in the @Process@ monad.
118noop :: Process ()
111noop :: Process ()
119noop = return ()
112noop = return ()
120
113
121−pause :: Int -> Process ()
122−pause delay =
123− void $ receiveTimeout delay [ match (\Private -> return ()) ]
124−
125synchronisedAssertion :: Eq a
114synchronisedAssertion :: Eq a
126 => String
115 => String
127 -> LocalNode
116 -> LocalNode
@@ -234,10 +223,10 @@ testProcessReport pid = do
234tryRunProcess :: LocalNode -> Process () -> IO ()
223tryRunProcess :: LocalNode -> Process () -> IO ()
235tryRunProcess node p = do
224tryRunProcess node p = do
236 tid <- liftIO myThreadId
225 tid <- liftIO myThreadId
237− void $ timeout (1000000 * 60 * 5 :: Int) $
238− runProcess node $ catch p (\e -> liftIO $ throwTo tid (e::SomeException))
226+ runProcess node $ catch p (\e -> liftIO $ throwTo tid (e::SomeException))
239
227
240tryForkProcess :: LocalNode -> Process () -> IO ProcessId
228tryForkProcess :: LocalNode -> Process () -> IO ProcessId
241tryForkProcess node p = do
229tryForkProcess node p = do
242 tid <- liftIO myThreadId
230 tid <- liftIO myThreadId
243 forkProcess node $ catch p (\e -> liftIO $ throwTo tid (e::SomeException))
231 forkProcess node $ catch p (\e -> liftIO $ throwTo tid (e::SomeException))
232+
distributed-process-tests/src/Control/Distributed/Process/Tests/Mx.hs+82−352
@@ -1,22 +1,17 @@
1−{-# LANGUAGE DeriveGeneric #-}
1+{-# LANGUAGE DeriveGeneric #-}
2−{-# LANGUAGE ParallelListComp #-}
3−
4module Control.Distributed.Process.Tests.Mx (tests) where
2module Control.Distributed.Process.Tests.Mx (tests) where
5
3
6import Control.Distributed.Process.Tests.Internal.Utils
4import Control.Distributed.Process.Tests.Internal.Utils
7import Network.Transport.Test (TestTransport(..))
5import Network.Transport.Test (TestTransport(..))
8−import Control.Distributed.Process hiding (bracket, finally, try)
6+
7+import Control.Concurrent (threadDelay)
9−import Control.Distributed.Process.Internal.Types
8+import Control.Distributed.Process
10− ( ProcessExitException(..)
11− , unsafeCreateUnencodedMessage
12− )
13import Control.Distributed.Process.Node
9import Control.Distributed.Process.Node
14import qualified Control.Distributed.Process.UnsafePrimitives as Unsafe
10import qualified Control.Distributed.Process.UnsafePrimitives as Unsafe
15 ( send
11 ( send
16 , nsend
12 , nsend
17 , nsendRemote
13 , nsendRemote
18 , usend
14 , usend
19− , sendChan
20 )
15 )
21import Control.Distributed.Process.Management
16import Control.Distributed.Process.Management
22 ( MxEvent(..)
17 ( MxEvent(..)
@@ -32,15 +27,14 @@ import Control.Distributed.Process.Management
32 , mxUpdateLocal
27 , mxUpdateLocal
33 , mxNotify
28 , mxNotify
34 , mxBroadcast
29 , mxBroadcast
30+ , mxGetId
35 )
31 )
36−import Control.Monad (void, unless)
32+import Control.Monad (void)
37−import Control.Monad.Catch(finally, bracket, try)
38−import Control.Rematch (equalTo)
39import Data.Binary
33import Data.Binary
40−import Data.List (find, sort, intercalate)
34+import Data.List (find, sort)
41−import Data.Maybe (isJust, fromJust, isNothing, fromMaybe, catMaybes)
35+import Data.Maybe (isJust)
42import Data.Typeable
36import Data.Typeable
43−import GHC.Generics hiding (from)
37+import GHC.Generics
44#if ! MIN_VERSION_base(4,6,0)
38#if ! MIN_VERSION_base(4,6,0)
45import Prelude hiding (catch, log)
39import Prelude hiding (catch, log)
46#endif
40#endif
@@ -56,17 +50,7 @@ data Publish = Publish
56
50
57instance Binary Publish where
51instance Binary Publish where
58
52
59−awaitExit :: ProcessId -> Process ()
60−awaitExit pid =
61− withMonitorRef pid $ \ref -> do
62− receiveWait
63− [ matchIf (\(ProcessMonitorNotification r _ _) -> r == ref)
64− (\_ -> return ())
65− ]
66− where
67− withMonitorRef p = bracket (monitor p) unmonitor
68−
69−testAgentBroadcast :: TestResult (Maybe ()) -> Process ()
53+testAgentBroadcast :: TestResult () -> Process ()
70testAgentBroadcast result = do
54testAgentBroadcast result = do
71 (resultSP, resultRP) <- newChan :: Process (SendPort (), ReceivePort ())
55 (resultSP, resultRP) <- newChan :: Process (SendPort (), ReceivePort ())
72
56
@@ -81,7 +65,7 @@ testAgentBroadcast result = do
81 mxNotify ()
65 mxNotify ()
82 -- Once the publisher has seen our message, it will broadcast the Publish
66 -- Once the publisher has seen our message, it will broadcast the Publish
83 -- and the consumer will see that and send the result to our typed channel.
67 -- and the consumer will see that and send the result to our typed channel.
84− stash result =<< receiveChanTimeout 10000000 resultRP
68+ stash result =<< receiveChan resultRP
85
69
86 kill publisher "finished"
70 kill publisher "finished"
87 kill consumer "finished"
71 kill consumer "finished"
@@ -89,7 +73,7 @@ testAgentBroadcast result = do
89testAgentDualInput :: TestResult (Maybe Int) -> Process ()
73testAgentDualInput :: TestResult (Maybe Int) -> Process ()
90testAgentDualInput result = do
74testAgentDualInput result = do
91 (sp, rp) <- newChan
75 (sp, rp) <- newChan
92− s <- mxAgent (MxAgentId "sum-agent") (0 :: Int) [
76+ _ <- mxAgent (MxAgentId "sum-agent") (0 :: Int) [
93 mxSink $ (\(i :: Int) -> do
77 mxSink $ (\(i :: Int) -> do
94 mxSetLocal . (+i) =<< mxGetLocal
78 mxSetLocal . (+i) =<< mxGetLocal
95 i' <- mxGetLocal
79 i' <- mxGetLocal
@@ -99,8 +83,6 @@ testAgentDualInput result = do
99 else mxReady)
83 else mxReady)
100 ]
84 ]
101
85
102− mRef <- monitor s
103−
104 mxNotify (1 :: Int)
86 mxNotify (1 :: Int)
105 nsend "sum-agent" (3 :: Int)
87 nsend "sum-agent" (3 :: Int)
106 mxNotify (2 :: Int)
88 mxNotify (2 :: Int)
@@ -108,10 +90,6 @@ testAgentDualInput result = do
108 mxNotify (5 :: Int)
90 mxNotify (5 :: Int)
109
91
110 stash result =<< receiveChanTimeout 10000000 rp
92 stash result =<< receiveChanTimeout 10000000 rp
111− died <- receiveTimeout 10000000 [
112− matchIf (\(ProcessMonitorNotification r _ _) -> r == mRef) (const $ return True)
113− ]
114− died `shouldBe` equalTo (Just True)
115
93
116testAgentPrioritisation :: TestResult [String] -> Process ()
94testAgentPrioritisation :: TestResult [String] -> Process ()
117testAgentPrioritisation result = do
95testAgentPrioritisation result = do
@@ -123,12 +101,12 @@ testAgentPrioritisation result = do
123
101
124 let name = "prioritising-agent"
102 let name = "prioritising-agent"
125 (sp, rp) <- newChan
103 (sp, rp) <- newChan
126− s <- mxAgent (MxAgentId name) ["first"] [
104+ void $ mxAgent (MxAgentId name) ["first"] [
127− mxSink (\(s :: String) -> do
105+ mxSink (\(s :: String) -> do
128− mxUpdateLocal (s:)
106+ mxUpdateLocal ((s:))
129 st <- mxGetLocal
107 st <- mxGetLocal
130 case length st of
108 case length st of
131− n | n == 5 -> do liftMX $ sendChan sp (sort st)
109+ n | n == 5 -> do liftMX $ sendChan sp st
132 mxDeactivate "finished"
110 mxDeactivate "finished"
133 _ -> mxReceive -- go to the mailbox
111 _ -> mxReceive -- go to the mailbox
134 )
112 )
@@ -139,33 +117,23 @@ testAgentPrioritisation result = do
139 mxNotify "fourth"
117 mxNotify "fourth"
140 nsend name "fifth"
118 nsend name "fifth"
141
119
142− stash result . fromMaybe [] =<< receiveChanTimeout 10000000 rp
120+ stash result . sort =<< receiveChan rp
143− awaitExit s
144
121
145testAgentMailboxHandling :: TestResult (Maybe ()) -> Process ()
122testAgentMailboxHandling :: TestResult (Maybe ()) -> Process ()
146testAgentMailboxHandling result = do
123testAgentMailboxHandling result = do
147 (sp, rp) <- newChan
124 (sp, rp) <- newChan
148− agent <- mxAgent (MxAgentId "mailbox-agent") () [
125+ agent <- mxAgent (MxAgentId "listener-agent") () [
149 mxSink $ \() -> (liftMX $ sendChan sp ()) >> mxReady
126 mxSink $ \() -> (liftMX $ sendChan sp ()) >> mxReady
150 ]
127 ]
151
128
152− nsend "mailbox-agent" ()
129+ nsend "listener-agent" ()
153
130
154 stash result =<< receiveChanTimeout 1000000 rp
131 stash result =<< receiveChanTimeout 1000000 rp
155 kill agent "finished"
132 kill agent "finished"
156
133
157testAgentEventHandling :: TestResult Bool -> Process ()
134testAgentEventHandling :: TestResult Bool -> Process ()
158testAgentEventHandling result = do
135testAgentEventHandling result = do
159− us <- getSelfPid
160− -- because this test is a bit racy, let's ensure it can't run indefinitely
161− timer <- spawnLocal $ do
162− pause (1000000 * 60 * 5)
163− -- okay we've waited 5 mins, let's kill the test off if it's stuck...
164− stash result False
165− kill us "Test Timed Out"
166−
167 let initState = [] :: [MxEvent]
136 let initState = [] :: [MxEvent]
168− (rc, rs) <- newChan
169 agentPid <- mxAgent (MxAgentId "lifecycle-listener-agent") initState [
137 agentPid <- mxAgent (MxAgentId "lifecycle-listener-agent") initState [
170 (mxSink $ \ev -> do
138 (mxSink $ \ev -> do
171 st <- mxGetLocal
139 st <- mxGetLocal
@@ -174,7 +142,7 @@ testAgentEventHandling result = do
174 (MxSpawned _) -> mxSetLocal (ev:st)
142 (MxSpawned _) -> mxSetLocal (ev:st)
175 (MxProcessDied _ _) -> mxSetLocal (ev:st)
143 (MxProcessDied _ _) -> mxSetLocal (ev:st)
176 _ -> return ()
144 _ -> return ()
177− act >> (liftMX $ sendChan rc ()) >> mxReady),
145+ act >> mxReady),
178 (mxSink $ \(ev, sp :: SendPort Bool) -> do
146 (mxSink $ \(ev, sp :: SendPort Bool) -> do
179 st <- mxGetLocal
147 st <- mxGetLocal
180 let found =
148 let found =
@@ -194,349 +162,111 @@ testAgentEventHandling result = do
194 mxReady)
162 mxReady)
195 ]
163 ]
196
164
197− -- TODO: yes, this is racy, but we're at the mercy of the scheduler here...
165+ _ <- monitor agentPid
198− faff 2000000 rs
199−
200− (sp, rp) <- newChan :: Process (SendPort (), ReceivePort ())
166+ (sp, rp) <- newChan
201−
202− pid <- spawnLocal $ expect >>= sendChan sp
167+ pid <- spawnLocal $ sendChan sp ()
168+ () <- receiveChan rp
203
169
204 -- By waiting for a monitor notification, we have a
170 -- By waiting for a monitor notification, we have a
205− -- higher probably that the agent has seen the spawn and died events
171+ -- higher probably that the agent has seen
206 monitor pid
172 monitor pid
207−
208− send pid ()
209− rct <- receiveChanTimeout 10000000 rp
210− unless (isJust rct) $ die "No response on channel"
211−
212− pmn <- receiveTimeout 2000000 [ match (\ProcessMonitorNotification{} -> return ()) ]
173+ receiveWait [ match (\(ProcessMonitorNotification _ _ _) -> return ()) ]
213− unless (isJust pmn) $ die "No monitor notification arrived"
214−
215− -- TODO: yes, this is racy, but we're at the mercy of the scheduler here...
216− faff 2000000 rs
217
174
218 (replyTo, reply) <- newChan :: Process (SendPort Bool, ReceivePort Bool)
175 (replyTo, reply) <- newChan :: Process (SendPort Bool, ReceivePort Bool)
219 mxNotify (MxSpawned pid, replyTo)
176 mxNotify (MxSpawned pid, replyTo)
220 mxNotify (MxProcessDied pid DiedNormal, replyTo)
177 mxNotify (MxProcessDied pid DiedNormal, replyTo)
221
178
222− seenAlive <- receiveChanTimeout 10000000 reply
179+ seenAlive <- receiveChan reply
223− seenDead <- receiveChanTimeout 10000000 reply
180+ seenDead <- receiveChan reply
224−
225− stash result $ foldr (&&) True $ catMaybes [seenAlive, seenDead]
226− kill timer "test-complete"
227− kill agentPid "test-complete"
228− where
229− faff delay port = do
230− res <- receiveChanTimeout delay port
231− unless (isNothing res) $ pause delay
232−
233−testMxRegEvents :: TestResult () -> Process ()
234−testMxRegEvents result = do
235− {- This test only deals with the local case, to ensure that we are being
236− notified in the expected order - the remote cases related to the
237− behaviour of the node controller are contained in the CH test suite. -}
238− ensure (stash result ()) $ do
239− let label = "testMxRegEvents"
240− let agentLabel = "mxRegEvents-agent"
241− let delay = 1000000
242− (regChan, regSink) <- newChan
243− (unRegChan, unRegSink) <- newChan
244− agent <- mxAgent (MxAgentId agentLabel) () [
245− mxSink $ \ev -> do
246− case ev of
247− MxRegistered pid label'
248− | label' == label -> liftMX $ sendChan regChan (label', pid)
249− MxUnRegistered pid label'
250− | label' == label -> liftMX $ sendChan unRegChan (label', pid)
251− _ -> return ()
252− mxReady
253− ]
254
181
255− p1 <- spawnLocal expect
256− p2 <- spawnLocal expect
257−
258− register label p1
259− reg1 <- receiveChanTimeout delay regSink
260− reg1 `shouldBe` equalTo (Just (label, p1))
261−
262− unregister label
263− unreg1 <- receiveChanTimeout delay unRegSink
264− unreg1 `shouldBe` equalTo (Just (label, p1))
265−
266− register label p2
267− reg2 <- receiveChanTimeout delay regSink
268− reg2 `shouldBe` equalTo (Just (label, p2))
269−
270− reregister label p1
271− unreg2 <- receiveChanTimeout delay unRegSink
272− unreg2 `shouldBe` equalTo (Just (label, p2))
273−
274− reg3 <- receiveChanTimeout delay regSink
275− reg3 `shouldBe` equalTo (Just (label, p1))
276−
277− mapM_ (flip kill $ "test-complete") [agent, p1, p2]
278−
279−testMxRegMon :: LocalNode -> TestResult () -> Process ()
280−testMxRegMon remoteNode result = do
281− ensure (stash result ()) $ do
182+ stash result $ seenAlive && seenDead
282− -- ensure that when a registered process dies, we get a notification that
283− -- it has been unregistered as well as seeing the name get removed
284− let label1 = "aaaaa"
285− let label2 = "bbbbb"
286− let isValid l = l == label1 || l == label2
287− let agentLabel = "mxRegMon-agent"
288− let delay = 1000000
289− (regChan, regSink) <- newChan
290− (unRegChan, unRegSink) <- newChan
291− agent <- mxAgent (MxAgentId agentLabel) () [
292− mxSink $ \ev -> do
293− case ev of
294− MxRegistered pid label
295− | isValid label -> liftMX $ sendChan regChan (label, pid)
296− MxUnRegistered pid label
297− | isValid label -> liftMX $ sendChan unRegChan (label, pid)
298− _ -> return ()
299− mxReady
300− ]
301
183
302− (sp, rp) <- newChan
184+
303− liftIO $ forkProcess remoteNode $ do
185+testMxUnsafeSendEvents :: LocalNode -> Process ()
304− getSelfPid >>= sendChan sp
186+testMxUnsafeSendEvents remoteNode = do
305− expect :: Process ()
187+
306−
188+ -- ensure that when a registered process dies, we get a notification that
307− p1 <- receiveChan rp
189+ -- it has been unregistered as well as seeing the name get removed
308−
190+
309− register label1 p1
191+ let label1 = "aaaaa"
310− reg1 <- receiveChanTimeout delay regSink
192+ let label2 = "bbbbb"
311− reg1 `shouldBe` equalTo (Just (label1, p1))
193+ let isValid l = l ==label1 || l == label2
312−
194+ let agentLabel = "listener-agent"
313− register label2 p1
195+ let delay = 1000000
314− reg2 <- receiveChanTimeout delay regSink
315− reg2 `shouldBe` equalTo (Just (label2, p1))
316−
317− n1 <- whereis label1
318− n1 `shouldBe` equalTo (Just p1)
319−
320− n2 <- whereis label2
321− n2 `shouldBe` equalTo (Just p1)
322−
323− kill p1 "goodbye"
324−
325− unreg1 <- receiveChanTimeout delay unRegSink
326− unreg2 <- receiveChanTimeout delay unRegSink
327−
328− let evts = [unreg1, unreg2]
329− -- we can't rely on the order of the values in the node controller's
330− -- map (it's either racy to do so, or no such guarantee exists for Data.Map),
331− -- so we simply verify that we received the un-registration events we expect
332− evts `shouldContain` (Just (label1, p1))
333− evts `shouldContain` (Just (label2, p1))
334−
335− kill agent "test-complete"
336−
337−ensure :: Process () -> Process () -> Process ()
338−ensure = flip finally
339−
340−testNSend :: (String -> () -> Process ())
341− -> Maybe LocalNode
342− -> Process ()
343−testNSend op n = do
344− us <- getSelfPid
345− let delay = 5000000
346− let label = "testMxSend" ++ (show us)
347− let isValid = isValidLabel n label
348−
349− testMxSend n label $ \p1 sink -> do
350− register label p1
351− reg1 <- receiveChanTimeout delay sink
352− case reg1 of
353− Just (MxRegistered pd lb)
354− | pd == p1 && isValid lb -> return ()
355− _ -> die $ "Reg-Failed: " ++ show reg1
356−
357− op label ()
358−
359− sent <- receiveChanTimeout delay sink
360− case sent of
361− Just (MxSentToName lb by _)
362− | by == us && isValid lb -> return True
363− _ -> die $ "Send-Failed: " ++ show sent
364−
365− where
366− isValidLabel nd l1 l2
367− | l2 == l2 = True
368− | isJust nd = l2 == l1 ++ "@" ++ show (localNodeId $ fromJust nd)
369− | otherwise = False
370−
371−testSend :: (ProcessId -> () -> Process ())
372− -> Maybe LocalNode
373− -> Process ()
374−testSend op n = do
375− us <- getSelfPid
376− let label = "testMxSend" ++ (show us)
377− testMxSend n label $ \p1 sink -> do
378− -- initiate a send
379− op p1 ()
380−
381− -- verify the management event
382− sent <- receiveChanTimeout 5000000 sink
383− case sent of
384− Just (MxSent pidTo pidFrom _)
385− | pidTo == p1 && pidFrom == us -> return True
386− _ -> return False
387−
388−testChan :: (SendPort () -> () -> Process ())
389− -> Maybe LocalNode
390− -> Process ()
391−testChan op n = testMxSend n "" $ \p1 sink -> do
392−
393− us <- getSelfPid
394− send p1 us
395−
396− cleared <- receiveChanTimeout 2000000 sink
397− case cleared of
398− Just (MxSent pidTo pidFrom _)
399− | pidTo == p1 && pidFrom == us -> return ()
400− _ -> die "Received uncleared Mx Event"
401−
402− chan <- receiveTimeout 5000000 [ match return ]
403− let ch' = fromMaybe (error "No reply chan received") chan
404−
405− -- initiate a send
406− op ch' ()
407−
408− -- verify the management event
409− sent <- receiveChanTimeout 5000000 sink
410− case sent of
411− Just (MxSentToPort sId spId _)
412− | sId == us && spId == sendPortId ch' -> return True
413− _ -> return False
414−
415−type SendTest = ProcessId -> ReceivePort MxEvent -> Process Bool
416−
417−testMxSend :: Maybe LocalNode -> String -> SendTest -> Process ()
418−testMxSend mNode label test = do
419− us <- getSelfPid
420− (sp, rp) <- newChan
196+ (regChan, regSink) <- newChan
421− (chan, sink) <- newChan
197+ (unRegChan, unRegSink) <- newChan
422− agent <- mxAgent (MxAgentId $ agentLabel us) () [
198+ agent <- mxAgent (MxAgentId agentLabel) () [
423 mxSink $ \ev -> do
199 mxSink $ \ev -> do
424 case ev of
200 case ev of
425− m@(MxSentToPort _ cid _)
426− | cid /= sendPortId chan
427− && cid /= sendPortId sp -> liftMX $ sendChan chan m
428− m@(MxSent _ fromPid _)
429− | fromPid == us -> liftMX $ sendChan chan m
430− m@(MxSentToName _ fromPid _)
431− | fromPid == us -> liftMX $ sendChan chan m
432− m@(MxRegistered _ name)
201+ MxRegistered pid label
433− | name == label -> liftMX $ sendChan chan m
202+ | isValid label -> liftMX $ sendChan regChan (label, pid)
203+ MxUnRegistered pid label
204+ | isValid label -> liftMX $ sendChan unRegChan (label, pid)
434 _ -> return ()
205 _ -> return ()
435 mxReady
206 mxReady
436 ]
207 ]
437
208
438− case mNode of
209+ (sp, rp) <- newChan
439− Nothing -> void $ spawnLocal (proc sp)
440− Just remoteNode -> void $ liftIO $ forkProcess remoteNode $ proc sp
210+ liftIO $ forkProcess remoteNode $ do
211+ getSelfPid >>= sendChan sp
212+ expect :: Process ()
441
213
442− p1 <- receiveChanTimeout 2000000 rp
214+ p1 <- receiveChan rp
443− unless (isJust p1) $ die "Timed out waiting for ProcessId"
444− res <- try $ test (fromJust p1) sink
445
215
446− kill agent "bye"
216+ register label1 p1
447− kill (fromJust p1) "bye"
217+ reg1 <- receiveChanTimeout delay regSink
218+ reg1 `shouldBe` equalTo (Just (label1, p1))
448
219
449− case res of
220+ register label2 p1
450− Left (ProcessExitException _ m) -> (liftIO $ putStrLn $ "SomeException-" ++ show m) >> die m
221+ reg2 <- receiveChanTimeout delay regSink
451− Right tr -> tr `shouldBe` equalTo True
222+ reg2 `shouldBe` equalTo (Just (label2, p1))
452
223
224+ n1 <- whereis label1
225+ n1 `shouldBe` equalTo (Just p1)
453
226
454− where
227+ n2 <- whereis label2
455− agentLabel s = "mx-unsafe-check-agent-" ++ show s
228+ n2 `shouldBe` equalTo (Just p1)
456− proc sp' = getSelfPid >>= sendChan sp' >> go Nothing
457
229
458− go :: Maybe (ReceivePort ()) -> Process ()
230+ kill p1 "goodbye"
459− go Nothing = receiveTimeout 5000000 [ match replyChannel ] >>= go
460− go c@(Just c') = receiveWait [ matchChan c' return ] >> go c
461
231
462− replyChannel p' = do
232+ unreg1 <- receiveChanTimeout delay unRegSink
463− (s, r) <- newChan
233+ unreg2 <- receiveChanTimeout delay unRegSink
464− send p' s
234+
465− return r
235+ sort [unreg1, unreg2]
236+ `shouldBe` equalTo [Just (label1, p1), Just (label2, p1)]
237+
238+ kill agent "test-complete"
466
239
467tests :: TestTransport -> IO [Test]
240tests :: TestTransport -> IO [Test]
468tests TestTransport{..} = do
241tests TestTransport{..} = do
469 node1 <- newLocalNode testTransport initRemoteTable
242 node1 <- newLocalNode testTransport initRemoteTable
470 node2 <- newLocalNode testTransport initRemoteTable
243 node2 <- newLocalNode testTransport initRemoteTable
471 return [
244 return [
472− testGroup "MxAgents" [
245+ testGroup "Mx Agents" [
473− testCase "EventHandling"
246+ testCase "Event Handling"
474 (delayedAssertion
247 (delayedAssertion
475 "expected True, but events where not as expected"
248 "expected True, but events where not as expected"
476 node1 True testAgentEventHandling)
249 node1 True testAgentEventHandling)
477− , testCase "InterAgentBroadcast"
250+ , testCase "Inter-Agent Broadcast"
478 (delayedAssertion
251 (delayedAssertion
479 "expected (), but no broadcast was received"
252 "expected (), but no broadcast was received"
480− node1 (Just ()) testAgentBroadcast)
253+ node1 () testAgentBroadcast)
481− , testCase "AgentMailboxHandling"
254+ , testCase "Agent Mailbox Handling"
482 (delayedAssertion
255 (delayedAssertion
483 "expected (Just ()), but no regular (mailbox) input was handled"
256 "expected (Just ()), but no regular (mailbox) input was handled"
484 node1 (Just ()) testAgentMailboxHandling)
257 node1 (Just ()) testAgentMailboxHandling)
485− , testCase "AgentDualInputHandling"
258+ , testCase "Agent Dual Input Handling"
486 (delayedAssertion
259 (delayedAssertion
487 "expected sum = 15, but the result was Nothing"
260 "expected sum = 15, but the result was Nothing"
488 node1 (Just 15 :: Maybe Int) testAgentDualInput)
261 node1 (Just 15 :: Maybe Int) testAgentDualInput)
489− , testCase "AgentInputPrioritisation"
262+ , testCase "Agent Input Prioritisation"
490 (delayedAssertion
263 (delayedAssertion
491 "expected [first, second, third, fourth, fifth], but result diverged"
264 "expected [first, second, third, fourth, fifth], but result diverged"
492 node1 (sort ["first", "second",
265 node1 (sort ["first", "second",
493 "third", "fourth",
266 "third", "fourth",
494 "fifth"]) testAgentPrioritisation)
267 "fifth"]) testAgentPrioritisation)
268+ testGroup "Mx Events" [
269+ testCase "Monitor Events"
270+ (runProcess node1 (testMxUnsafeSendEvents node2))
495 ]
271 ]
496− , testGroup "MxEvents" [
497− testCase "NameRegistrationEvents"
498− (delayedAssertion
499− "expected registration events to map to the correct ProcessId"
500− node1 () testMxRegEvents)
501− , testCase "PostDeathNameUnRegistrationEvents"
502− (delayedAssertion
503− "expected process deaths to result in unregistration events"
504− node1 () (testMxRegMon node2))
505− , testGroup "SendEvents" $ buildTestCases node1 node2
506− ]
272+ ]]
507− ]
508− where
509− buildTestCases n1 n2 = let nid = localNodeId n2 in build n1 n2 [
510− ("NSend", [
511− ("nsend", testNSend nsend)
512− , ("Unsafe.nsend", testNSend Unsafe.nsend)
513− , ("nsendRemote", testNSend (nsendRemote nid))
514− , ("Unsafe.nsendRemote", testNSend (Unsafe.nsendRemote nid))
515− ])
516− , ("Send", [
517− ("send", testSend send)
518− , ("Unsafe.send", testSend Unsafe.send)
519− , ("usend", testSend usend)
520− , ("Unsafe.usend", testSend Unsafe.usend)
521− , ("sendChan", testChan sendChan)
522− , ("Unsafe.sendChan", testChan Unsafe.sendChan)
523− ])
524− , ("Forward", [
525− ("forward", testSend (\p m -> forward (unsafeCreateUnencodedMessage m) p))
526− , ("uforward", testSend (\p m -> uforward (unsafeCreateUnencodedMessage m) p))
527− ])
528− ]
529−
530− build :: LocalNode
531− -> LocalNode
532− -> [(String, [(String, (Maybe LocalNode -> Process ()))])]
533− -> [Test]
534− build n ln specs =
535− [ testGroup (intercalate "-" [groupName, caseSuffix]) [
536− testCase (intercalate "-" [caseName, caseSuffix])
537− (runProcess n (caseImpl caseNode))
538− | (caseName, caseImpl) <- groupCases
539− ]
540− | (groupName, groupCases) <- specs
541− , (caseSuffix, caseNode) <- [("RemotePid", Just ln), ("LocalPid", Nothing)]
542− ]
distributed-process-tests/src/Control/Distributed/Process/Tests/Receive.hs+8−7
@@ -10,12 +10,13 @@ import Network.Transport.Test (TestTransport(..))
10
10
11import Network.Transport (Transport)
11import Network.Transport (Transport)
12import Control.Distributed.Process
12import Control.Distributed.Process
13+import Control.Distributed.Process.Closure
13import Control.Distributed.Process.Node
14import Control.Distributed.Process.Node
14
15
15import Control.Monad
16import Control.Monad
16
17
17import Test.HUnit (Assertion, (@?=))
18import Test.HUnit (Assertion, (@?=))
18−import Test.Framework (Test)
19+import Test.Framework (Test, defaultMain)
19import Test.Framework.Providers.HUnit (testCase)
20import Test.Framework.Providers.HUnit (testCase)
20
21
21-- Tests:
22-- Tests:
@@ -42,7 +43,7 @@ recTest2 :: ReceivePort ()
42 -> SendPort String
43 -> SendPort String
43 -> ReceivePort String -> ReceivePort String
44 -> ReceivePort String -> ReceivePort String
44 -> Process ()
45 -> Process ()
45−recTest2 wait sync r1 _ = do
46+recTest2 wait sync r1 r2 = do
46 forever $ do
47 forever $ do
47 receiveChan wait
48 receiveChan wait
48 r <- receiveWait
49 r <- receiveWait
@@ -55,7 +56,7 @@ recTest3 :: ReceivePort ()
55 -> SendPort String
56 -> SendPort String
56 -> ReceivePort String -> ReceivePort String
57 -> ReceivePort String -> ReceivePort String
57 -> Process ()
58 -> Process ()
58−recTest3 wait sync r1 _ = do
59+recTest3 wait sync r1 r2 = do
59 forever $ do
60 forever $ do
60 receiveChan wait
61 receiveChan wait
61 r <- receiveWait
62 r <- receiveWait
@@ -68,7 +69,7 @@ recTest4 :: ReceivePort ()
68 -> SendPort String
69 -> SendPort String
69 -> ReceivePort String -> ReceivePort String
70 -> ReceivePort String -> ReceivePort String
70 -> Process ()
71 -> Process ()
71−recTest4 wait sync r1 _ = do
72+recTest4 wait sync r1 r2 = do
72 forever $ do
73 forever $ do
73 receiveChan wait
74 receiveChan wait
74 r <- receiveWait
75 r <- receiveWait
@@ -82,11 +83,11 @@ master :: Process ()
82master = do
83master = do
83 (waits,waitr) <- newChan
84 (waits,waitr) <- newChan
84 (syncs,syncr) <- newChan
85 (syncs,syncr) <- newChan
85− let go expected = do
86+ let go expect = do
86 sendChan waits ()
87 sendChan waits ()
87 r <- receiveChan syncr
88 r <- receiveChan syncr
88− liftIO $ print (r, expected, r == expected)
89+ liftIO $ print (r,expect, r == expect)
89− liftIO $ r @?= expected
90+ liftIO $ r @?= expect
90
91
91 liftIO $ putStrLn "---- Test 1 ----"
92 liftIO $ putStrLn "---- Test 1 ----"
92 (s1,r1) <- newChan
93 (s1,r1) <- newChan
distributed-process-tests/src/Control/Distributed/Process/Tests/Stats.hs+7−11
@@ -29,7 +29,7 @@ import Test.Framework.Providers.HUnit (testCase)
29
29
30testLocalDeadProcessInfo :: TestResult (Maybe ProcessInfo) -> Process ()
30testLocalDeadProcessInfo :: TestResult (Maybe ProcessInfo) -> Process ()
31testLocalDeadProcessInfo result = do
31testLocalDeadProcessInfo result = do
32− pid <- spawnLocal $ do (_ :: String) <- expect; return ()
32+ pid <- spawnLocal $ do "finish" <- expect; return ()
33 mref <- monitor pid
33 mref <- monitor pid
34 send pid "finish"
34 send pid "finish"
35 _ <- receiveWait [
35 _ <- receiveWait [
@@ -52,10 +52,8 @@ testLocalLiveProcessInfo result = do
52 link self
52 link self
53 mRef <- monitor self
53 mRef <- monitor self
54 stash mon mRef
54 stash mon mRef
55− res <- expect
55+ "die" <- expect
56− case res of
57− "die" -> return ()
56+ return ()
58− _ -> die $ "unexpected message received: " ++ res
59
57
60 monRef <- liftIO $ takeMVar mon
58 monRef <- liftIO $ takeMVar mon
61
59
@@ -86,9 +84,9 @@ testRemoteLiveProcessInfo TestTransport{..} node1 = do
86 -- our send op shouldn't overtake link or monitor requests AFAICT
84 -- our send op shouldn't overtake link or monitor requests AFAICT
87 -- so a little table tennis should get us synchronised properly
85 -- so a little table tennis should get us synchronised properly
88 send serverPid (self, "ping")
86 send serverPid (self, "ping")
89− pong <- expect
87+ "pong" <- expect
90 pInfo <- getProcessInfo serverPid
88 pInfo <- getProcessInfo serverPid
91− stash result $ pong == "pong" && pInfo /= Nothing
89+ stash result $ pInfo /= Nothing
92 where
90 where
93 launchRemote :: MVar ProcessId -> IO ()
91 launchRemote :: MVar ProcessId -> IO ()
94 launchRemote locMV = do
92 launchRemote locMV = do
@@ -99,10 +97,8 @@ testRemoteLiveProcessInfo TestTransport{..} node1 = do
99 _ <- receiveWait [
97 _ <- receiveWait [
100 match (\(pid, "ping") -> send pid "pong")
98 match (\(pid, "ping") -> send pid "pong")
101 ]
99 ]
102− res <- expect
100+ "stop" <- expect
103− case res of
104− "stop" -> return ()
101+ return ()
105− _ -> die $ "unexpected message received: " ++ res
106 return ()
102 return ()
107
103
108 withActiveRemote :: LocalNode
104 withActiveRemote :: LocalNode
distributed-process-tests/src/Control/Distributed/Process/Tests/Tracing.hs+11−23
@@ -125,11 +125,9 @@ testTraceSending result = do
125 (\ev ->
125 (\ev ->
126 case ev of
126 case ev of
127 (MxSent to from msg) -> do
127 (MxSent to from msg) -> do
128− mS <- unwrapMessage msg :: Process (Maybe String)
128+ (Just s) <- unwrapMessage msg :: Process (Maybe String)
129− case mS of
130− (Just s) -> do stash res (to == pid && from == self && s == "hello there")
129+ stash res (to == pid && from == self && s == "hello there")
131− stash res (to == pid && from == self)
130+ stash res (to == pid && from == self)
132− _ -> die "failed state invariant, message type unmatched..."
133 _ ->
131 _ ->
134 return ()) $ do
132 return ()) $ do
135 send pid "hello there"
133 send pid "hello there"
@@ -248,8 +246,12 @@ testRemoteTraceRelay TestTransport{..} result =
248 -- Here we set up that relay, and then wait for a signal
246 -- Here we set up that relay, and then wait for a signal
249 -- that the tracer (on node1) has seen the expected
247 -- that the tracer (on node1) has seen the expected
250 -- MxSpawned message, at which point we're finished
248 -- MxSpawned message, at which point we're finished
249+ (Just log') <- whereis "logger"
250+ pid <- liftIO $ forkProcess node2 $ do
251+ logRelay <- spawnLocal $ relay log'
252+ reregister "logger" logRelay
253+ getSelfNode >>= stash mvNid >> (expect :: Process ())
251
254
252− pid <- splinchLogger node2 mvNid
253 nid <- liftIO $ takeMVar mvNid
255 nid <- liftIO $ takeMVar mvNid
254 mref <- monitor pid
256 mref <- monitor pid
255 observedPid <- liftIO $ newEmptyMVar
257 observedPid <- liftIO $ newEmptyMVar
@@ -289,17 +291,6 @@ testRemoteTraceRelay TestTransport{..} result =
289 -- and just to be polite...
291 -- and just to be polite...
290 liftIO $ closeLocalNode node2
292 liftIO $ closeLocalNode node2
291
293
292− where
293− splinchLogger n2 mv = do
294− mLog <- whereis "logger"
295− case mLog of
296− Nothing -> die "no logger registered"
297− Just log' -> do liftIO $ forkProcess n2 $ do
298− logRelay <- spawnLocal $ relay log'
299− reregister "logger" logRelay
300− getSelfNode >>= stash mv >> (expect :: Process ())
301−
302−
303-- | Sets the value of an environment variable while executing the given IO
294-- | Sets the value of an environment variable while executing the given IO
304-- computation and restores the preceeding value upon completion.
295-- computation and restores the preceeding value upon completion.
305withEnv :: String -> String -> IO a -> IO a
296withEnv :: String -> String -> IO a -> IO a
@@ -328,12 +319,9 @@ testSystemLoggerMsg t action interestingMessage =
328 -- Wait for the trace message.
319 -- Wait for the trace message.
329 receiveWait [ matchIf interestingMessage' $ const $ return () ]
320 receiveWait [ matchIf interestingMessage' $ const $ return () ]
330 -- Only one interesting message should arrive.
321 -- Only one interesting message should arrive.
331− expectedTimeout <- receiveTimeout
322+ Nothing <- receiveTimeout 100000
332− 100000
333− [ matchIf interestingMessage' $ const $ return () ]
323+ [ matchIf interestingMessage' $ const $ return () ]
334− case expectedTimeout of
335− Nothing -> return ()
324+ return ()
336− Just _ -> die "Unexpected message arrived..."
337
325
338
326
339-- | Tests that one and only one trace message is produced when a message is
327-- | Tests that one and only one trace message is produced when a message is
distributed-process-tests/tests/runInMemory.hs+1
@@ -8,6 +8,7 @@ import Network.Transport.Test (TestTransport(..))
8import Network.Transport.InMemory
8import Network.Transport.InMemory
9import Test.Framework (defaultMainWithArgs)
9import Test.Framework (defaultMainWithArgs)
10
10
11+import Control.Concurrent (threadDelay)
11import System.Environment (getArgs)
12import System.Environment (getArgs)
12
13
13main :: IO ()
14main :: IO ()
distributed-process-tests/tests/runTCP.hs+2−2
@@ -10,7 +10,6 @@ import Network.Transport.TCP
10 ( createTransportExposeInternals
10 ( createTransportExposeInternals
11 , TransportInternals(socketBetween)
11 , TransportInternals(socketBetween)
12 , defaultTCPParameters
12 , defaultTCPParameters
13− , defaultTCPAddr
14 , TCPParameters(..)
13 , TCPParameters(..)
15 )
14 )
16import Test.Framework (defaultMainWithArgs)
15import Test.Framework (defaultMainWithArgs)
@@ -25,7 +24,8 @@ main = do
25 hSetBuffering stdout LineBuffering
24 hSetBuffering stdout LineBuffering
26 hSetBuffering stderr LineBuffering
25 hSetBuffering stderr LineBuffering
27 Right (transport, internals) <-
26 Right (transport, internals) <-
28− createTransportExposeInternals (defaultTCPAddr "127.0.0.1" "8080")
27+ createTransportExposeInternals "127.0.0.1" "8080"
28+ (\sn -> ("127.0.0.1", sn))
29 defaultTCPParameters { transportConnectTimeout = Just 3000000 }
29 defaultTCPParameters { transportConnectTimeout = Just 3000000 }
30 ts <- tests TestTransport
30 ts <- tests TestTransport
31 { testTransport = transport
31 { testTransport = transport
distributed-process.cabal+2−2
@@ -6,7 +6,7 @@ License: BSD3
6License-File: LICENSE
6License-File: LICENSE
7Copyright: Well-Typed LLP, Tweag I/O Limited
7Copyright: Well-Typed LLP, Tweag I/O Limited
8Author: Duncan Coutts, Nicolas Wu, Edsko de Vries
8Author: Duncan Coutts, Nicolas Wu, Edsko de Vries
9−Maintainer: Tim Watson <watson.timothy@gmail.com>
9+Maintainer: Facundo Domínguez <facundo.dominguez@tweag.io>
10Stability: experimental
10Stability: experimental
11Homepage: http://haskell-distributed.github.com/
11Homepage: http://haskell-distributed.github.com/
12Bug-Reports: https://github.com/haskell-distributed/distributed-process/issues
12Bug-Reports: https://github.com/haskell-distributed/distributed-process/issues
@@ -21,7 +21,7 @@ Description: This is an implementation of Cloud Haskell, as described in
21
21
22 You will probably also want to install a Cloud Haskell backend such
22 You will probably also want to install a Cloud Haskell backend such
23 as distributed-process-simplelocalnet.
23 as distributed-process-simplelocalnet.
24−Tested-With: GHC==7.10.3 GHC==8.0.2 GHC==8.2.2 GHC==8.4.4
24+Tested-With: GHC==7.6.3 GHC==7.8.4 GHC==7.10.3 GHC==8.0.1
25Category: Control
25Category: Control
26extra-source-files: ChangeLog
26extra-source-files: ChangeLog
27
27
src/Control/Distributed/Process/Internal/Primitives.hs+41−60
@@ -76,7 +76,6 @@ module Control.Distributed.Process.Internal.Primitives
76 , unlink
76 , unlink
77 , monitor
77 , monitor
78 , unmonitor
78 , unmonitor
79− , unmonitorAsync
80 , withMonitor
79 , withMonitor
81 , withMonitor_
80 , withMonitor_
82 -- * Logging
81 -- * Logging
@@ -254,15 +253,17 @@ send them msg = do
254 node = processNode proc
253 node = processNode proc
255 nodeId = localNodeId node
254 nodeId = localNodeId node
256 destNode = (processNodeId them) in do
255 destNode = (processNodeId them) in do
256+ case destNode == nodeId of
257+ True -> sendLocal them msg
258+ False -> liftIO $ sendMessage (processNode proc)
259+ (ProcessIdentifier (processId proc))
260+ (ProcessIdentifier them)
261+ NoImplicitReconnect
262+ msg
263+ -- We do not fire the trace event until after the sending is complete;
264+ -- In the remote case, 'sendMessage' can block in the networking stack.
257 liftIO $ traceEvent (localEventBus node)
265 liftIO $ traceEvent (localEventBus node)
258 (MxSent them us (createUnencodedMessage msg))
266 (MxSent them us (createUnencodedMessage msg))
259− if destNode == nodeId
260− then sendLocal them msg
261− else liftIO $ sendMessage (processNode proc)
262− (ProcessIdentifier (processId proc))
263− (ProcessIdentifier them)
264− NoImplicitReconnect
265− msg
266
267
267-- | /Unsafe/ variant of 'send'. This function makes /no/ attempt to serialize
268-- | /Unsafe/ variant of 'send'. This function makes /no/ attempt to serialize
268-- and (in the case when the destination process resides on the same local
269-- and (in the case when the destination process resides on the same local
@@ -282,12 +283,9 @@ unsafeSend = Unsafe.send
282--
283--
283usend :: Serializable a => ProcessId -> a -> Process ()
284usend :: Serializable a => ProcessId -> a -> Process ()
284usend them msg = do
285usend them msg = do
285− proc <- ask
286+ here <- getSelfNode
286 let there = processNodeId them
287 let there = processNodeId them
287− let (us, node) = (processId proc, processNode proc)
288+ if here == there
288− let msg' = wrapMessage msg
289− liftIO $ traceEvent (localEventBus node) (MxSent them us msg')
290− if localNodeId (processNode proc) == there
291 then sendLocal them msg
289 then sendLocal them msg
292 else sendCtrlMsg (Just there) $ UnreliableSend (processLocalId them)
290 else sendCtrlMsg (Just there) $ UnreliableSend (processLocalId them)
293 (createMessage msg)
291 (createMessage msg)
@@ -307,13 +305,7 @@ expect = receiveWait [match return]
307-- Channels --
305-- Channels --
308--------------------------------------------------------------------------------
306--------------------------------------------------------------------------------
309
307
310−-- | Create a new typed channel, bound to the calling Process
308+-- | Create a new typed channel
311−--
312−-- Note that the channel is bound to the lifecycle of the process that evaluates
313−-- this function, such that when it dies/exits, the channel will no longer
314−-- function, but will remain accessible. Thus reading from the ReceivePort will
315−-- fail silently thereafter, blocking indefinitely (unless a timeout is used).
316−--
317newChan :: Serializable a => Process (SendPort a, ReceivePort a)
309newChan :: Serializable a => Process (SendPort a, ReceivePort a)
318newChan = do
310newChan = do
319 proc <- ask
311 proc <- ask
@@ -341,16 +333,13 @@ newChan = do
341sendChan :: Serializable a => SendPort a -> a -> Process ()
333sendChan :: Serializable a => SendPort a -> a -> Process ()
342sendChan (SendPort cid) msg = do
334sendChan (SendPort cid) msg = do
343 proc <- ask
335 proc <- ask
344− let node = processNode proc
345− pid = processId proc
346− us = localNodeId node
336+ let node = localNodeId (processNode proc)
347− them = processNodeId (sendPortProcessId cid) in do
337+ destNode = processNodeId (sendPortProcessId cid) in do
348− liftIO $ traceEvent (localEventBus node) (MxSentToPort pid cid $ wrapMessage msg)
349− case them == us of
338+ case destNode == node of
350 True -> sendChanLocal cid msg
339 True -> sendChanLocal cid msg
351 False -> do
340 False -> do
352− liftIO $ sendBinary node
341+ liftIO $ sendBinary (processNode proc)
353− (ProcessIdentifier pid)
342+ (ProcessIdentifier (processId proc))
354 (SendPortIdentifier cid)
343 (SendPortIdentifier cid)
355 NoImplicitReconnect
344 NoImplicitReconnect
356 msg
345 msg
@@ -411,11 +400,8 @@ newtype Match b = Match { unMatch :: MatchOn Message (Process b) }
411receiveWait :: [Match b] -> Process b
400receiveWait :: [Match b] -> Process b
412receiveWait ms = do
401receiveWait ms = do
413 queue <- processQueue <$> ask
402 queue <- processQueue <$> ask
414− mProc <- liftIO $ dequeue queue Blocking (map unMatch ms)
403+ Just proc <- liftIO $ dequeue queue Blocking (map unMatch ms)
415− case mProc of
416− Just proc' -> proc'
404+ proc
417− Nothing -> die $ "System Invariant Violation: CQueue.hs returned `Nothing` "
418− ++ "in the absence of a timeout value."
419
405
420-- | Like 'receiveWait' but with a timeout.
406-- | Like 'receiveWait' but with a timeout.
421--
407--
@@ -494,14 +480,15 @@ forward msg them = do
494 us = processId proc
480 us = processId proc
495 nid = localNodeId node
481 nid = localNodeId node
496 destNode = (processNodeId them) in do
482 destNode = (processNodeId them) in do
497− liftIO $ traceEvent (localEventBus node) (MxSent them us msg)
498− if destNode == nid
483+ case destNode == nid of
499− then sendCtrlMsg Nothing (LocalSend them msg)
484+ True -> sendCtrlMsg Nothing (LocalSend them msg)
500− else liftIO $ sendPayload (processNode proc)
485+ False ->
486+ liftIO $ do
501− (ProcessIdentifier (processId proc))
487+ sendPayload (processNode proc) (ProcessIdentifier (processId proc)) (ProcessIdentifier them) NoImplicitReconnect (messageToPayload msg)
502− (ProcessIdentifier them)
488+ -- We do not fire the trace event until after the sending is complete;
503− NoImplicitReconnect
489+ -- In this remote case, 'sendMessage' can block in the networking stack.
504− (messageToPayload msg)
490+ liftIO $ traceEvent (localEventBus node)
491+ (MxSent them us msg)
505
492
506-- | Forward a raw 'Message' to the given 'ProcessId'.
493-- | Forward a raw 'Message' to the given 'ProcessId'.
507--
494--
@@ -515,10 +502,14 @@ uforward msg them = do
515 us = processId proc
502 us = processId proc
516 nid = localNodeId node
503 nid = localNodeId node
517 destNode = (processNodeId them) in do
504 destNode = (processNodeId them) in do
518− liftIO $ traceEvent (localEventBus node) (MxSent them us msg)
519− if destNode == nid
505+ case destNode == nid of
520− then sendCtrlMsg Nothing (LocalSend them msg)
506+ True -> sendCtrlMsg Nothing (LocalSend them msg)
507+ False -> do
521− else sendCtrlMsg (Just destNode) $ UnreliableSend (processLocalId them) msg
508+ sendCtrlMsg (Just destNode) $ UnreliableSend (processLocalId them) msg
509+ -- We do not fire the trace event until after the sending is complete;
510+ -- In the remote case, 'sendCtrlMsg' can block in the networking stack.
511+ liftIO $ traceEvent (localEventBus node)
512+ (MxSent them us msg)
522
513
523-- | Wrap a 'Serializable' value in a 'Message'. Note that 'Message's are
514-- | Wrap a 'Serializable' value in a 'Message'. Note that 'Message's are
524-- 'Serializable' - like the datum they contain - but also note, deserialising
515-- 'Serializable' - like the datum they contain - but also note, deserialising
@@ -1227,12 +1218,8 @@ whereisRemoteAsync nid label = do
1227
1218
1228-- | Named send to a process in the local registry (asynchronous)
1219-- | Named send to a process in the local registry (asynchronous)
1229nsend :: Serializable a => String -> a -> Process ()
1220nsend :: Serializable a => String -> a -> Process ()
1230−nsend label msg = do
1221+nsend label msg =
1231− proc <- ask
1232− let msg' = createUnencodedMessage msg
1233− liftIO $ traceEvent (localEventBus (processNode proc))
1234− (MxSentToName label (processId proc) msg')
1235− sendCtrlMsg Nothing (NamedSend label msg')
1222+ sendCtrlMsg Nothing (NamedSend label (createUnencodedMessage msg))
1236
1223
1237-- | Named send to a process in the local registry (asynchronous).
1224-- | Named send to a process in the local registry (asynchronous).
1238-- This function makes /no/ attempt to serialize and (in the case when the
1225-- This function makes /no/ attempt to serialize and (in the case when the
@@ -1244,15 +1231,9 @@ unsafeNSend = Unsafe.nsend
1244-- | Named send to a process in a remote registry (asynchronous)
1231-- | Named send to a process in a remote registry (asynchronous)
1245nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()
1232nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()
1246nsendRemote nid label msg = do
1233nsendRemote nid label msg = do
1247− proc <- ask
1234+ here <- getSelfNode
1248− let us = processId proc
1249− let node = processNode proc
1250− if localNodeId node == nid
1251− then nsend label msg
1235+ if here == nid then nsend label msg
1252− else let lbl = label ++ "@" ++ show nid in do
1253− liftIO $ traceEvent (localEventBus node)
1254− (MxSentToName lbl us (wrapMessage msg))
1255− sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))
1236+ else sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))
1256
1237
1257-- | Named send to a process in a remote registry (asynchronous)
1238-- | Named send to a process in a remote registry (asynchronous)
1258-- This function makes /no/ attempt to serialize and (in the case when the
1239-- This function makes /no/ attempt to serialize and (in the case when the
src/Control/Distributed/Process/Management.hs+2−34
@@ -76,40 +76,6 @@
76--
76--
77-- * Whether messages will be taken from the mailbox first, or the event bus.
77-- * Whether messages will be taken from the mailbox first, or the event bus.
78--
78--
79−-- Since the event bus uses STM broadcast channels to communicate with agents,
80−-- no message written to the bus successfully can be lost.
81−--
82−-- Agents can also receive messages via their mailboxes - these are subject to
83−-- the same guarantees as all inter-process message sending.
84−--
85−-- Messages dispatched on an STM broadcast channel (i.e., management event bus)
86−-- are guaranteed to be delivered with the same FIFO ordering guarantees that
87−-- exist between two communicating processes, such that communication from the
88−-- node controller's threads (i.e., MxEvent's) will never be re-ordered, but
89−-- messages dispatched to the event bus by other processes (including, but not
90−-- limited to agents) are only guaranteed to be ordered between one sender and
91−-- one receiver.
92−--
93−-- No guarantee exists for the ordering in which messages sent to an agent's
94−-- mailbox will be delivered, vs messages dispatched via the event bus.
95−--
96−-- Because of the above, there are no ordering guarantees for messages sent
97−-- between agents, or for processes to agents, except for those that apply to
98−-- messages sent between regular processes, since agents are
99−-- implemented as such.
100−--
101−-- The event bus is serial and single threaded. Anything that is published by
102−-- the node controller will be seen in FIFO order. There are no ordering
103−-- guarantees pertaining to entries published to the event bus by other
104−-- processes or agents.
105−--
106−-- It should not be possible to see, for example, an @MxReceived@ before the
107−-- corresponding @MxSent@ event, since the places where we issue the @MxSent@
108−-- write directly to the event bus (using STM) in the calling (green) thread,
109−-- before dispatching instructions to the node controller to perform the
110−-- necessary routing to deliver the message to a process (or registered name,
111−-- or typed channel) locally or remotely.
112−--
113-- [Management Data API]
79-- [Management Data API]
114--
80--
115-- Both management agents and clients of the API have access to a variety of
81-- Both management agents and clients of the API have access to a variety of
@@ -178,6 +144,8 @@
178-- > monitorNames = getSelfPid >>= nsend "name-monitor"
144-- > monitorNames = getSelfPid >>= nsend "name-monitor"
179-- > monitorNames2 = getSelfPid >>= mxNotify
145-- > monitorNames2 = getSelfPid >>= mxNotify
180--
146--
147+-- For some real-world examples, see the distributed-process-platform package.
148+--
181-- [Performance, Stablity and Scalability]
149-- [Performance, Stablity and Scalability]
182--
150--
183-- /Management Agents/ offer numerous advantages over regular processes:
151-- /Management Agents/ offer numerous advantages over regular processes:
src/Control/Distributed/Process/Management/Internal/Trace/Primitives.hs+2−8
@@ -33,7 +33,6 @@ import Control.Distributed.Process.Internal.Primitives
33 ( whereis
33 ( whereis
34 , newChan
34 , newChan
35 , receiveChan
35 , receiveChan
36− , die
37 )
36 )
38import Control.Distributed.Process.Management.Internal.Trace.Types
37import Control.Distributed.Process.Management.Internal.Trace.Types
39 ( TraceArg(..)
38 ( TraceArg(..)
@@ -169,11 +168,6 @@ withRegisteredTracer act = do
169 withLocalTracer $ \t -> liftIO $ Tracer.getCurrentTraceClient t sp
168 withLocalTracer $ \t -> liftIO $ Tracer.getCurrentTraceClient t sp
170 currentTracer <- receiveChan rp
169 currentTracer <- receiveChan rp
171 case currentTracer of
170 case currentTracer of
172− Nothing -> do mTP <- whereis "tracer.initial"
171+ Nothing -> do { (Just p') <- whereis "tracer.initial"; act p' }
173− -- NB: this should NOT ever happen, but forcing pattern matches
174− -- is not considered cool in later versions of MonadFail
175− case mTP of
176− Just p' -> act p'
177− Nothing -> die $ "System Invariant Violation: Tracer Process "
178− ++ "Name Not Found (whereis tracer.initial)"
179 (Just p) -> act p
172 (Just p) -> act p
173+
src/Control/Distributed/Process/Management/Internal/Types.hs−7
@@ -21,7 +21,6 @@ import Control.Concurrent.STM
21import Control.Distributed.Process.Internal.Types
21import Control.Distributed.Process.Internal.Types
22 ( Process
22 ( Process
23 , ProcessId
23 , ProcessId
24− , SendPortId
25 , Message
24 , Message
26 , DiedReason
25 , DiedReason
27 , NodeId
26 , NodeId
@@ -31,7 +30,6 @@ import qualified Control.Monad.State as ST
31 ( MonadState
30 ( MonadState
32 , StateT
31 , StateT
33 )
32 )
34−import Control.Monad.Fix (MonadFix)
35import Data.Binary
33import Data.Binary
36import Data.Typeable (Typeable)
34import Data.Typeable (Typeable)
37import GHC.Generics
35import GHC.Generics
@@ -60,12 +58,8 @@ data MxEvent =
60 -- ^ fired whenever a message is sent from a local process
58 -- ^ fired whenever a message is sent from a local process
61 | MxSentToName String ProcessId Message
59 | MxSentToName String ProcessId Message
62 -- ^ fired whenever a named send occurs
60 -- ^ fired whenever a named send occurs
63− | MxSentToPort ProcessId SendPortId Message
64− -- ^ fired whenever a sendChan occurs
65 | MxReceived ProcessId Message
61 | MxReceived ProcessId Message
66 -- ^ fired whenever a message is received by a local process
62 -- ^ fired whenever a message is received by a local process
67− | MxReceivedPort SendPortId Message
68− -- ^ fired whenever a message is received via a typed channel
69 | MxConnected ConnectionId EndPointAddress
63 | MxConnected ConnectionId EndPointAddress
70 -- ^ fired when a network-transport connection is first established
64 -- ^ fired when a network-transport connection is first established
71 | MxDisconnected ConnectionId EndPointAddress
65 | MxDisconnected ConnectionId EndPointAddress
@@ -120,7 +114,6 @@ newtype MxAgent s a =
120 } deriving ( Functor
114 } deriving ( Functor
121 , Monad
115 , Monad
122 , MonadIO
116 , MonadIO
123− , MonadFix
124 , ST.MonadState (MxAgentState s)
117 , ST.MonadState (MxAgentState s)
125 , Typeable
118 , Typeable
126 , Applicative
119 , Applicative
src/Control/Distributed/Process/Node.hs+19−38
@@ -28,12 +28,9 @@ import Data.Binary (decode)
28import Data.Map (Map)
28import Data.Map (Map)
29import qualified Data.Map as Map
29import qualified Data.Map as Map
30 ( empty
30 ( empty
31− , delete
32 , toList
31 , toList
33 , fromList
32 , fromList
34− , insert
33+ , filter
35− , lookup
36− , partition
37 , partitionWithKey
34 , partitionWithKey
38 , elems
35 , elems
39 , size
36 , size
@@ -186,9 +183,6 @@ import Control.Distributed.Process.Internal.Types
186import Control.Distributed.Process.Management.Internal.Agent
183import Control.Distributed.Process.Management.Internal.Agent
187 ( mxAgentController
184 ( mxAgentController
188 )
185 )
189−import Control.Distributed.Process.Management.Internal.Types
190− ( MxEvent(..)
191− )
192import qualified Control.Distributed.Process.Management.Internal.Trace.Remote as Trace
186import qualified Control.Distributed.Process.Management.Internal.Trace.Remote as Trace
193 ( remoteTable
187 ( remoteTable
194 )
188 )
@@ -201,6 +195,9 @@ import Control.Distributed.Process.Management.Internal.Trace.Types
201 , traceLogFmt
195 , traceLogFmt
202 , enableTrace
196 , enableTrace
203 )
197 )
198+import Control.Distributed.Process.Management.Internal.Types
199+ ( MxEvent(..)
200+ )
204import Control.Distributed.Process.Serializable (Serializable)
201import Control.Distributed.Process.Serializable (Serializable)
205import Control.Distributed.Process.Internal.Messaging
202import Control.Distributed.Process.Internal.Messaging
206 ( sendBinary
203 ( sendBinary
@@ -213,9 +210,6 @@ import Control.Distributed.Process.Internal.Primitives
213 , match
210 , match
214 , sendChan
211 , sendChan
215 , unwrapMessage
212 , unwrapMessage
216− , monitor
217− , unmonitorAsync
218− , getSelfNode
219 , SayMessage(..)
213 , SayMessage(..)
220 )
214 )
221import Control.Distributed.Process.Internal.Types (SendPort, Tracer(..))
215import Control.Distributed.Process.Internal.Types (SendPort, Tracer(..))
@@ -492,7 +486,7 @@ type IncomingConnection = (NT.EndPointAddress, IncomingTarget)
492data IncomingTarget =
486data IncomingTarget =
493 Uninit
487 Uninit
494 | ToProc ProcessId (Weak (CQueue Message))
488 | ToProc ProcessId (Weak (CQueue Message))
495− | ToChan SendPortId TypedChannel
489+ | ToChan TypedChannel
496 | ToNode
490 | ToNode
497
491
498data ConnectionState = ConnectionState {
492data ConnectionState = ConnectionState {
@@ -547,17 +541,13 @@ handleIncomingMessages node = go initConnectionState
547 enqueue queue msg -- 'enqueue' is strict
541 enqueue queue msg -- 'enqueue' is strict
548 trace node (MxReceived pid msg)
542 trace node (MxReceived pid msg)
549 go st
543 go st
550− Just (_, ToChan chId (TypedChannel chan')) -> do
544+ Just (_, ToChan (TypedChannel chan')) -> do
551 mChan <- deRefWeak chan'
545 mChan <- deRefWeak chan'
552 -- If mChan is Nothing, the process has given up the read end of
546 -- If mChan is Nothing, the process has given up the read end of
553 -- the channel and we simply ignore the incoming message
547 -- the channel and we simply ignore the incoming message
554− forM_ mChan $ \chan -> do
548+ forM_ mChan $ \chan -> atomically $
555− msg' <- atomically $ do
556− msg <- return $! decode (BSL.fromChunks payload)
557− -- We make sure the message is fully decoded when it is enqueued
549+ -- We make sure the message is fully decoded when it is enqueued
558− writeTQueue chan msg
550+ writeTQueue chan $! decode (BSL.fromChunks payload)
559− return msg
560− trace node $ MxReceivedPort chId $ unsafeCreateUnencodedMessage msg'
561 go st
551 go st
562 Just (_, ToNode) -> do
552 Just (_, ToNode) -> do
563 let ctrlMsg = decode . BSL.fromChunks $ payload
553 let ctrlMsg = decode . BSL.fromChunks $ payload
@@ -584,7 +574,7 @@ handleIncomingMessages node = go initConnectionState
584 mChannel <- withMVar (processState proc) $ return . (^. typedChannelWithId lcid)
574 mChannel <- withMVar (processState proc) $ return . (^. typedChannelWithId lcid)
585 case mChannel of
575 case mChannel of
586 Just channel ->
576 Just channel ->
587− go (incomingAt cid ^= Just (src, ToChan chId channel) $ st)
577+ go (incomingAt cid ^= Just (src, ToChan channel) $ st)
588 Nothing ->
578 Nothing ->
589 invalidRequest cid st $
579 invalidRequest cid st $
590 "incoming attempt to connect to unknown channel of"
580 "incoming attempt to connect to unknown channel of"
@@ -913,11 +903,7 @@ ncEffectDied ident reason = do
913
903
914 modify' $ (links ^= unaffectedLinks') . (monitors ^= unaffectedMons')
904 modify' $ (links ^= unaffectedLinks') . (monitors ^= unaffectedMons')
915
905
916− -- we now consider all labels for this identifier unregistered
917− let toDrop pid = not $ ident `impliesDeathOf` ProcessIdentifier pid
906+ modify' $ registeredHere ^: Map.filter (\pid -> not $ ident `impliesDeathOf` ProcessIdentifier pid)
918− (keepNames, dropNames) <- Map.partition toDrop <$> gets (^. registeredHere)
919− mapM_ (\(p, l) -> liftIO $ trace node (MxUnRegistered l p)) (Map.toList dropNames)
920− modify' $ registeredHere ^= keepNames
921
907
922 remaining <- fmap Map.toList (gets (^. registeredOnNodes)) >>=
908 remaining <- fmap Map.toList (gets (^. registeredOnNodes)) >>=
923 mapM (\(pid,nidlist) ->
909 mapM (\(pid,nidlist) ->
@@ -969,11 +955,7 @@ ncEffectRegister from label atnode mPid reregistration = do
969 do modify' $ registeredHereFor label ^= mPid
955 do modify' $ registeredHereFor label ^= mPid
970 updateRemote node currentVal mPid
956 updateRemote node currentVal mPid
971 case mPid of
957 case mPid of
972− (Just p) -> do
973− if reregistration
974− then liftIO $ trace node (MxUnRegistered (fromJust currentVal) label)
975− else return ()
976− liftIO $ trace node (MxRegistered p label)
958+ (Just p) -> liftIO $ trace node (MxRegistered p label)
977 Nothing -> liftIO $ trace node (MxUnRegistered (fromJust currentVal) label)
959 Nothing -> liftIO $ trace node (MxUnRegistered (fromJust currentVal) label)
978 newVal <- gets (^. registeredHereFor label)
960 newVal <- gets (^. registeredHereFor label)
979 ncSendToProcess from $ unsafeCreateUnencodedMessage $
961 ncSendToProcess from $ unsafeCreateUnencodedMessage $
@@ -1054,12 +1036,11 @@ ncEffectLocalPortSend from msg = do
1054 -- If ch is Nothing, the process has given up the read end of
1036 -- If ch is Nothing, the process has given up the read end of
1055 -- the channel and we simply ignore the incoming message - this
1037 -- the channel and we simply ignore the incoming message - this
1056 ch <- deRefWeak chan'
1038 ch <- deRefWeak chan'
1057− forM_ ch $ \chan -> deliverChan node from msg chan
1039+ forM_ ch $ \chan -> deliverChan msg chan
1058− where deliverChan :: forall a . LocalNode -> SendPortId -> Message -> TQueue a -> IO ()
1040+ where deliverChan :: forall a . Message -> TQueue a -> IO ()
1059− deliverChan n p (UnencodedMessage _ raw) chan' = do
1041+ deliverChan (UnencodedMessage _ raw) chan' =
1060 atomically $ writeTQueue chan' ((unsafeCoerce raw) :: a)
1042 atomically $ writeTQueue chan' ((unsafeCoerce raw) :: a)
1061− trace n (MxReceivedPort p $ unsafeCreateUnencodedMessage raw)
1062− deliverChan _ _ (EncodedMessage _ _) _ =
1043+ deliverChan (EncodedMessage _ _) _ =
1063 -- this will not happen unless someone screws with Primitives.hs
1044 -- this will not happen unless someone screws with Primitives.hs
1064 error "invalid local channel delivery"
1045 error "invalid local channel delivery"
1065
1046
@@ -1089,7 +1070,7 @@ ncEffectGetInfo from pid =
1089 case mProc of
1070 case mProc of
1090 Nothing -> dispatch (isLocal node (ProcessIdentifier from))
1071 Nothing -> dispatch (isLocal node (ProcessIdentifier from))
1091 from (ProcessInfoNone DiedUnknownId)
1072 from (ProcessInfoNone DiedUnknownId)
1092− Just proc -> do
1073+ Just proc -> do
1093 itsLinks <- Set.map fst . BiMultiMap.lookupBy1st them <$>
1074 itsLinks <- Set.map fst . BiMultiMap.lookupBy1st them <$>
1094 gets (^. links)
1075 gets (^. links)
1095 itsMons <- BiMultiMap.lookupBy1st them <$> gets (^. monitors)
1076 itsMons <- BiMultiMap.lookupBy1st them <$> gets (^. monitors)
@@ -1103,8 +1084,8 @@ ncEffectGetInfo from pid =
1103 infoNode = (processNodeId pid)
1084 infoNode = (processNodeId pid)
1104 , infoRegisteredNames = reg
1085 , infoRegisteredNames = reg
1105 , infoMessageQueueLength = size
1086 , infoMessageQueueLength = size
1106− , infoMonitors = Set.toList itsMons
1087+ , infoMonitors = Set.toList itsMons
1107− , infoLinks = Set.toList itsLinks
1088+ , infoLinks = Set.toList itsLinks
1108 }
1089 }
1109 where dispatch :: (Serializable a)
1090 where dispatch :: (Serializable a)
1110 => Bool
1091 => Bool
src/Control/Distributed/Process/UnsafePrimitives.hs+55−47
@@ -104,26 +104,39 @@ nsend :: Serializable a => String -> a -> Process ()
104nsend label msg = do
104nsend label msg = do
105 proc <- ask
105 proc <- ask
106 let us = processId proc
106 let us = processId proc
107+ let node = localNodeId (processNode proc)
107 let msg' = wrapMessage msg
108 let msg' = wrapMessage msg
108 -- see [note: tracing]
109 -- see [note: tracing]
109 liftIO $ traceEvent (localEventBus (processNode proc))
110 liftIO $ traceEvent (localEventBus (processNode proc))
110 (MxSentToName label us msg')
111 (MxSentToName label us msg')
111 sendCtrlMsg Nothing (NamedSend label msg')
112 sendCtrlMsg Nothing (NamedSend label msg')
112
113
114+-- [note: tracing]
115+-- In the remote case, we do not fire a trace event until after sending is
116+-- complete, since 'sendMessage' can block in the networking stack.
117+--
118+-- In addition, MxSent trace messages are dispatched by the node controller's
119+-- thread for local sends, which also covers local nsend, and local usend.
120+--
121+-- Also note that tracing writes to the local node's control channel, and this
122+-- module explicitly specifies to its clients that it does unsafe message
123+-- encoding. The same is true for the messages it puts onto the Management
124+-- event bus, however we do *not* want unevaluated thunks hitting the event
125+-- bus control thread. Hence the word /Unsafe/ in this module's name!
126+--
127+
113-- | Named send to a process in a remote registry (asynchronous)
128-- | Named send to a process in a remote registry (asynchronous)
114nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()
129nsendRemote :: Serializable a => NodeId -> String -> a -> Process ()
115nsendRemote nid label msg = do
130nsendRemote nid label msg = do
116 proc <- ask
131 proc <- ask
117 let us = processId proc
132 let us = processId proc
118 let node = processNode proc
133 let node = processNode proc
119− if localNodeId node == nid
134+ if localNodeId (processNode proc) == nid
120 then nsend label msg
135 then nsend label msg
121− else
122− let lbl = label ++ "@" ++ show nid in do
123− -- see [note: tracing] NB: this is a remote call to another NC...
136+ else do -- see [note: tracing] NB: this is a remote call to another NC...
137+ sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))
124− liftIO $ traceEvent (localEventBus node)
138+ liftIO $ traceEvent (localEventBus node)
125− (MxSentToName lbl us (wrapMessage msg))
139+ (MxSentToName label us (wrapMessage msg))
126− sendCtrlMsg (Just nid) (NamedSend label (createMessage msg))
127
140
128-- | Send a message
141-- | Send a message
129send :: Serializable a => ProcessId -> a -> Process ()
142send :: Serializable a => ProcessId -> a -> Process ()
@@ -131,18 +144,17 @@ send them msg = do
131 proc <- ask
144 proc <- ask
132 let node = localNodeId (processNode proc)
145 let node = localNodeId (processNode proc)
133 destNode = (processNodeId them)
146 destNode = (processNodeId them)
134− us = (processId proc)
147+ us = (processId proc) in do
135− msg' = wrapMessage msg in do
148+ case destNode == node of
149+ True -> unsafeSendLocal them msg
150+ False -> liftIO $ do sendMessage (processNode proc)
151+ (ProcessIdentifier (processId proc))
152+ (ProcessIdentifier them)
153+ NoImplicitReconnect
154+ msg
136− -- see [note: tracing]
155+ -- see [note: tracing]
137− liftIO $ traceEvent (localEventBus (processNode proc))
156+ liftIO $ traceEvent (localEventBus (processNode proc))
138− (MxSent them us msg')
157+ (MxSent them us (wrapMessage msg))
139− if destNode == node
140− then sendCtrlMsg Nothing $ LocalSend them msg'
141− else liftIO $ sendMessage (processNode proc)
142− (ProcessIdentifier (processId proc))
143− (ProcessIdentifier them)
144− NoImplicitReconnect
145− msg
146
158
147-- | Send a message unreliably.
159-- | Send a message unreliably.
148--
160--
@@ -158,42 +170,38 @@ usend them msg = do
158 proc <- ask
170 proc <- ask
159 let there = processNodeId them
171 let there = processNodeId them
160 let (us, node) = (processId proc, processNode proc)
172 let (us, node) = (processId proc, processNode proc)
161− let msg' = wrapMessage msg
162− -- see [note: tracing]
163− liftIO $ traceEvent (localEventBus node) (MxSent them us msg')
164 if localNodeId (processNode proc) == there
173 if localNodeId (processNode proc) == there
165− then sendCtrlMsg Nothing $ LocalSend them msg'
174+ then unsafeSendLocal them msg
166− else sendCtrlMsg (Just there) $ UnreliableSend (processLocalId them)
175+ else do sendCtrlMsg (Just there) $ UnreliableSend (processLocalId them)
167− (createMessage msg)
176+ (createMessage msg)
177+ -- I would assert that is it *still* cheaper to not encode here...
178+ liftIO $ traceEvent (localEventBus node)
179+ (MxSent them us (wrapMessage msg))
168
180
181+unsafeSendLocal :: (Serializable a) => ProcessId -> a -> Process ()
182+unsafeSendLocal pid msg =
183+ let msg' = wrapMessage msg in do
184+ sendCtrlMsg Nothing $ LocalSend pid msg'
169−-- [note: tracing]
185+ -- see [note: tracing]
170−-- Note that tracing writes to the local node's control channel, and this
171−-- module explicitly specifies to its clients that it does unsafe message
172−-- encoding. The same is true for the messages it puts onto the Management
173−-- event bus, however we do *not* want unevaluated thunks hitting the event
174−-- bus control thread. Hence the word /Unsafe/ in this module's name!
175−--
176
186
177-- | Send a message on a typed channel
187-- | Send a message on a typed channel
178sendChan :: Serializable a => SendPort a -> a -> Process ()
188sendChan :: Serializable a => SendPort a -> a -> Process ()
179sendChan (SendPort cid) msg = do
189sendChan (SendPort cid) msg = do
180 proc <- ask
190 proc <- ask
181− let node = processNode proc
182− pid = processId proc
183− us = localNodeId node
191+ let node = localNodeId (processNode proc)
184− them = processNodeId (sendPortProcessId cid)
192+ destNode = processNodeId (sendPortProcessId cid) in do
185− msg' = wrapMessage msg in do
193+ case destNode == node of
186− liftIO $ traceEvent (localEventBus node) (MxSentToPort pid cid msg')
187− if them == us
188− then unsafeSendChanLocal cid msg' -- NB: we wrap to P.Message !!!
194+ True -> unsafeSendChanLocal cid msg
195+ False -> do
189− else liftIO $ sendBinary node
196+ liftIO $ sendBinary (processNode proc)
190− (ProcessIdentifier pid)
197+ (ProcessIdentifier (processId proc))
191− (SendPortIdentifier cid)
198+ (SendPortIdentifier cid)
192− NoImplicitReconnect
199+ NoImplicitReconnect
193− msg
200+ msg
194 where
201 where
195− unsafeSendChanLocal :: SendPortId -> Message -> Process ()
202+ unsafeSendChanLocal :: (Serializable a) => SendPortId -> a -> Process ()
203+ unsafeSendChanLocal spId msg' =
196− unsafeSendChanLocal p m = sendCtrlMsg Nothing $ LocalPortSend p m
204+ sendCtrlMsg Nothing $ LocalPortSend spId (wrapMessage msg')
197
205
198-- | Create an unencoded @Message@ for any @Serializable@ type.
206-- | Create an unencoded @Message@ for any @Serializable@ type.
199wrapMessage :: Serializable a => a -> Message
207wrapMessage :: Serializable a => a -> Message
stack-ghc-7.10.3.yaml+1−5
@@ -3,18 +3,14 @@ resolver: nightly-2016-03-08
3packages:
3packages:
4- '.'
4- '.'
5- distributed-process-tests/
5- distributed-process-tests/
6−- location:
7− git: https://github.com/haskell-distributed/network-transport-tcp.git
8− commit: d87d6f55697a94a1fbf211ff9c1bb769a1e129cd
9− extra-dep: true
10
6
11extra-deps:
7extra-deps:
12- rank1dynamic-0.4.0
8- rank1dynamic-0.4.0
13- distributed-static-0.3.8
9- distributed-static-0.3.8
14- network-transport-0.5.2
10- network-transport-0.5.2
11+- network-transport-tcp-0.6.0
15- network-transport-inmemory-0.5.2
12- network-transport-inmemory-0.5.2
16- rematch-0.2.0.0
13- rematch-0.2.0.0
17−- exceptions-0.8.2.1 # test dependency
18
14
19flags:
15flags:
20 distributed-process-tests:
16 distributed-process-tests:
stack-ghc-8.0.2.yaml+2−8
@@ -1,22 +1,16 @@
1−resolver: lts-9.21
1+resolver: lts-9.0
2−# added -location for network-transport-tcp-0.6.0
3−# since c7fd79 broke the client facing API for createTransport :/
4
2
5packages:
3packages:
6- '.'
4- '.'
7- distributed-process-tests/
5- distributed-process-tests/
8−- location:
9− git: https://github.com/haskell-distributed/network-transport-tcp.git
10− commit: d87d6f55697a94a1fbf211ff9c1bb769a1e129cd
11− extra-dep: true
12
6
13extra-deps:
7extra-deps:
14- rank1dynamic-0.4.0
8- rank1dynamic-0.4.0
15- distributed-static-0.3.8
9- distributed-static-0.3.8
16- network-transport-0.5.2
10- network-transport-0.5.2
11+- network-transport-tcp-0.6.0
17- network-transport-inmemory-0.5.2
12- network-transport-inmemory-0.5.2
18- rematch-0.2.0.0
13- rematch-0.2.0.0
19−# - exceptions-0.8.2.1 # test dependency
20
14
21flags:
15flags:
22 distributed-process-tests:
16 distributed-process-tests:
stack-ghc-8.2.2.yamldeleted−21
@@ -1,21 +0,0 @@
1−resolver: lts-11.22
2−
3−packages:
4−- '.'
5−- distributed-process-tests/
6−- location:
7− git: https://github.com/haskell-distributed/network-transport-tcp.git
8− commit: d87d6f55697a94a1fbf211ff9c1bb769a1e129cd
9− extra-dep: true
10−
11−extra-deps:
12−- rank1dynamic-0.4.0
13−- distributed-static-0.3.8
14−- network-transport-0.5.2
15−- network-transport-inmemory-0.5.2
16−- rematch-0.2.0.0
17−# - exceptions-0.8.2.1 # test dependency
18−
19−flags:
20− distributed-process-tests:
21− tcp: true
stack-ghc-8.4.4.yamldeleted−20
@@ -1,20 +0,0 @@
1−resolver: lts-12.18
2−
3−packages:
4−- '.'
5−- distributed-process-tests/
6−- location:
7− git: https://github.com/haskell-distributed/network-transport-tcp.git
8− commit: d87d6f55697a94a1fbf211ff9c1bb769a1e129cd
9− extra-dep: true
10−
11−extra-deps:
12−- rank1dynamic-0.4.0
13−- distributed-static-0.3.8
14−- network-transport-0.5.2
15−- network-transport-inmemory-0.5.2
16−- rematch-0.2.0.0
17−
18−flags:
19− distributed-process-tests:
20− tcp: true
stack-nightly.yamldeleted−26
@@ -1,26 +0,0 @@
1−resolver: nightly
2−
3−packages:
4−- '.'
5−- distributed-process-tests/
6−- location:
7− git: https://github.com/haskell-distributed/distributed-static.git
8− commit: c51e6aadd3a44c044e96a8e986c11d3a94b45b11
9− extra-dep: true
10−- location:
11− git: https://github.com/haskell-distributed/network-transport-tcp.git
12− commit: d58250e2388b86a1865ac4139373cbdc106e72e1
13− extra-dep: true
14−- location:
15− git: https://github.com/haskell-distributed/network-transport-inmemory.git
16− commit: 0ce97924f15a8c1570b2355151834305c9bd2a28
17− extra-dep: true
18−
19−extra-deps:
20−- rank1dynamic-0.4.0
21−- network-transport-0.5.2
22−- rematch-0.2.0.0
23−
24−flags:
25− distributed-process-tests:
26− tcp: true
stack.yaml+2−7
@@ -1,19 +1,14 @@
1−resolver: lts-12.18 # Use GHC 8.4.4
1+resolver: nightly-2017-08-20
2−
3−# TODO: we can enable nightly-2018-11-11 once PR #319 is merged
4
2
5packages:
3packages:
6- '.'
4- '.'
7- distributed-process-tests/
5- distributed-process-tests/
8−- location:
9− git: https://github.com/haskell-distributed/network-transport-tcp.git
10− commit: d87d6f55697a94a1fbf211ff9c1bb769a1e129cd
11− extra-dep: true
12
6
13extra-deps:
7extra-deps:
14- rank1dynamic-0.4.0
8- rank1dynamic-0.4.0
15- distributed-static-0.3.8
9- distributed-static-0.3.8
16- network-transport-0.5.2
10- network-transport-0.5.2
11+- network-transport-tcp-0.6.0
17- network-transport-inmemory-0.5.2
12- network-transport-inmemory-0.5.2
18- rematch-0.2.0.0
13- rematch-0.2.0.0
19
14