hspec-pg-transact (empty) → 0.1.0.0

raw patch · 5 files changed

+115/−0 lines, 5 filesdep +basedep +bytestringdep +hspecsetup-changed

Dependencies added: base, bytestring, hspec, pg-transact, postgresql-simple, resource-pool, text, tmp-postgres

Files

• LICENSE +30/−0
• README.md +1/−0
• Setup.hs +2/−0
• hspec-pg-transact.cabal +32/−0
• src/Test/Hspec/DB.hs +50/−0

@@ -0,0 +1,30 @@+Copyright Author name here (c) 2017++All rights reserved.++Redistribution and use in source and binary forms, with or without+modification, are permitted provided that the following conditions are met:++    * Redistributions of source code must retain the above copyright+      notice, this list of conditions and the following disclaimer.++    * Redistributions in binary form must reproduce the above+      copyright notice, this list of conditions and the following+      disclaimer in the documentation and/or other materials provided+      with the distribution.++    * Neither the name of Author name here nor the names of other+      contributors may be used to endorse or promote products derived+      from this software without specific prior written permission.++THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS+"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT+LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR+A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT+OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,+SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT+LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,+DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY+THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE+OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

@@ -0,0 +1,1 @@+# hspec-pg-transact

@@ -0,0 +1,2 @@+import Distribution.Simple+main = defaultMain

@@ -0,0 +1,32 @@+name:                hspec-pg-transact+version:             0.1.0.0+synopsis: Helpers for creating database tests with hspec and pg-transact+description: Helpers for creating database tests with hspec and pg-transact+homepage:            https://github.com/jfischoff/hspec-pg-transact#readme+license:             BSD3+license-file:        LICENSE+author:              Jonathan Fischoff+maintainer:          jonathangfischoff@gmail.com+copyright:           2017 Jonathan Fischoff+category:            Web+build-type:          Simple+extra-source-files:  README.md+cabal-version:       >=1.10++library+  hs-source-dirs:      src+  exposed-modules: Test.Hspec.DB+  build-depends: base >= 4.7 && < 5+               , hspec+               , tmp-postgres+               , postgresql-simple+               , pg-transact+               , bytestring+               , text+               , resource-pool+  default-language:    Haskell2010+  ghc-options: -Wall -Wno-unused-do-bind++source-repository head+  type:     git+  location: https://github.com/githubuser/hspec-pg-transact

@@ -0,0 +1,50 @@+{-# LANGUAGE RecordWildCards #-}+module Test.Hspec.DB where+import           Control.Exception+import           Control.Monad+import qualified Data.ByteString.Char8        as BSC+import           Data.Pool+import qualified Database.Postgres.Temp       as Temp+import           Database.PostgreSQL.Simple+import           Database.PostgreSQL.Transact+import           Test.Hspec++data TestDB = TestDB+  { tempDB     :: Temp.DB+  , pool :: Pool Connection+  }++setupDB :: (Connection -> IO ()) -> IO TestDB+setupDB migrate = do+  tempDB     <- either throwIO return =<< Temp.startAndLogToTmp []+  putStrLn $ Temp.connectionString tempDB+  pool <- createPool+    (connectPostgreSQL (BSC.pack $ Temp.connectionString tempDB))+    close+    1+    100000000+    50+  withResource pool migrate+  return TestDB {..}++teardownDB :: TestDB -> IO ()+teardownDB TestDB {..} = do+  destroyAllResources pool+  void $ Temp.stop tempDB++withPool :: TestDB -> (Connection -> IO a) -> IO a+withPool testDB = withResource (pool testDB)++withDB :: DB a -> TestDB -> IO a+withDB action testDB =+  withResource (pool testDB) (runDBTSerializable action)++runDB :: TestDB -> DB a -> IO a+runDB = flip withDB++itDB :: String -> DB a -> SpecWith TestDB+itDB msg action = it msg $ void . withDB action++describeDB :: (Connection -> IO ()) -> String -> SpecWith TestDB -> Spec+describeDB migrate str =+  beforeAll (setupDB migrate) . afterAll teardownDB . describe str