From f3f5166d8bd5cacb7cf01ac44a57f48dd50752a9 Mon Sep 17 00:00:00 2001
From: Felipe Lessa <felipe.lessa@gmail.com>
Date: Tue, 28 Apr 2015 00:15:46 -0300
Subject: [PATCH] Update cleanDB with current entities.

---
 test/Test.hs | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/test/Test.hs b/test/Test.hs
index 96979cd..ad4beb7 100644
--- a/test/Test.hs
+++ b/test/Test.hs
@@ -1197,9 +1197,18 @@ cleanDB
   :: (forall m. RunDbMonad m
   => SqlPersistT (R.ResourceT m) ())
 cleanDB = do
-  delete $ from $ \(blogpost :: SqlExpr (Entity BlogPost))-> return ()
-  delete $ from $ \(follow   :: SqlExpr (Entity Follow))  -> return ()
-  delete $ from $ \(person   :: SqlExpr (Entity Person))  -> return ()
+  delete $ from $ \(_ :: SqlExpr (Entity BlogPost))   -> return ()
+  delete $ from $ \(_ :: SqlExpr (Entity Follow))     -> return ()
+  delete $ from $ \(_ :: SqlExpr (Entity Person))     -> return ()
+
+  delete $ from $ \(_ :: SqlExpr (Entity ArticleTag)) -> return ()
+  delete $ from $ \(_ :: SqlExpr (Entity Article))    -> return ()
+  delete $ from $ \(_ :: SqlExpr (Entity Article2))   -> return ()
+  delete $ from $ \(_ :: SqlExpr (Entity Tag))        -> return ()
+  delete $ from $ \(_ :: SqlExpr (Entity Frontcover)) -> return ()
+
+  delete $ from $ \(_ :: SqlExpr (Entity Circle))     -> return ()
+  delete $ from $ \(_ :: SqlExpr (Entity Point))      -> return ()
 #endif
 
 
@@ -1233,7 +1242,7 @@ run_worker act =
 #endif
   runSqlConn .
 #if defined (WITH_POSTGRESQL) || defined (WITH_MYSQL)
-  (runMigrationSilent migrateAll >>) $ (cleanDB >> act)
+  (runMigration migrateAll >>) $ (cleanDB >> act)
 #else
   (runMigrationSilent migrateAll >>) $ act
 #endif