mirror of
https://github.com/unclechu/gRPC-haskell.git
synced 2024-12-26 19:59:45 +01:00
77 lines
2 KiB
Haskell
77 lines
2 KiB
Haskell
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module GeneratedTests where
|
|
|
|
import Test.Tasty
|
|
import Test.Tasty.HUnit (testCase, (@?=))
|
|
|
|
import Data.String
|
|
import Proto3.Suite.DotProto.Generate
|
|
|
|
import Turtle hiding (err)
|
|
|
|
generatedTests :: TestTree
|
|
generatedTests = testGroup "Code generator tests"
|
|
[ testServerGeneration
|
|
, testClientGeneration ]
|
|
|
|
testServerGeneration :: TestTree
|
|
testServerGeneration = testCase "server generation" $ do
|
|
mktree hsTmpDir
|
|
mktree pyTmpDir
|
|
|
|
compileDotProtoFileOrDie [] hsTmpDir ["tests"] "simple.proto"
|
|
|
|
do exitCode <- proc "tests/simple-server.sh" [hsTmpDir] empty
|
|
exitCode @?= ExitSuccess
|
|
|
|
do exitCode <- proc "tests/protoc.sh" [pyTmpDir] empty
|
|
exitCode @?= ExitSuccess
|
|
|
|
runManaged $ do
|
|
serverExitCodeA <- fork (shell (hsTmpDir <> "/simple-server") empty)
|
|
clientExitCodeA <- fork
|
|
(export "PYTHONPATH" pyTmpDir >> shell "tests/test-client.sh" empty)
|
|
|
|
liftIO $ do
|
|
serverExitCode <- liftIO (wait serverExitCodeA)
|
|
clientExitCode <- liftIO (wait clientExitCodeA)
|
|
|
|
serverExitCode @?= ExitSuccess
|
|
clientExitCode @?= ExitSuccess
|
|
|
|
rmtree hsTmpDir
|
|
rmtree pyTmpDir
|
|
|
|
testClientGeneration :: TestTree
|
|
testClientGeneration = testCase "client generation" $ do
|
|
mktree hsTmpDir
|
|
mktree pyTmpDir
|
|
|
|
compileDotProtoFileOrDie [] hsTmpDir ["tests"] "simple.proto"
|
|
|
|
do exitCode <- proc "tests/simple-client.sh" [hsTmpDir] empty
|
|
exitCode @?= ExitSuccess
|
|
|
|
do exitCode <- proc "tests/protoc.sh" [pyTmpDir] empty
|
|
exitCode @?= ExitSuccess
|
|
|
|
runManaged $ do
|
|
serverExitCodeA <- fork
|
|
(export "PYTHONPATH" pyTmpDir >> shell "tests/test-server.sh" empty)
|
|
clientExitCodeA <- fork (shell (hsTmpDir <> "/simple-client") empty)
|
|
|
|
liftIO $ do
|
|
serverExitCode <- liftIO (wait serverExitCodeA)
|
|
clientExitCode <- liftIO (wait clientExitCodeA)
|
|
|
|
serverExitCode @?= ExitSuccess
|
|
clientExitCode @?= ExitSuccess
|
|
|
|
rmtree hsTmpDir
|
|
rmtree pyTmpDir
|
|
|
|
hsTmpDir, pyTmpDir :: IsString a => a
|
|
hsTmpDir = "tests/tmp"
|
|
pyTmpDir = "tests/py-tmp"
|