From 54b3aaecd5cddce52da936c7e70f624b6210fe9f Mon Sep 17 00:00:00 2001
From: Denis Tereshkin <denis@kasan.ws>
Date: Fri, 9 Jul 2021 13:53:32 +0700
Subject: [PATCH] BrokerServer: test order submission from different endpoints

---
 test/TestBrokerServer.hs | 41 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 41 insertions(+)

diff --git a/test/TestBrokerServer.hs b/test/TestBrokerServer.hs
index d0a09c3..bfa47a4 100644
--- a/test/TestBrokerServer.hs
+++ b/test/TestBrokerServer.hs
@@ -29,6 +29,7 @@ import           System.ZMQ4
 unitTests :: TestTree
 unitTests = testGroup "Broker.Server" [testBrokerServerStartStop
   , testBrokerServerSubmitOrder
+  , testBrokerServerSubmitOrderDifferentIdentities
   , testBrokerServerSubmitOrderToUnknownAccount
   , testBrokerServerCancelOrder
   , testBrokerServerCancelUnknownOrder
@@ -54,6 +55,15 @@ connectAndSendOrder step sock order ep = do
   send sock [] (BL.toStrict . encode $ RequestSubmitOrder 1 "identity" order)
   threadDelay 10000
 
+connectAndSendOrderWithIdentity :: (Sender a) => (String -> IO ()) -> Socket a -> Order -> ClientIdentity -> T.Text -> IO ()
+connectAndSendOrderWithIdentity step sock order clientIdentity ep = do
+  step "Connecting"
+  connect sock (T.unpack ep)
+
+  step $ "Sending request for identity: " ++ show clientIdentity
+  send sock [] (BL.toStrict . encode $ RequestSubmitOrder 1 clientIdentity order)
+  threadDelay 10000
+
 defaultOrder :: Order
 defaultOrder = mkOrder {
     orderId = 25,
@@ -102,6 +112,37 @@ testBrokerServerSubmitOrder = testCaseSteps "Broker Server submits order" $ \ste
         Just _                          -> assertFailure "Invalid response"
         Nothing                         -> assertFailure "Invalid response"
 
+testBrokerServerSubmitOrderDifferentIdentities :: TestTree
+testBrokerServerSubmitOrderDifferentIdentities = testCaseSteps "Broker Server submits order: different identities" $ \step -> withContext $ \ctx -> do
+  step "Setup"
+  (mockBroker, broState) <- mkMockBroker ["demo"]
+  ep <- makeEndpoint
+  let orderId1 = 42
+  let orderId2 = 76
+  bracket (startBrokerServer [mockBroker] ctx ep [] defaultServerSecurityParams) stopBrokerServer $ \_ -> do
+    withSocket ctx Req $ \sock1 -> do
+      withSocket ctx Req $ \sock2 -> do
+        connectAndSendOrderWithIdentity step sock1 defaultOrder {orderId = orderId1} "identity1" ep
+        connectAndSendOrderWithIdentity step sock2 defaultOrder {orderId = orderId2} "identity2" ep
+
+        step "Checking that orders are submitted to BrokerInterface"
+        s <- readIORef broState
+        (length . orders) s @?= 2
+
+        step "Reading response for identity1"
+        resp <- decode . BL.fromStrict <$> receive sock1
+        case resp of
+          Just (ResponseOrderSubmitted localOrderId) -> localOrderId @=? orderId1
+          Just _                          -> assertFailure "Invalid response"
+          Nothing                         -> assertFailure "Invalid response"
+
+        step "Reading response for identity2"
+        resp <- decode . BL.fromStrict <$> receive sock2
+        case resp of
+          Just (ResponseOrderSubmitted localOrderId) -> localOrderId @=? orderId2
+          Just _                          -> assertFailure "Invalid response"
+          Nothing                         -> assertFailure "Invalid response"
+
 testBrokerServerSubmitOrderToUnknownAccount :: TestTree
 testBrokerServerSubmitOrderToUnknownAccount = testCaseSteps "Broker Server returns error if account is unknown" $
   \step -> withContext (\ctx -> do