module Test.FIT.FixtureTest where

import Test.HUnit
import Test.FIT.Fixture
import Test.FIT.Parse
import Test.FIT.LoadFixture
import Data.HashTable as Hash

fixtureTests = TestList
  [ testTraverseOneTable
  , testTraverseTwoRows ]


testTraverseTwoRows = TestCase $ do
  let
    inputHtml =
      -- we need at least three rows in the fixture
      -- because the first two are skipped
      "<table>"
      ++ "<tr><td>Test.FIT.Fixture</td><td>two</td><td>three</td></tr>"
      ++ "<tr><td>one</td><td>two</td><td>three</td></tr>"
      ++ "<tr><td>one</td><td>two</td><td>three</td></tr>"
      ++ "</table>"
      ++ "<table>"
      ++ "<tr><td>Test.FIT.Fixture</td><td>two</td><td>three</td></tr>"
      ++ "<tr><td>one</td><td>two</td><td>three</td></tr>"
      ++ "<tr><td>one</td><td>two</td><td>three</td></tr>"
      ++ "</table>"
    expectHtml =
      "<table>"
      ++ "<tr><td>Test.FIT.Fixture</td><td>two</td><td>three</td></tr>"
      ++ "<tr><td>one</td><td>two</td><td>three</td></tr>"
      ++ "<tr><td bgcolor=\"#efefef\">one</td>"
      ++ "<td bgcolor=\"#efefef\">two</td>"
      ++ "<td bgcolor=\"#efefef\">three</td></tr>"
      ++ "</table>"
      ++ "<table>"
      ++ "<tr><td>Test.FIT.Fixture</td><td>two</td><td>three</td></tr>"
      ++ "<tr><td>one</td><td>two</td><td>three</td></tr>"
      ++ "<tr><td bgcolor=\"#efefef\">one</td>"
      ++ "<td bgcolor=\"#efefef\">two</td>"
      ++ "<td bgcolor=\"#efefef\">three</td></tr>"
      ++ "</table>"
  case parse inputHtml ["table", "tr", "td"] 0 0 of
    Left e -> assertFailure e
    Right p -> do
      summary <- Hash.new (==) hashString
      p' <- doTables p summary "" -- >>= return . fixtureParse
      assertEqual "traverse" expectHtml (printParse p')
      return ()

testTraverseOneTable = TestCase $ do
  let
    inputHtml =
      "<table>"
      ++ "<tr><td>Test.FIT.Fixture</td><td>two</td><td>three</td></tr>"
      -- ++ "<tr><td>one</td><td>two</td><td>three</td></tr>"
      ++ "</table>"
    expectHtml =
      "<table>"
      ++ "<tr><td>Test.FIT.Fixture</td><td>two</td><td>three</td></tr>"
      -- ++ "<tr><td>one</td><td>two</td><td>three</td></tr>"
      ++ "</table>"
  case parse inputHtml ["table", "tr", "td"] 0 0 of
    Left e -> assertFailure e
    Right p -> do
      summary <- Hash.new (==) hashString
      p' <- doTables p summary "" -- >>= return . fixtureParse
      assertEqual "traverse" expectHtml (printParse p')
      return ()