--------------------------------------------------------------------------------
module Hakyll.Images.CompressJpg.Tests
  ( tests,
  )
where

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

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

import Control.Exception (ErrorCall, catch, evaluate)
import qualified Data.ByteString as B
import Hakyll.Images.Common
import Hakyll.Images.CompressJpg
import Test.Tasty (TestTree, testGroup)
import Test.Tasty.HUnit (Assertion, assertBool, assertFailure, testCase)
import Text.Printf (printf)

testJpg :: IO Image
testJpg = Image Jpeg <$> (B.readFile "tests/data/piccolo.jpg")

fromAssertions ::
  -- | Name
  String ->
  -- | Cases
  [Assertion] ->
  -- | Result tests
  [TestTree]
fromAssertions name =
  zipWith testCase [printf "[%2d] %s" n name | n <- [1 :: Int ..]]

-- Test that the standard Image compressed to quality 25/100 is smaller
-- than the initial image
testCompressionFromImage :: Assertion
testCompressionFromImage = do
  im <- testJpg
  let initialSize = (B.length . image) im
      finalSize = (B.length . image . compressJpg 25) im

  assertBool "Image was not compressed" (initialSize > finalSize)

-- Test that specifying a JPG encoding below 0 will fail
testJpgEncodingOutOfLowerBound :: Assertion
testJpgEncodingOutOfLowerBound = do
  im <- testJpg
  -- Catching exceptions is an idea from here:
  -- https://stackoverflow.com/questions/46330592/is-it-possible-to-assert-an-error-case-in-hunit
  -- Since compressJpg is a "pure" function, we need to evaluate it in an IO context
  -- to catch errors.
  errored <- catch (evaluate (compressJpg (-10) im) >> pure False) handler
  if errored
    then pure ()
    else assertFailure "did not catch expected error"
  where
    handler :: ErrorCall -> IO Bool
    handler _ = pure True

-- Test that specifying a JPG encoding above 100 will fail
testJpgEncodingOutOfUpperBound :: Assertion
testJpgEncodingOutOfUpperBound = do
  im <- testJpg
  -- Catching exceptions is an idea from here:
  -- https://stackoverflow.com/questions/46330592/is-it-possible-to-assert-an-error-case-in-hunit
  -- Since compressJpg is a "pure" function, we need to evaluate it in an IO context
  -- to catch errors.
  errored <- catch (evaluate (compressJpg 111 im) >> pure False) handler
  if errored
    then pure ()
    else assertFailure "did not catch expected error"
  where
    handler :: ErrorCall -> IO Bool
    handler _ = pure True

--------------------------------------------------------------------------------
tests :: TestTree
tests =
  testGroup "Hakyll.Images.CompressJpg.Tests" $
    concat
      [ fromAssertions
          "compressJpg"
          [ testCompressionFromImage,
            testJpgEncodingOutOfLowerBound,
            testJpgEncodingOutOfUpperBound
          ]
      ]