Support both IO and clean version of Generate monad.

[hs-java.git] / TestGen.hs

diff --git a/TestGen.hs b/TestGen.hs
index c8cd0ef3c91578696e3db0355efc384f76225343..31e71c0676ae0255d345a46d8ac1d6a502dd1dd5 100644 (file)
--- a/TestGen.hs
+++ b/TestGen.hs
@@ -11,7 +11,7 @@ import Java.ClassPath
 import qualified Java.Lang
 import qualified Java.IO
 
-test :: Generate ()
+test :: GenerateIO ()
 test = do
   withClassPath $ do
       -- Add current directory (with Hello.class) to ClassPath
@@ -45,6 +45,7 @@ test = do
       invokeStatic Java.Lang.integer Java.Lang.valueOfInteger
       aastore
       invokeVirtual Java.IO.printStream Java.IO.printf
+      -- Call Hello.hello()
       invokeStatic "Hello" helloJava
       pop
       i0 RETURN
@@ -62,6 +63,6 @@ test = do
 
 main :: IO ()
 main = do
-  testClass <- generate [] "Test" test
+  testClass <- generateIO [] "Test" test
   B.writeFile "Test.class" (encodeClass testClass)