1 files changed, 24 insertions, 3 deletions
diff --git a/src/test/java/com/seaofnodes/simple/PhaseTest.java b/src/test/java/com/seaofnodes/simple/PhaseTest.java
index 5ff7fcd6..7252378a 100644
--- a/src/test/java/com/seaofnodes/simple/PhaseTest.java
+++ b/src/test/java/com/seaofnodes/simple/PhaseTest.java
@@ -22,22 +22,43 @@ public class PhaseTest {
         code.driver(Phase.TypeCheck);
         code.driver(Phase.TypeCheck); // A repeated request does not rerun anything.
         code.driver(Phase.LocalSched);
-        assertEquals("Parse Iter Opto IterPostOpto FinalizeIR TypeCheck Lower LoopTree Schedule LocalSched",
+        assertEquals("Parse Iter Opto IterPostOpto FinalizeIR TypeCheck Lower LoopTree Unlink Schedule LocalSched",
                      String.join(" ",phases));
+        assertNull(code._serial);
         assertNull(code._mach);
         assertEquals("45",Eval2.eval(code,10));
     }
 
-    @Test public void testEncodingAndResume() {
+    @Test public void testEncodingWithoutSerialization() {
         for( String cpu : new String[]{"x86_64_v2","arm","riscv"} ) {
             CodeGen code = new CodeGen(SRC).driver(Phase.TypeCheck);
-            code.driver(Phase.LoopTree,cpu,"SystemV");
+            code.driver(Phase.Unlink,cpu,"SystemV");
             code.driver(Phase.Select,cpu,"SystemV");
             code.driver(Phase.Encoding,cpu,"SystemV");
+            assertNull(code._serial);
             byte[] staged = code._encoding._bits.toByteArray();
             CodeGen direct = new CodeGen(SRC).driver(Phase.Encoding,cpu,"SystemV");
             assertArrayEquals(staged,direct._encoding._bits.toByteArray());
         }
     }
 
+    @Test public void testExplicitSerializationAndMemoryExport() {
+        CodeGen code = new CodeGen(SRC).driver(Phase.Serialize);
+        assertNotNull(code._serial);
+        assertEquals(Phase.Serialize,code._phase);
+        code.driver(Phase.Encoding,"riscv","SystemV");
+        assertNotNull(code._serial);
+
+        CodeGen memory = new CodeGen(SRC).driver("riscv","SystemV",true,false);
+        assertNull(memory._serial);
+        assertEquals(Phase.Export,memory._phase);
+
+        CodeGen late = new CodeGen(SRC).driver(Phase.Encoding,"riscv","SystemV");
+        try {
+            late.exportELF(false,false);
+            fail("Cannot recover ideal IR after lowering and instruction selection");
+        } catch( IllegalStateException expected ) {
+            assertTrue(expected.getMessage().contains("Serialize before Lower"));
+        }
+    }
 }