-----------------------------------------------------------------------------
-- Copyright 2016, Ideas project team. This file is distributed under the
-- terms of the Apache License 2.0. For more information, see the files
-- "LICENSE.txt" and "NOTICE.txt", which are included in the distribution.
-----------------------------------------------------------------------------
-- |
-- Maintainer  :  bastiaan.heeren@ou.nl
-- Stability   :  provisional
-- Portability :  portable (depends on ghc)
--
-----------------------------------------------------------------------------

module Ideas.Common.Traversal.Tests
   ( testIterator, testNavigator, tests
   , uniGen, listGen
   ) where

import Control.Monad
import Data.Maybe
import Ideas.Common.Traversal.Iterator
import Ideas.Common.Traversal.Navigator
import Ideas.Common.Traversal.Utils
import Ideas.Utils.TestSuite
import Ideas.Utils.Uniplate
import Test.QuickCheck hiding ((===))

testIterator :: (Show a, Eq a, Iterator a) => String -> Gen a -> TestSuite
testIterator s gen = suite (s ++ " Iterator")
   [ suite "previous/next"
        [ prop gen "previous; next" $  hasPrevious ==>>  previous >=> next ==! id
        , prop gen "next; previous" $  hasNext     ==>>  next >=> previous ==! id
        ]

   , suite "next/final"
        [ prop gen "isFinal"       $  isFinal . final
        , prop gen "next to final" $  fixp next === final
        ]

   , suite "previous/first"
        [ prop gen "isFirst"           $  isFirst . first
        , prop gen "previous to first" $  fixp previous === first
        ]

   , suite "position"
        [ prop gen "pos previous" $
             hasPrevious ==>> fmap position . previous ==! pred . position
        , prop gen "pos next" $
             hasNext ==>> fmap position . next ==! succ . position
        , prop gen "pos first" $
             (==0) . position . first
        , prop gen "pos final" $
             position . final === position . fixp next
        ]
   ]

testNavigator :: (Show a, Eq a, Navigator a) => String -> Gen a -> TestSuite
testNavigator s gen = suite (s ++ " Navigator")
   [ suite "up/down"
        [ prop gen "down; up"     $  hasDown ==>>      down >=> up ==! id
        , prop gen "up; down"     $  hasUp   ==>>      up >=> down ==! leftMost
        , prop gen "up; downLast" $  hasUp   ==>>  up >=> downLast ==! rightMost
        ]

   , suite "left/right"
        [ prop gen "right; left" $  hasRight ==>>  right >=> left ==! id
        , prop gen "left; right" $  hasLeft  ==>>  left >=> right ==! id
        ]

   , suite "up/left+right"
        [ prop gen "left; up"  $  hasLeft  ==>>   left >=> up === up
        , prop gen "right; up" $  hasRight ==>>  right >=> up === up
        ]

   , suite "down/downLast"
        [ prop gen "down; rightMost"       $  fmap rightMost . down === downLast
        , prop gen "downLast; leftMost"    $  fmap leftMost . downLast === down
        , prop gen "down is leftMost"      $  isNothing . (down >=> left)
        , prop gen "downLast is rightMost" $  isNothing . (downLast >=> right)
        ]

   , suite "location"
        [ prop gen "loc up" $ hasUp    ==>>
             fmap locationList . up ==! init . locationList
        , prop gen "loc down" $ hasDown  ==>>
             fmap locationList . down ==! (++[0]) . locationList
        , prop gen "loc downLast" $ hasDown  ==>>
             fmap locationList . downLast ==! (\a -> locationList a ++ [arity a-1])
        , prop gen "loc left" $ hasLeft  ==>>
             fmap locationList . left ==! changeLast pred . locationList
        , prop gen "loc right" $ hasRight ==>>
             fmap locationList . right ==! changeLast succ . locationList
        , prop gen "childnr" $
             childnr === fromMaybe 0 . listToMaybe . reverse . locationList
        ]
   ]

locationList :: Navigator a => a -> [Int]
locationList = fromLocation . location

-------------------------------------------------------------------------
-- tests

tests :: TestSuite
tests =
   suite "Iterators"
      [ testIterator "List" listGen
      , testIterator "Mirror"     $ makeMirror     <$> listGen
      , testIterator "Leafs"      $ makeLeafs      <$> uniGen
      , testIterator "PreOrder"   $ makePreOrder   <$> uniGen
      , testIterator "PostOrder"  $ makePostOrder  <$> uniGen
      , testIterator "Horizontal" $ makeHorizontal <$> uniGen
      , testIterator "LevelOrder" $ makeLevelOrder <$> uniGen
      ] <>
   suite "Navigators"
      [ testNavigator "Uniplate" uniGen
      , testNavigator "Mirror" $ makeMirror <$> uniGen
      ]

_go :: IO ()
_go = runTestSuiteResult True tests >>= print

-------------------------------------------------------------------------
-- test utils

infixr 0 ===, ==!

(===) :: Eq b => (a -> b) -> (a -> b) -> a -> Bool
(f === g) a = f a == g a

(==!) :: Eq b => (a -> Maybe b) -> (a -> b) -> a -> Bool
f ==! g = f === Just . g

infixr 0 ==>>

(==>>) :: Testable prop => (a -> Bool) -> (a -> prop) -> a -> Property
(p ==>> f) a = p a ==> f a

prop :: (Testable prop, Show a) => Gen a -> String -> (a -> prop) -> TestSuite
prop gen s = useProperty s . forAll gen

changeLast :: (a -> a) -> [a] -> [a]
changeLast _ []     = []
changeLast f [x]    = [f x]
changeLast f (x:xs) = x:changeLast f xs

data T a = T a [T a] deriving (Show, Eq)

instance Uniplate (T a) where
   uniplate (T a xs) = plate (T a) ||* xs

instance Arbitrary a => Arbitrary (T a) where
   arbitrary = sized genT
    where
      genT n = do
         a  <- arbitrary
         i  <- if n==0 then return 0 else choose (0, 5)
         xs <- vectorOf i (genT (n `div` 2))
         return (T a xs)

listGen :: Gen (ListIterator Int)
listGen = arbitrary

uniGen :: Gen (UniplateNavigator (T Int))
uniGen = arbitrary