Add QueryCommand back to command handler

usnistgov · Jan 11, 2024 · 52cc75d · 52cc75d
1 parent cd19acc
commit 52cc75d
Show file tree

Hide file tree

Showing 2 changed files with 49 additions and 49 deletions.
diff --git a/metaschema-cli/src/main/java/gov/nist/secauto/metaschema/cli/CLI.java b/metaschema-cli/src/main/java/gov/nist/secauto/metaschema/cli/CLI.java
@@ -27,6 +27,7 @@
 package gov.nist.secauto.metaschema.cli;
 
 import gov.nist.secauto.metaschema.cli.commands.GenerateSchemaCommand;
+import gov.nist.secauto.metaschema.cli.commands.QueryCommand;
 import gov.nist.secauto.metaschema.cli.commands.ValidateContentUsingModuleCommand;
 import gov.nist.secauto.metaschema.cli.commands.ValidateModuleCommand;
 import gov.nist.secauto.metaschema.cli.processor.CLIProcessor;
@@ -59,6 +60,7 @@ public static ExitStatus runCli(String... args) {
     processor.addCommandHandler(new ValidateModuleCommand());
     processor.addCommandHandler(new GenerateSchemaCommand());
     processor.addCommandHandler(new ValidateContentUsingModuleCommand());
+    processor.addCommandHandler(new QueryCommand());
 
     CommandService.getInstance().getCommands().stream().forEach(command -> {
       assert command != null;

diff --git a/metaschema-cli/src/test/java/gov/nist/secauto/metaschema/cli/CLITest.java b/metaschema-cli/src/test/java/gov/nist/secauto/metaschema/cli/CLITest.java
@@ -32,19 +32,13 @@
 
 import gov.nist.secauto.metaschema.cli.processor.ExitCode;
 import gov.nist.secauto.metaschema.cli.processor.ExitStatus;
-import gov.nist.secauto.metaschema.cli.CLI;
 
-import org.junit.jupiter.api.Test;
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.Arguments;
 import org.junit.jupiter.params.provider.MethodSource;
 
-import java.nio.file.Path;
-import java.nio.file.Paths;
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.List;
-import java.util.Map;
 import java.util.stream.Stream;
 
 import edu.umd.cs.findbugs.annotations.NonNull;
@@ -53,50 +47,54 @@
  * Unit test for simple CLI.
  */
 public class CLITest {
-	void evaluateResult(@NonNull ExitStatus status, @NonNull ExitCode expectedCode) {
-		status.generateMessage(true);
-		assertAll(() -> assertEquals(expectedCode, status.getExitCode(), "exit code mismatch"),
-				() -> assertNull(status.getThrowable(), "expected null Throwable"));
-	}
+  void evaluateResult(@NonNull ExitStatus status, @NonNull ExitCode expectedCode) {
+    status.generateMessage(true);
+    assertAll(() -> assertEquals(expectedCode, status.getExitCode(), "exit code mismatch"),
+        () -> assertNull(status.getThrowable(), "expected null Throwable"));
+  }
 
-	void evaluateResult(@NonNull ExitStatus status, @NonNull ExitCode expectedCode,
-			@NonNull Class<? extends Throwable> thrownClass) {
-		status.generateMessage(true);
-		Throwable thrown = status.getThrowable();
-		assert thrown != null;
-		assertAll(() -> assertEquals(expectedCode, status.getExitCode(), "exit code mismatch"),
-				() -> assertEquals(thrownClass, thrown.getClass(), "expected Throwable mismatch"));
-	}
+  void evaluateResult(@NonNull ExitStatus status, @NonNull ExitCode expectedCode,
+      @NonNull Class<? extends Throwable> thrownClass) {
+    status.generateMessage(true);
+    Throwable thrown = status.getThrowable();
+    assert thrown != null;
+    assertAll(() -> assertEquals(expectedCode, status.getExitCode(), "exit code mismatch"),
+        () -> assertEquals(thrownClass, thrown.getClass(), "expected Throwable mismatch"));
+  }
 
-	private static Stream<Arguments> providesValues() {
-		ExitCode noExpectedExceptionClass = null;
-		List<Arguments> values = new ArrayList<>();
-		values.add(Arguments.of(new String[] {}, ExitCode.INVALID_COMMAND, noExpectedExceptionClass));
-		values.add(Arguments.of(new String[] { "-h" }, ExitCode.OK, noExpectedExceptionClass));
-		values.add(Arguments.of(new String[] { "generate-schema", "--help" }, ExitCode.INVALID_COMMAND,
-				noExpectedExceptionClass));
-		values.add(Arguments.of(new String[] { "validate", "--help" }, ExitCode.OK, noExpectedExceptionClass));
-		values.add(Arguments.of(new String[] { "validate-content", "--help" }, ExitCode.INVALID_COMMAND,
-				noExpectedExceptionClass));
-		values.add(Arguments.of(
-				new String[] { "validate",
-						"../databind/src/test/resources/metaschema/fields_with_flags/metaschema.xml" },
-				ExitCode.OK, noExpectedExceptionClass));
-		values.add(Arguments.of(new String[] { "generate-schema", "--overwrite", "--as", "JSON",
-				"../databind/src/test/resources/metaschema/fields_with_flags/metaschema.xml",
-				"target/schema-test.json" }, ExitCode.OK, noExpectedExceptionClass));
-		values.add(Arguments.of(new String[] { "query", "-m", "module.xml", "-i", "content-instance.xml", "\"2 + 2\""}, ExitCode.INVALID_COMMAND, noExpectedExceptionClass));
-		return values.stream();
-	}
+  private static Stream<Arguments> providesValues() {
+    ExitCode noExpectedExceptionClass = null;
+    List<Arguments> values = new ArrayList<>();
+    values.add(Arguments.of(new String[] {}, ExitCode.INVALID_COMMAND, noExpectedExceptionClass));
+    values.add(Arguments.of(new String[] { "-h" }, ExitCode.OK, noExpectedExceptionClass));
+    values.add(Arguments.of(new String[] { "generate-schema", "--help" }, ExitCode.INVALID_COMMAND,
+        noExpectedExceptionClass));
+    values.add(Arguments.of(new String[] { "validate", "--help" }, ExitCode.OK, noExpectedExceptionClass));
+    values.add(Arguments.of(new String[] { "validate-content", "--help" }, ExitCode.INVALID_COMMAND,
+        noExpectedExceptionClass));
+    values.add(Arguments.of(
+        new String[] { "validate",
+            "../databind/src/test/resources/metaschema/fields_with_flags/metaschema.xml" },
+        ExitCode.OK, noExpectedExceptionClass));
+    values.add(Arguments.of(new String[] { "generate-schema", "--overwrite", "--as", "JSON",
+        "../databind/src/test/resources/metaschema/fields_with_flags/metaschema.xml",
+        "target/schema-test.json" }, ExitCode.OK, noExpectedExceptionClass));
+    values.add(Arguments.of(
+        new String[] { "query", "-m", "../databind/src/test/resources/metaschema/fields_with_flags/metaschema.xml",
+            "-i",
+            "../databind/src/test/resources/metaschema/fields_with_flags/example.json", "2 + 2" },
+        ExitCode.OK, noExpectedExceptionClass));
+    return values.stream();
+  }
 
-	@ParameterizedTest
-	@MethodSource("providesValues")
-	void testAllCommands(@NonNull String[] args, @NonNull ExitCode expectedExitCode,
-			Class<? extends Throwable> expectedThrownClass) {
-		if (expectedThrownClass == null) {
-			evaluateResult(CLI.runCli(args), expectedExitCode);
-		} else {
-			evaluateResult(CLI.runCli(args), expectedExitCode, expectedThrownClass);
-		}
-	}
+  @ParameterizedTest
+  @MethodSource("providesValues")
+  void testAllCommands(@NonNull String[] args, @NonNull ExitCode expectedExitCode,
+      Class<? extends Throwable> expectedThrownClass) {
+    if (expectedThrownClass == null) {
+      evaluateResult(CLI.runCli(args), expectedExitCode);
+    } else {
+      evaluateResult(CLI.runCli(args), expectedExitCode, expectedThrownClass);
+    }
+  }
 }