TimestampSpec.hs

module PostgresqlTypes.TimestampSpec (spec) where

import Data.Data (Proxy (Proxy))
import qualified Data.Time as Time
import qualified PostgresqlTypes.Timestamp as Timestamp
import Test.Hspec
import Test.QuickCheck
import qualified UnitTests.Scripts as Scripts

spec :: Spec
spec = do
  describe "Show/Read laws" do
    Scripts.testShowRead (Proxy @Timestamp.Timestamp)

  describe "IsScalar laws" do
    Scripts.testIsScalar (Proxy @Timestamp.Timestamp)

  describe "Constructors" do
    describe "normalizeFromLocalTime" do
      it "creates Timestamp from LocalTime" do
        let day = Time.fromGregorian 2023 6 15
            tod = Time.TimeOfDay 12 30 45
            localTime = Time.LocalTime day tod
            pgTimestamp = Timestamp.normalizeFromLocalTime localTime
        Timestamp.toLocalTime pgTimestamp `shouldBe` localTime

      it "handles epoch" do
        let day = Time.fromGregorian 2000 1 1
            tod = Time.TimeOfDay 0 0 0
            localTime = Time.LocalTime day tod
            pgTimestamp = Timestamp.normalizeFromLocalTime localTime
        Timestamp.toLocalTime pgTimestamp `shouldBe` localTime

  describe "Accessors" do
    describe "toLocalTime" do
      it "extracts LocalTime value" do
        let day = Time.fromGregorian 2023 6 15
            tod = Time.TimeOfDay 12 30 45
            localTime = Time.LocalTime day tod
            pgTimestamp = Timestamp.normalizeFromLocalTime localTime
        Timestamp.toLocalTime pgTimestamp `shouldBe` localTime

  describe "Property Tests" do
    it "roundtrips through toLocalTime and normalizeFromLocalTime" do
      property \(pgTimestamp :: Timestamp.Timestamp) ->
        let localTime = Timestamp.toLocalTime pgTimestamp
            pgTimestamp' = Timestamp.normalizeFromLocalTime localTime
         in pgTimestamp' === pgTimestamp

    it "roundtrips through normalizeFromLocalTime and toLocalTime for valid LocalTimes" do
      property \(pgTimestamp :: Timestamp.Timestamp) ->
        let localTime = Timestamp.toLocalTime pgTimestamp
            restored = Timestamp.normalizeFromLocalTime localTime
         in Timestamp.toLocalTime restored === localTime