Implement FromField UTCTime

test/Database/PostgreSQL/Opium/FromFieldSpec.hs

@@ -4,7 +4,15 @@
 module Database.PostgreSQL.Opium.FromFieldSpec (spec) where
 
 import Data.ByteString (ByteString)
-import Data.Time (Day (..), DiffTime, TimeOfDay (..), fromGregorian, secondsToDiffTime)
+import Data.Time
+  ( Day (..)
+  , DiffTime
+  , TimeOfDay (..)
+  , UTCTime (..)
+  , fromGregorian
+  , secondsToDiffTime
+  , timeOfDayToTime
+  )
 import Data.Text (Text)
 import Database.PostgreSQL.LibPQ (Connection)
 import Database.PostgreSQL.Opium (FromRow)
@@ -93,6 +101,12 @@ newtype ATimeOfDay = ATimeOfDay
 
 instance FromRow ATimeOfDay where
 
+newtype AUTCTime = AUTCTime
+  { utctime :: UTCTime
+  } deriving (Eq, Generic, Show)
+
+instance FromRow AUTCTime where
+
 newtype ARawField = ARawField
   { raw :: Opium.RawField ByteString
   } deriving (Eq, Generic, Show)
@@ -247,6 +261,22 @@ spec = do
       shouldFetch conn "SELECT time '00:01:00' AS timeofday" [ATimeOfDay $ TimeOfDay 0 1 0]
       shouldFetch conn "SELECT time '13:07:43' AS timeofday" [ATimeOfDay $ TimeOfDay 13 7 43]
 
+  describe "FromField UTCTime" $ do
+    it "Decodes timestamp" $ \conn -> do
+      let ts0 = UTCTime (fromGregorian 2023 10 2) (timeOfDayToTime $ TimeOfDay 12 42 23)
+      shouldFetch conn "SELECT timestamp '2023-10-02 12:42:23' AS utctime" [AUTCTime ts0]
+
+      let ts1 = UTCTime (fromGregorian 294275 12 31) (timeOfDayToTime $ TimeOfDay 23 59 59)
+      shouldFetch conn "SELECT timestamp '294275-12-31 23:59:59' AS utctime" [AUTCTime ts1]
+
+      let ts2 = UTCTime (fromGregorian 1 1 1) (timeOfDayToTime $ TimeOfDay 0 0 0)
+      shouldFetch conn "SELECT timestamp '0001-01-01 00:00:00' AS utctime" [AUTCTime ts2]
+
+      {- TODO: How does "BC" work here?
+      let ts3 = UTCTime (fromGregorian 0 1 1) (timeOfDayToTime $ TimeOfDay 0 0 0)
+      shouldFetch conn "SELECT timestamp 'January 1, 1 BC 00:00:00' AS utctime" [AUTCTime ts3]
+      -}
+
   describe "FromField RawField" $ do
     it "Simply returns the bytestring without decoding it" $ \conn -> do
       shouldFetch conn "SELECT 'Hello, World!'::bytea AS raw" [ARawField $ Opium.RawField "Hello, World!"]