Added failing test case for custom primary keys (prowdsponsor#87)

meteogrid · Dec 23, 2014 · e22f232 · e22f232
1 parent abf91a1
commit e22f232
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/test/Test.hs b/test/Test.hs
@@ -59,6 +59,10 @@ share [mkPersist sqlSettings, mkMigrate "migrateAll"] [persistUpperCase|
     follower PersonId
     followed PersonId
     deriving Eq Show
+  Frontcover
+    number Int
+    Primary number
+    deriving Eq Show
 |]
 
 -- | this could be achieved with S.fromList, but not all lists
@@ -909,6 +913,16 @@ main = do
 
           liftIO $ ret `shouldBe` [ Value (3) ]
 
+      it "works with custom primary key" $
+        run $ do
+          let fc = Frontcover number
+              number = 101
+              Right thePk = keyFromValues [PersistInt64 $ fromIntegral number]
+          fcPk <- insert fc
+          [Entity _ ret] <- select $ from $ return
+          liftIO $ do
+            ret `shouldBe` fc
+            fcPk `shouldBe` thePk
 
 ----------------------------------------------------------------------