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.
50 lines
1.6 KiB
50 lines
1.6 KiB
{-# 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"))) |
|
|
|
|