courier-0.1.1.5: tests/Tests.hs

module Main where

-- local imports

import TestUtils

-- external imports

import System.IO
import System.Log.Formatter
import System.Log.Handler (setFormatter)
import System.Log.Handler.Simple
import System.Log.Logger

import Test.Framework

-- Test modules
import qualified TestMailbox as MB
import qualified TestMemory as M
import qualified TestRPC as R
import qualified TestRPCTyped as RT
import qualified TestTCP as T
import qualified TestUDP as U

-----------------------------------------------------------------------------
-----------------------------------------------------------------------------

main :: IO ()
main = do
  initLogging
  ipv6 <- isIPv6Available
  t4 <- T.tests4
  t6 <- T.tests6
  u4 <- U.tests4
  u6 <- U.tests6
  let allTests = tests ++ t4 ++ u4
  if ipv6
    then defaultMain (allTests ++ t6 ++ u6)
    else defaultMain allTests

initLogging :: IO ()
initLogging = do
  s <- streamHandler stdout INFO
  let fs = setFormatter s $ simpleLogFormatter "$time [$prio] - $msg"
  updateGlobalLogger rootLoggerName (setLevel WARNING)
  updateGlobalLogger rootLoggerName $ setHandlers [fs]

tests :: [Test.Framework.Test]
tests =
  MB.tests
  ++ M.tests
  ++ R.tests
  ++ RT.tests