Add additional function signature test

CraftTweaker · Dec 15, 2020 · f4f0ebc · f4f0ebc
1 parent a51dd36
commit f4f0ebc
Showing 1 changed file with 15 additions and 5 deletions.
diff --git a/src/test/java/stanhebben/zenscript/tests/TestFunctionTypesInSignatures.java b/src/test/java/stanhebben/zenscript/tests/TestFunctionTypesInSignatures.java
@@ -1,10 +1,7 @@
 package stanhebben.zenscript.tests;
 
-import com.google.gson.internal.*;
-import org.junit.jupiter.api.BeforeAll;
-import org.junit.jupiter.api.BeforeEach;
-import org.junit.jupiter.api.Test;
-import stanhebben.zenscript.TestHelper;
+import org.junit.jupiter.api.*;
+import stanhebben.zenscript.*;
 
 import java.util.*;
 
@@ -45,4 +42,17 @@ public void TestFunctionExpressionInClassSignature() {
         TestHelper.run(joiner.toString());
         assertMany("inside fun: abc");
     }
+
+    @Test
+    public void TestFuncSupplier() {
+        final StringJoiner joiner = new StringJoiner("\n");
+        joiner.add("global funcCreator as function(int)function()string = function(i as int) as function() string {");
+        joiner.add("    return function() as string {return 'Hello, Nr. ' ~ i;};");
+        joiner.add("};");
+        joiner.add("print(funcCreator(10)());");
+        joiner.add("print(funcCreator(42)());");
+
+        TestHelper.run(joiner.toString());
+        assertMany("Hello, Nr. 10", "Hello, Nr. 42");
+    }
 }