{-# 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")))