{-# LANGUAGE OverloadedStrings #-}
module Main (main) where

import Control.Concurrent
import Control.Exception
import qualified Data.ByteString as B
import Data.BEncode
import Data.Map
import System.Environment
import System.Process
import System.FilePath

import Test.HUnit hiding (Test)
import Test.Framework
import Test.Framework.Providers.HUnit

import Network.KRPC
import Shared


addr :: RemoteAddr
addr = (0, 6000)

withServ :: FilePath -> IO () -> IO ()
withServ serv_path = bracket up terminateProcess . const
  where
    up = do
      (_, _, _, h) <- createProcess (proc serv_path [])
      threadDelay 1000000
      return h

main :: IO ()
main = do
  let serv_path = "dist" </> "build" </> "test-server" </> "test-server"
  withServ serv_path $
    defaultMain tests


(==?) :: (Eq a, Show a) => a -> IO a -> Assertion
expected ==? action = do
  actual <- action
  expected @=? actual

tests :: [Test]
tests =
  [ testCase "unit" $
      () ==? call addr unitM ()

  , testCase "echo int" $
      1234 ==? call addr echoM 1234

  , testCase "reverse 1..100" $
      reverse [1..100] ==? call addr reverseM [1..100]

  , testCase "reverse empty list" $
      reverse [] ==? call addr reverseM []

  , testCase "reverse singleton list" $
      reverse [1] ==? call addr reverseM [1]

  , testCase "swap pair" $
      (1, 0) ==? call addr swapM (0, 1)

  , testCase "shift triple" $
      ([2..10], (), 1) ==? call addr shiftR ((), 1, [2..10])

  , testCase "echo bytestring" $
      let bs = B.replicate 400 0 in
      bs ==? call addr echoBytes bs

  , testCase "raw method" $
      BInteger 10 ==? call addr rawM (BInteger 10)

  , testCase "raw dict" $
      let dict = BDict $ fromList
                 [ ("some_int", BInteger 100)
                 , ("some_list", BList [BInteger 10])
                 ]
      in dict ==? call addr rawDictM dict
  ]