aboutsummaryrefslogtreecommitdiffstats
path: root/test/TestUtils.hs
blob: 672f8017e7f5c2f99af2d16716f8a7267ee1b778 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
{-# OPTIONS_GHC -Wno-missing-export-lists #-}
module TestUtils where

import Utils

testConfig :: Config
testConfig = Config {
    configScriptFileName = Nothing,
    configVerboseMode = False,
    configShowHelp = False,
    configPrintEvaled = PrintEvaledOff,
    configPrintCallStack = False,
    configUseREPL = False
}

testRunL :: Env -> LContext a -> IO (Either LException (a, LState))
testRunL env = runL LState {
    stateConfig = testConfig,
    stateEnv = env,
    stateDepth = 0,
    statePure = Impure
}

expectSuccessL :: Env -> LContext a -> IO (a, LState)
expectSuccessL env lc =
     do res <- testRunL env lc
        case res of
            Left (LException _ err) -> error $ "unexpected error: " ++ err
            Right val -> return val

expectErrorL :: Show a => Env -> LContext a -> IO String
expectErrorL env lc =
     do res <- testRunL env lc
        case res of
            Left (LException _ err) -> return err
            Right (val, _) -> error $ "unexpected success: " ++ show val

ast :: ASTNode -> AST
ast node = AST { astNode = node }

astInteger a = ast $ ASTInteger a
astDouble a = ast $ ASTDouble a
astSymbol a = ast $ ASTSymbol a
astBoolean a = ast $ ASTBoolean a
astString a = ast $ ASTString a
astVector a = ast $ ASTVector a
astFunctionCall a = ast $ ASTFunctionCall a
astHashMap a = ast $ ASTHashMap a
astUnit = ast $ ASTUnit
astHole = ast $ ASTHole