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.
51 lines
1.6 KiB
51 lines
1.6 KiB
9 years ago
|
{-# LANGUAGE OverloadedStrings #-}
|
||
|
|
|
||
|
|
module TestQuoteSourceServer (
|
||
|
|
unitTests
|
||
|
|
) where
|
||
|
|
|
||
|
|
import Test.Tasty
|
||
|
|
import Test.Tasty.SmallCheck as SC
|
||
|
|
import Test.Tasty.QuickCheck as QC
|
||
|
|
import Test.Tasty.HUnit
|
||
|
|
|
||
|
|
import ATrade.Types
|
||
|
|
import qualified Data.ByteString.Lazy as BL
|
||
|
|
import ATrade.QuoteSource.Server
|
||
|
|
import Control.Monad
|
||
|
|
import Control.Monad.Loops
|
||
|
|
import Control.Concurrent.MVar
|
||
|
|
import Control.Concurrent.BoundedChan
|
||
|
|
import Control.Concurrent hiding (writeChan)
|
||
|
|
import Control.Exception
|
||
|
|
import System.ZMQ4
|
||
|
|
import Data.Time.Clock
|
||
|
|
import Data.Time.Calendar
|
||
|
|
import Data.Maybe
|
||
|
|
|
||
|
|
unitTests = testGroup "QuoteSource.Server" [testStartStop, testTickStream]
|
||
|
|
|
||
|
|
testStartStop = testCase "QuoteSource Server starts and stops" $ withContext (\ctx -> do
|
||
|
|
chan <- newBoundedChan 1000
|
||
|
|
qss <- startQuoteSourceServer chan ctx "inproc://quotesource-server"
|
||
|
|
stopQuoteSourceServer qss)
|
||
|
|
|
||
|
|
testTickStream = testCase "QuoteSource Server sends ticks" $ withContext (\ctx -> do
|
||
|
|
chan <- newBoundedChan 1000
|
||
|
|
bracket (startQuoteSourceServer chan ctx "inproc://quotesource-server") stopQuoteSourceServer (\qs ->
|
||
|
|
withSocket ctx Sub (\s -> do
|
||
|
|
connect s "inproc://quotesource-server"
|
||
|
|
subscribe s "FOOBAR"
|
||
|
|
let tick = Tick {
|
||
|
|
security = "FOOBAR",
|
||
|
|
datatype = Price,
|
||
|
|
timestamp = UTCTime (fromGregorian 2016 9 27) 16000,
|
||
|
|
value = 1000,
|
||
|
|
volume = 1}
|
||
|
|
tryWriteChan chan (Just tick)
|
||
|
|
packet <- fmap BL.fromStrict <$> receiveMulti s
|
||
|
|
case deserializeTick packet of
|
||
|
|
Just recvdTick -> tick @=? recvdTick
|
||
|
|
Nothing -> assertFailure "Unable to deserialize tick")))
|
||
|
|
|