Fix unit tests

Rollczi · Dec 19, 2024 · 03ddf92 · 03ddf92
1 parent 6a9b297
commit 03ddf92
Show file tree

Hide file tree

Showing 4 changed files with 31 additions and 1 deletion.
diff --git a/...otations/test/dev/rollczi/litecommands/annotations/schematic/SchematicConfiguredTest.java b/...otations/test/dev/rollczi/litecommands/annotations/schematic/SchematicConfiguredTest.java
@@ -1,6 +1,7 @@
 
 package dev.rollczi.litecommands.annotations.schematic;
 
+import dev.rollczi.litecommands.unit.Parsers;
 import dev.rollczi.litecommands.unit.annotations.LiteTestSpec;
 import dev.rollczi.litecommands.annotations.argument.Arg;
 import dev.rollczi.litecommands.annotations.async.Async;
@@ -20,6 +21,7 @@
 class SchematicConfiguredTest extends LiteTestSpec {
 
     static LiteTestConfig config = builder -> builder
+        .argumentParser(ServerRank.class, Parsers.of(new ServerRank()))
         .schematicGenerator(SchematicFormat.parentheses());
 
     static class ServerRank {}

diff --git a/...mmands-annotations/test/dev/rollczi/litecommands/annotations/schematic/SchematicTest.java b/...mmands-annotations/test/dev/rollczi/litecommands/annotations/schematic/SchematicTest.java
@@ -1,5 +1,10 @@
 package dev.rollczi.litecommands.annotations.schematic;
 
+import dev.rollczi.litecommands.argument.Argument;
+import dev.rollczi.litecommands.argument.parser.ParseResult;
+import dev.rollczi.litecommands.argument.resolver.ArgumentResolver;
+import dev.rollczi.litecommands.invocation.Invocation;
+import dev.rollczi.litecommands.unit.Parsers;
 import dev.rollczi.litecommands.unit.annotations.LiteTestSpec;
 import dev.rollczi.litecommands.annotations.argument.Arg;
 import dev.rollczi.litecommands.annotations.flag.Flag;
@@ -18,6 +23,9 @@
 
 class SchematicTest extends LiteTestSpec {
 
+    static LiteTestConfig config = builder -> builder
+        .argumentParser(ServerRank.class, Parsers.of(new ServerRank()));
+
     static class ServerRank {}
 
     @Command(name = "rank")

diff --git a/litecommands-core/src/dev/rollczi/litecommands/event/SimpleEventPublisher.java b/litecommands-core/src/dev/rollczi/litecommands/event/SimpleEventPublisher.java
@@ -106,7 +106,7 @@ public void invoke(Event event) {
                 throw new LiteCommandsReflectInvocationException(declaredMethod, "Cannot access method", exception);
             }
             catch (InvocationTargetException exception) {
-                throw new LiteCommandsReflectInvocationException(declaredMethod, "Cannot invoke method", exception);
+                throw new LiteCommandsReflectInvocationException(declaredMethod, "Cannot invoke method", exception.getCause());
             }
         }
 

diff --git a/litecommands-unit/src/dev/rollczi/litecommands/unit/Parsers.java b/litecommands-unit/src/dev/rollczi/litecommands/unit/Parsers.java
@@ -0,0 +1,20 @@
+package dev.rollczi.litecommands.unit;
+
+import dev.rollczi.litecommands.argument.Argument;
+import dev.rollczi.litecommands.argument.parser.ParseResult;
+import dev.rollczi.litecommands.argument.parser.Parser;
+import dev.rollczi.litecommands.argument.resolver.ArgumentResolver;
+import dev.rollczi.litecommands.invocation.Invocation;
+
+public final class Parsers {
+
+    public static <S, T> Parser<S, T> of(T result) {
+        return new ArgumentResolver<S, T>() {
+            @Override
+            protected ParseResult<T> parse(Invocation<S> invocation, Argument<T> context, String argument) {
+                return ParseResult.success(result);
+            }
+        };
+    }
+
+}