@@ -1,16 +1,15 @@
11module Test.Compiler (runTests) where
22
3−import Prelude (Unit, ($), bind, discard, show)
4−import Debug
3+import Prelude (Unit, bind, discard, show)
54import Data.Semigroup ((<>))
65import Data.Foldable (foldMap)
76import Data.Tuple (Tuple(Tuple))
8−import Effect
7+import Effect (Effect)
98import Test.Unit (suite, test)
109import Test.Unit.Main (runTest)
1110import Test.Unit.Assert (equal)
1211import Test.Compiler.ServerMock (settings, setupSrv)
13−import Compiler (Code, runCompiler)
12+import Compiler (Code, runCompiler, appendMain)
1413
1514mkImport :: String -> String -> Tuple Code Code
1615mkImport url mod = let
@@ -24,9 +23,10 @@ runTests :: Effect Unit
2423runTests = runTest do
2524 suite "compile" do
2625 test "produces expected output" do
27− let expected = "compiled-code"
26+ let srv = "compiled-code"
2827 compile = runCompiler settings
29− actual <- setupSrv expected (compile "code")
28+ expected = appendMain srv
29+ actual <- setupSrv srv (compile "code")
3030 expected `equal` actual
3131 test "rename imports" do
3232 let url = settings.protocol
@@ -37,4 +37,4 @@ runTests = runTest do
3737 Tuple code expected = mkImport url `foldMap` [ "Prelude", "Effect" ]
3838 compile = runCompiler settings
3939 actual <- setupSrv code (compile code)
40− expected `equal` actual
40+ (appendMain expected) `equal` actual