--
-- Copyright 2018, akashche at redhat.com
--
-- Licensed under the Apache License, Version 2.0 (the "License");
-- you may not use this file except in compliance with the License.
-- You may obtain a copy of the License at
--
-- http://www.apache.org/licenses/LICENSE-2.0
--
-- Unless required by applicable law or agreed to in writing, software
-- distributed under the License is distributed on an "AS IS" BASIS,
-- WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
-- See the License for the specific language governing permissions and
-- limitations under the License.

{-# LANGUAGE DeriveGeneric #-}
{-# LANGUAGE DuplicateRecordFields #-}
{-# LANGUAGE OverloadedStrings #-}
{-# LANGUAGE ScopedTypeVariables #-}
{-# LANGUAGE Strict #-}

module HUnitTest ( hunitTest ) where

import Test.HUnit
import Prelude (Int, (+), ($), return)
import Data.IORef (newIORef, readIORef, writeIORef)
import Data.Vector (fromList)

import VtUtils.HUnit

testMain :: Test
testMain = TestLabel "testMain" $ TestCase $ do
    -- uses args, hard to test
    return ()

testRun :: Test
testRun = TestLabel "testRun" $ TestCase $ do
    ref <- newIORef (0 :: Int)
    let tst = TestCase $ do
            val <- readIORef ref
            _ <- writeIORef ref (val + 1)
            return ()
    _ <- hunitRun (fromList
        [ TestLabel "group1" (TestList
            [ TestLabel "test1" tst
            ])
        ])
    after <- readIORef ref
    assertEqual "test case called" 1 after
    return ()

testRunGroup :: Test
testRunGroup = TestLabel "testRunGroup" $ TestCase $ do
    ref <- newIORef (0 :: Int)
    before <- readIORef ref
    assertEqual "before" 0 before
    let tst = TestCase $ do
            val <- readIORef ref
            _ <- writeIORef ref (val + 1)
            return ()
    _ <- hunitRunGroup (fromList
        [ TestLabel "group1" (TestList
            [ TestLabel "test1" tst
            ])
        , TestLabel "group2" (TestList
            [ TestLabel "test1" tst
            ])
        ]) "group1"
    after <- readIORef ref
    assertEqual "specified group called" 1 after
    return ()

testRunSingle :: Test
testRunSingle = TestLabel "testRunSingle" $ TestCase $ do
    ref <- newIORef (0 :: Int)
    before <- readIORef ref
    assertEqual "before" 0 before
    let tst = TestCase $ do
            val <- readIORef ref
            _ <- writeIORef ref (val + 1)
            return ()
    _ <- hunitRunSingle (fromList
        [ TestLabel "group1" (TestList
            [ TestLabel "test1" tst
            , TestLabel "test2" tst
            ])
        ]) "group1" "test1"
    after <- readIORef ref
    assertEqual "specified test called" 1 after
    return ()

hunitTest :: Test
hunitTest = TestLabel "HUnitTest" (TestList
    [ testMain
    , testRun
    , testRunGroup
    , testRunSingle
    ])