Refactor.

[hs-java.git] / TestGen.hs

diff --git a/TestGen.hs b/TestGen.hs
index 175e10b1b79a1f8a153baf35058f78d8027edf45..900a422dd5778892483b970457018cfd32094ad0 100644 (file)
--- a/TestGen.hs
+++ b/TestGen.hs
@@ -13,11 +13,15 @@ import qualified Java.IO
 test :: Generate ()
 test = do
   newMethod [ACC_PUBLIC] "<init>" [] ReturnsVoid $ do
+      setStackSize 1
+
       aload_ I0
       invokeSpecial Java.Lang.object Java.Lang.objectInit
       i0 RETURN
 
   hello <- newMethod [ACC_PUBLIC, ACC_STATIC] "hello" [IntType] ReturnsVoid $ do
+      setStackSize 8
+
       getStaticField Java.Lang.system Java.IO.out
       loadString "Здравствуй, мир!"
       invokeVirtual Java.IO.printStream Java.IO.println
@@ -35,15 +39,15 @@ test = do
       i0 RETURN
 
   newMethod [ACC_PUBLIC, ACC_STATIC] "main" [arrayOf Java.Lang.stringClass] ReturnsVoid $ do
+      setStackSize 1
+
       iconst_5
       invokeStatic "Test" hello
       i0 RETURN
 
   return ()
 
-testClass ::  Class Resolved
-testClass = generate "Test" test
-
 main = do
+  testClass <- generate [] "Test" test
   B.writeFile "Test.class" (encodeClass testClass)