Duane JohnsonandJoe Nelson
a46b6f5020
- allows queries to refer to current_setting('app.settings.foo') to retrieve variables
- useful for 12-factor apps (app data can be in environment)
- provides workaround for AWS Relational Database Service (RDS) not
allowing `ALTER DATABASE SET 'app.[KEY]' TO '[VALUE]'` on database.
117 lines
5.0 KiB
Haskell
117 lines
5.0 KiB
Haskell
module Main where
|
|
|
|
import Test.Hspec
|
|
import SpecHelper
|
|
|
|
import qualified Hasql.Pool as P
|
|
|
|
import PostgREST.App (postgrest)
|
|
import PostgREST.Config (pgVersion96, configSettings)
|
|
import PostgREST.DbStructure (getDbStructure, getPgVersion, fillSessionWithSettings)
|
|
import PostgREST.Types (DbStructure(..))
|
|
import Data.Function (id)
|
|
import Data.IORef
|
|
|
|
import qualified Feature.AuthSpec
|
|
import qualified Feature.AsymmetricJwtSpec
|
|
import qualified Feature.BinaryJwtSecretSpec
|
|
import qualified Feature.AudienceJwtSecretSpec
|
|
import qualified Feature.ConcurrentSpec
|
|
import qualified Feature.CorsSpec
|
|
import qualified Feature.DeleteSpec
|
|
import qualified Feature.InsertSpec
|
|
import qualified Feature.NoJwtSpec
|
|
import qualified Feature.QueryLimitedSpec
|
|
import qualified Feature.QuerySpec
|
|
import qualified Feature.RangeSpec
|
|
import qualified Feature.StructureSpec
|
|
import qualified Feature.SingularSpec
|
|
import qualified Feature.UnicodeSpec
|
|
import qualified Feature.ProxySpec
|
|
import qualified Feature.AndOrParamsSpec
|
|
import qualified Feature.RpcSpec
|
|
import qualified Feature.NonexistentSchemaSpec
|
|
import qualified Feature.PgVersion96Spec
|
|
|
|
import Protolude
|
|
|
|
main :: IO ()
|
|
main = do
|
|
testDbConn <- getEnvVarWithDefault "POSTGREST_TEST_CONNECTION" "postgres://postgrest_test@localhost/postgrest_test"
|
|
setupDb testDbConn
|
|
|
|
pool <- P.acquire (3, 10, toS testDbConn)
|
|
|
|
result <- P.use pool $ getDbStructure "test" =<< getPgVersion
|
|
|
|
dbStructure <- pure $ either (panic.show) id result
|
|
|
|
refDbStructure <- newIORef $ Just dbStructure
|
|
|
|
let withApp = return $ postgrest (testCfg testDbConn) refDbStructure pool $ pure ()
|
|
ltdApp = return $ postgrest (testLtdRowsCfg testDbConn) refDbStructure pool $ pure ()
|
|
unicodeApp = return $ postgrest (testUnicodeCfg testDbConn) refDbStructure pool $ pure ()
|
|
proxyApp = return $ postgrest (testProxyCfg testDbConn) refDbStructure pool $ pure ()
|
|
noJwtApp = return $ postgrest (testCfgNoJWT testDbConn) refDbStructure pool $ pure ()
|
|
binaryJwtApp = return $ postgrest (testCfgBinaryJWT testDbConn) refDbStructure pool $ pure ()
|
|
audJwtApp = return $ postgrest (testCfgAudienceJWT testDbConn) refDbStructure pool $ pure ()
|
|
asymJwkApp = return $ postgrest (testCfgAsymJWK testDbConn) refDbStructure pool $ pure ()
|
|
nonexistentSchemaApp = return $ postgrest (testNonexistentSchemaCfg testDbConn) refDbStructure pool $ pure ()
|
|
|
|
let reset :: IO ()
|
|
reset = P.use pool (fillSessionWithSettings (configSettings $ testCfg testDbConn)) >> resetDb testDbConn
|
|
|
|
actualPgVersion = pgVersion dbStructure
|
|
pg96spec | actualPgVersion >= pgVersion96 = [("Feature.PgVersion96Spec" , Feature.PgVersion96Spec.spec)]
|
|
| otherwise = []
|
|
|
|
specs = uncurry describe <$> [
|
|
("Feature.AuthSpec" , Feature.AuthSpec.spec)
|
|
, ("Feature.ConcurrentSpec" , Feature.ConcurrentSpec.spec)
|
|
, ("Feature.CorsSpec" , Feature.CorsSpec.spec)
|
|
, ("Feature.DeleteSpec" , Feature.DeleteSpec.spec)
|
|
, ("Feature.InsertSpec" , Feature.InsertSpec.spec)
|
|
, ("Feature.QuerySpec" , Feature.QuerySpec.spec)
|
|
, ("Feature.RpcSpec" , Feature.RpcSpec.spec)
|
|
, ("Feature.RangeSpec" , Feature.RangeSpec.spec)
|
|
, ("Feature.SingularSpec" , Feature.SingularSpec.spec)
|
|
, ("Feature.StructureSpec" , Feature.StructureSpec.spec)
|
|
, ("Feature.AndOrParamsSpec" , Feature.AndOrParamsSpec.spec)
|
|
, ("Feature.NonexistentSchemaSpec" , Feature.NonexistentSchemaSpec.spec)
|
|
] ++ pg96spec
|
|
|
|
hspec $ do
|
|
mapM_ (beforeAll_ reset . before withApp) specs
|
|
|
|
-- this test runs with a different server flag
|
|
beforeAll_ reset . before ltdApp $
|
|
describe "Feature.QueryLimitedSpec" Feature.QueryLimitedSpec.spec
|
|
|
|
-- this test runs with a different schema
|
|
beforeAll_ reset . before unicodeApp $
|
|
describe "Feature.UnicodeSpec" Feature.UnicodeSpec.spec
|
|
|
|
-- this test runs with a proxy
|
|
beforeAll_ reset . before proxyApp $
|
|
describe "Feature.ProxySpec" Feature.ProxySpec.spec
|
|
|
|
-- this test runs without a JWT secret
|
|
beforeAll_ reset . before noJwtApp $
|
|
describe "Feature.NoJwtSpec" Feature.NoJwtSpec.spec
|
|
|
|
-- this test runs with a binary JWT secret
|
|
beforeAll_ reset . before binaryJwtApp $
|
|
describe "Feature.BinaryJwtSecretSpec" Feature.BinaryJwtSecretSpec.spec
|
|
|
|
-- this test runs with a binary JWT secret and an audience claim
|
|
beforeAll_ reset . before audJwtApp $
|
|
describe "Feature.AudienceJwtSecretSpec" Feature.AudienceJwtSecretSpec.spec
|
|
|
|
-- this test runs with asymmetric JWK
|
|
beforeAll_ reset . before asymJwkApp $
|
|
describe "Feature.AsymmetricJwtSpec" Feature.AsymmetricJwtSpec.spec
|
|
|
|
-- this test runs with a nonexistent db-schema
|
|
beforeAll_ reset . before nonexistentSchemaApp $
|
|
describe "Feature.NonexistentSchemaSpec" Feature.NonexistentSchemaSpec.spec
|