serversession-backend-persistent-2.0.4: tests/Main.hs

{-# LANGUAGE TypeApplications #-}

module Main (main) where

import Control.Monad (forM_)
import Control.Monad.Logger (runStderrLoggingT, runNoLoggingT)
import Data.Pool (destroyAllResources)
import Database.Persist.Postgresql (createPostgresqlPool)
import Database.Persist.Sqlite (createSqlitePool)
import Test.Hspec
import Web.ServerSession.Backend.Persistent
import Web.ServerSession.Core.StorageTests
import Database.Persist.Sql.Migration (Migration)
import Data.Proxy (Proxy(..))
import Database.Persist.TH (migrateModels)

import qualified Database.Persist.Sql as P

migrateAll :: Migration
migrateAll = migrateModels $ [(P.entityDef (Proxy @PersistentSessionBySessionMap))]

main :: IO ()
main = hspec $
  forM_ [ ("PostgreSQL", createPostgresqlPool "host=localhost port=5432 user=test dbname=test password=test" 20)
        , ("SQLite",     createSqlitePool "test.db" 1) ] $
    \(rdbms, createPool) ->
  describe ("SqlStorage on " ++ rdbms) $ do
    pool <-
      runIO $ do
        pool <- runNoLoggingT createPool
        runStderrLoggingT $ P.runSqlPool (P.runMigration migrateAll) pool
        return pool
    afterAll_ (destroyAllResources pool) $
      allStorageTests (SqlStorage pool) it runIO parallel shouldBe shouldReturn shouldThrow