git.iko.soy/rel8
on fce7213 help sign in
rel8/Doctest.hs
33 lines 1.5 kB history raw
1{-# language BlockArguments #-}
2{-# language OverloadedStrings #-}
4module Main where
6import Build_doctests ( flags, pkgs, module_sources )
7import Control.Exception ( bracket, throwIO )
8import Data.ByteString.Char8 ( unpack )
9import Data.Foldable ( traverse_ )
10import Database.PostgreSQL.Simple ( connectPostgreSQL, close, execute_ )
11import Database.Postgres.Temp ( toConnectionString, with, withConfig, verboseConfig )
12import System.Environment ( setEnv )
13import System.Environment.Compat ( unsetEnv )
14import Test.DocTest ( doctest )
16main :: IO ()
17main = do
18 unsetEnv "GHC_ENVIRONMENT" -- see 'Notes'; you may not need this
19 either throwIO return =<< with \db -> do
20 setEnv "TEST_DATABASE_URL" (unpack (toConnectionString db))
21 bracket (connectPostgreSQL (toConnectionString db)) close \conn -> do
22 execute_ conn "create table author ( author_id serial primary key, name text not null, url text )"
23 execute_ conn "create table project ( author_id int not null references author (author_id), name text not null )"
25 execute_ conn "insert into author ( name, url ) values ( 'Ollie', 'https://ocharles.org.uk' )"
26 execute_ conn "insert into author ( name, url ) values ( 'Bryan O''Sullivan', null )"
27 execute_ conn "insert into project ( author_id, name ) values ( 1, 'rel8' )"
28 execute_ conn "insert into project ( author_id, name ) values ( 2, 'aeson' )"
30 doctest args
32 where
33 args = flags ++ pkgs ++ module_sources