unchecked dispatch tests

Author: Frotty

de.peeeq.wurstscript/src/test/java/tests/wurstscript/tests/GenericsWithTypeclassesTests.java

@@ -2091,11 +2091,49 @@ public void fullArrayListTest() throws IOException {
        assertEquals(count, 2);
 
         assertFalse(compiled.contains("ArrayList_nextFreeIndex_"));
+        // In checked mode, virtual method calls should go through dispatch.
+        assertTrue(compiled.contains("dispatch_ArrayList"));
+    }
+
+    @Test
+    public void fullArrayListTestUncheckedDispatch() throws IOException {
+        test().withStdLib().uncheckedDispatch().executeProg().executeTests().file(new File(TEST_DIR + "arrayList.wurst"));
+
+        String compiled = Files.toString(
+            new File(TEST_OUTPUT_PATH + "GenericsWithTypeclassesTests_fullArrayListTestUncheckedDispatch_no_opts.jim"),
+            Charsets.UTF_8);
+
+        // In unchecked dispatch mode, monomorphic ArrayList<T>.get should be lowered directly.
+        assertTrue(compiled.contains("ArrayList_get"));
+        assertFalse(compiled.contains("dispatch_ArrayList"));
     }
 
     @Test
     public void fullReactiveGenericDispatchTest() throws IOException {
         test().withStdLib().executeProg().executeTests().file(new File(TEST_DIR + "reactiveGenericsDispatch.wurst"));
+
+        String compiled = Files.toString(
+            new File(TEST_OUTPUT_PATH + "GenericsWithTypeclassesTests_fullReactiveGenericDispatchTest_no_opts.jim"),
+            Charsets.UTF_8);
+
+        // In checked mode, polymorphic interface calls are dispatched and guarded.
+        assertTrue(compiled.contains("dispatch_ReactiveSource_ReactiveSource_subscribe"));
+        assertTrue(compiled.contains("Nullpointer exception when calling ReactiveSource.subscribe"));
+        assertTrue(compiled.contains("Called ReactiveSource.subscribe on invalid object."));
+    }
+
+    @Test
+    public void fullReactiveGenericDispatchTestUncheckedDispatch() throws IOException {
+        test().withStdLib().uncheckedDispatch().executeProg().executeTests().file(new File(TEST_DIR + "reactiveGenericsDispatch.wurst"));
+
+        String compiled = Files.toString(
+            new File(TEST_OUTPUT_PATH + "GenericsWithTypeclassesTests_fullReactiveGenericDispatchTestUncheckedDispatch_no_opts.jim"),
+            Charsets.UTF_8);
+
+        // ReactiveSource is polymorphic, so dispatch remains, but safety checks should be removed.
+        assertTrue(compiled.contains("dispatch_ReactiveSource_ReactiveSource_subscribe"));
+        assertFalse(compiled.contains("Nullpointer exception when calling ReactiveSource.subscribe"));
+        assertFalse(compiled.contains("Called ReactiveSource.subscribe on invalid object."));
     }
 
     @Test

de.peeeq.wurstscript/src/test/java/tests/wurstscript/tests/WurstScriptTest.java

@@ -70,6 +70,7 @@ class TestConfig {
         private boolean stopOnFirstError = true;
         private boolean runCompiletimeFunctions;
         private boolean testLua = false;
+        private boolean uncheckedDispatch = false;
 
         TestConfig(String name) {
             this.name = name;
@@ -123,6 +124,15 @@ public TestConfig executeProgOnlyAfterTransforms(boolean b) {
             return this;
         }
 
+        public TestConfig uncheckedDispatch() {
+            return uncheckedDispatch(true);
+        }
+
+        public TestConfig uncheckedDispatch(boolean b) {
+            this.uncheckedDispatch = b;
+            return this;
+        }
+
         TestConfig expectError(String expectedError) {
             this.expectedError = expectedError;
             return this;
@@ -186,6 +196,9 @@ private CompilationResult testScript() {
             if (withStdLib) {
                 runArgs = runArgs.with("-lib", StdLib.getLib());
             }
+            if (uncheckedDispatch) {
+                runArgs = runArgs.with("-uncheckedDispatch");
+            }
             if (runCompiletimeFunctions) {
                 runArgs = runArgs.with("-runcompiletimefunctions");
             }