You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.8 KiB
56 lines
1.8 KiB
{-# LANGUAGE OverloadedStrings #-} |
|
|
|
module TestQuoteSourceClient ( |
|
unitTests |
|
) where |
|
|
|
import Test.Tasty |
|
import Test.Tasty.HUnit |
|
|
|
import ATrade.Types |
|
import ATrade.QuoteSource.Server |
|
import ATrade.QuoteSource.Client |
|
import Control.Monad |
|
import Control.Concurrent.BoundedChan |
|
import Control.Concurrent hiding (writeChan, readChan) |
|
import Control.Exception |
|
import System.ZMQ4 |
|
import Data.Time.Clock |
|
import Data.Time.Calendar |
|
import qualified Data.Text as T |
|
import Data.UUID as U |
|
import Data.UUID.V4 as UV4 |
|
|
|
makeEndpoint :: IO T.Text |
|
makeEndpoint = do |
|
uid <- toText <$> UV4.nextRandom |
|
return $ "inproc://server" `T.append` uid |
|
|
|
unitTests :: TestTree |
|
unitTests = testGroup "QuoteSource.Client" [testStartStop, testTickStream] |
|
|
|
testStartStop :: TestTree |
|
testStartStop = testCase "QuoteSource client connects and disconnects" $ withContext (\ctx -> do |
|
ep <- makeEndpoint |
|
chan <- newBoundedChan 1000 |
|
clientChan <- newBoundedChan 1000 |
|
bracket (startQuoteSourceServer chan ctx ep Nothing) stopQuoteSourceServer (\_ -> |
|
bracket (startQuoteSourceClient clientChan [] ctx ep) stopQuoteSourceClient (const yield))) |
|
|
|
testTickStream :: TestTree |
|
testTickStream = testCase "QuoteSource clients receives ticks" $ withContext (\ctx -> do |
|
ep <- makeEndpoint |
|
chan <- newBoundedChan 1000 |
|
clientChan <- newBoundedChan 1000 |
|
bracket (startQuoteSourceServer chan ctx ep Nothing) stopQuoteSourceServer (\_ -> |
|
bracket (startQuoteSourceClient clientChan ["FOOBAR"] ctx ep) stopQuoteSourceClient (\_ -> do |
|
let tick = Tick { |
|
security = "FOOBAR", |
|
datatype = LastTradePrice, |
|
timestamp = UTCTime (fromGregorian 2016 9 27) 16000, |
|
value = 1000, |
|
volume = 1} |
|
forkIO $ forever $ writeChan chan (QSSTick tick) |
|
recvdTick <- readChan clientChan |
|
tick @=? recvdTick))) |
|
|
|
|