generated from tpl/purs
104 lines
3.3 KiB
Haskell
104 lines
3.3 KiB
Haskell
module Test.Data.Postgres.Custom where
|
|
|
|
import Prelude
|
|
|
|
import Control.Monad.Error.Class (liftEither)
|
|
import Control.Monad.Except (runExceptT)
|
|
import Data.Bifunctor (lmap)
|
|
import Data.DateTime (DateTime(..))
|
|
import Data.Generic.Rep (class Generic)
|
|
import Data.Maybe (Maybe(..))
|
|
import Data.Newtype (unwrap)
|
|
import Data.Postgres (class Deserialize, class Serialize, deserialize, serialize, smash)
|
|
import Data.Postgres.Custom.Enum (class CustomEnum, create, enumDeserialize, enumPrintExpr, enumSerialize, genericEnumVariants, genericParseEnum, genericPrintEnum, parseEnum, printEnum)
|
|
import Data.Show.Generic (genericShow)
|
|
import Effect.Class (liftEffect)
|
|
import Test.Spec (Spec, describe, it)
|
|
import Test.Spec.Assertions (shouldEqual)
|
|
import Unsafe.Coerce (unsafeCoerce)
|
|
|
|
data Enum1 = E1A
|
|
|
|
derive instance Generic Enum1 _
|
|
derive instance Eq Enum1
|
|
instance Show Enum1 where
|
|
show = genericShow
|
|
|
|
instance Serialize Enum1 where
|
|
serialize a = enumSerialize a
|
|
|
|
instance Deserialize Enum1 where
|
|
deserialize a = enumDeserialize a
|
|
|
|
instance CustomEnum Enum1 "enum_1" where
|
|
printEnum = genericPrintEnum
|
|
parseEnum = genericParseEnum
|
|
enumVariants = genericEnumVariants
|
|
|
|
data Enum2 = E2A | E2B
|
|
|
|
derive instance Generic Enum2 _
|
|
derive instance Eq Enum2
|
|
instance Show Enum2 where
|
|
show = genericShow
|
|
|
|
instance Serialize Enum2 where
|
|
serialize a = enumSerialize a
|
|
|
|
instance Deserialize Enum2 where
|
|
deserialize a = enumDeserialize a
|
|
|
|
instance CustomEnum Enum2 "enum_2" where
|
|
printEnum a = genericPrintEnum a
|
|
parseEnum a = genericParseEnum a
|
|
enumVariants = genericEnumVariants
|
|
|
|
data Enum5 = E5A | E5B | E5C | E5D | E5E
|
|
|
|
derive instance Generic Enum5 _
|
|
derive instance Eq Enum5
|
|
instance Show Enum5 where
|
|
show = genericShow
|
|
|
|
instance Serialize Enum5 where
|
|
serialize a = enumSerialize a
|
|
|
|
instance Deserialize Enum5 where
|
|
deserialize a = enumDeserialize a
|
|
|
|
instance CustomEnum Enum5 "enum_5" where
|
|
printEnum a = genericPrintEnum a
|
|
parseEnum a = genericParseEnum a
|
|
enumVariants = genericEnumVariants
|
|
|
|
spec :: Spec Unit
|
|
spec =
|
|
describe "Test.Data.Postgres.Custom" do
|
|
describe "Enum" do
|
|
it "serialize" do
|
|
act <- liftEffect $ smash $ serialize E5A
|
|
exp <- liftEffect $ smash $ serialize "E5A"
|
|
act `shouldEqual` exp
|
|
it "deserialize" do
|
|
act <- liftEffect $ smash $ deserialize $ unsafeCoerce "E5A"
|
|
act `shouldEqual` E5A
|
|
it "create" do
|
|
(_.text $ unwrap $ create @Enum1) `shouldEqual` "create type enum_1 as enum ('E1A');"
|
|
(_.text $ unwrap $ create @Enum2) `shouldEqual` "create type enum_2 as enum ('E2A', 'E2B');"
|
|
(_.text $ unwrap $ create @Enum5) `shouldEqual` "create type enum_5 as enum ('E5A', 'E5B', 'E5C', 'E5D', 'E5E');"
|
|
it "parseEnum" do
|
|
parseEnum "E1A" `shouldEqual` Just E1A
|
|
parseEnum "E2A" `shouldEqual` Just E2A
|
|
parseEnum "E2B" `shouldEqual` Just E2B
|
|
parseEnum "E5B" `shouldEqual` Just E5B
|
|
it "printEnum" do
|
|
printEnum E1A `shouldEqual` "E1A"
|
|
printEnum E2A `shouldEqual` "E2A"
|
|
printEnum E2B `shouldEqual` "E2B"
|
|
printEnum E5D `shouldEqual` "E5D"
|
|
it "enumPrintExpr" do
|
|
enumPrintExpr E1A `shouldEqual` Just "'E1A' :: enum_1"
|
|
enumPrintExpr E2A `shouldEqual` Just "'E2A' :: enum_2"
|
|
enumPrintExpr E2B `shouldEqual` Just "'E2B' :: enum_2"
|
|
enumPrintExpr E5D `shouldEqual` Just "'E5D' :: enum_5"
|