module Main (main) where import Data.Time.Calendar.BankHoliday.Germany import Data.Time.Calendar.BankHoliday.Germany.ExtraHolidays qualified as EH import Data.Time.Calendar.WeekDate import Data.Time import qualified Hedgehog.Gen as Gen import qualified Hedgehog.Range as Range import Test.Hspec (describe, hspec, it, shouldBe) import Test.Hspec.Hedgehog (forAll, hedgehog, (===)) import Test.DocTest day :: Year -> Int -> Int -> Day day = fromGregorian year :: Day -> Year year = (\(y, _, _) -> y) . toGregorian main :: IO () main = do doctest ["src/"] hspec $ do describe "Data.Time.Calendar.BankHoliday.Germany" $ do describe "holidaysBetween" $ do it "works for Christmas" $ holidaysBetween (day 2024 12 1) (day 2024 12 30) `shouldBe` [(day 2024 12 24, ChristmasEve), (day 2024 12 25, ChristmasDay), (day 2024 12 26, SecondChristmasDay)] it "is empty list when there are no holidays" $ holidaysBetween (day 2024 12 1) (day 2024 12 3) `shouldBe` [] it "counts 11 bank holidays and 9 public holidays per year (2010 to 2020)" $ hedgehog $ do y <- forAll $ Gen.integral (Range.linear 2010 2020) let bankHolidays = holidaysBetween (day y 1 1) (day y 12 31) length bankHolidays === 11 length (filter (isPublicHoliday . snd) bankHolidays) === 9 describe "toDay" $ do it "always returns a day in the given year" $ hedgehog $ do y <- forAll $ Gen.integral (Range.linear 0 5000) d <- forAll Gen.enumBounded year (toDay y d) === y it "Easter Monday is always a Monday" $ hedgehog $ do y <- forAll $ Gen.integral (Range.linear 0 5000) let (_, _, d) = toWeekDate $ toDay y EasterMonday d === 1 -- 1 = Monday describe "calculateEasterSunday" $ do it "is always between Mar 22 and Apr 25" $ hedgehog $ do y <- forAll $ Gen.integral (Range.linear 0 10000) (calculateEasterSunday y >= day y 3 22) === True (calculateEasterSunday y <= day y 4 25) === True it "is correct for some years" $ do calculateEasterSunday 2024 `shouldBe` day 2024 3 31 calculateEasterSunday 2025 `shouldBe` day 2025 4 20 calculateEasterSunday 2026 `shouldBe` day 2026 4 5 it "is always a Sunday" $ hedgehog $ do y <- forAll $ Gen.integral (Range.linear 0 5000) let (_, _, d) = toWeekDate $ calculateEasterSunday y d === 7 -- 7 = Sunday describe "isPublicHoliday" $ do it "is False only for Chrismas Eve and New Year's Eve" $ length (filter (not . isPublicHoliday) [minBound..maxBound]) `shouldBe` 2 describe "yearFromDay" $ do it "works for any year" $ hedgehog $ do y <- forAll $ Gen.integral (Range.linear 0 5000) m <- forAll $ Gen.integral (Range.linear 1 12) d <- forAll $ Gen.integral (Range.linear 1 28) yearFromDay (day y m d) === y describe "Data.Time.Calendar.BankHoliday.Germany.ExtraHolidays" $ do describe "FederalState" $ it "has right number of states" $ length [minBound .. maxBound :: EH.FederalState] `shouldBe` 16 describe "toDay" $ it "works for Fronleichnam (depending on Easter Sunday)" $ do -- https://www.arbeitstage.org/feiertage/fronleichnam/ map (\y -> show $ EH.toDay y EH.Fronleichnam) [2024..2027] `shouldBe` ["2024-05-30", "2025-06-19", "2026-06-04", "2027-05-27"] describe "holidaysBetween" $ do it "only has Bavaria's extra holidays" $ map snd (EH.holidaysBetween EH.Bayern (day 2024 11 1) (day 2024 12 31)) `shouldBe` [EH.Allerheiligen] it "has no holidays for other states yet => otherwise, please add tests" $ do let statesExceptBavaria = filter (/=EH.Bayern) [minBound..maxBound] let holidays = concatMap (\x -> EH.holidaysBetween x (day 2024 1 1) (day 2024 12 31)) statesExceptBavaria holidays `shouldBe` []