{-# LANGUAGE GeneralizedNewtypeDeriving #-}
{-# LANGUAGE UndecidableInstances #-}
module CustomPrimaryKeyReferenceTest where
import Init
share [mkPersist persistSettings { mpsGeneric = False }, mkMigrate "migration"] [persistLowerCase|
Tweet
tweetId Int
statusText Text sqltype=varchar(170)
Primary tweetId
UniqueTweetId tweetId
deriving Show
TweetUrl
tweetId TweetId
tweetUrl Text sqltype=varchar(255)
finalUrl Text Maybe sqltype=varchar(255)
UniqueTweetIdTweetUrl tweetId tweetUrl
deriving Show
|]
cleanDB :: (MonadIO m, PersistQuery backend, PersistEntityBackend Tweet ~ backend) => ReaderT backend m ()
cleanDB = do
deleteWhere ([] :: [Filter Tweet])
deleteWhere ([] :: [Filter TweetUrl])
specsWith :: (MonadFail m, MonadIO m) => RunDb SqlBackend m -> Spec
specsWith runDb = describe "custom primary key reference" $ do
let tweet = Tweet {tweetTweetId = 1, tweetStatusText = "Hello!"}
it "can insert a Tweet" $ runDb $ do
tweetId <- insert tweet
let url = TweetUrl {tweetUrlTweetId = tweetId, tweetUrlTweetUrl = "http://google.com", tweetUrlFinalUrl = Just "http://example.com"}
insert_ url