feat (cli): Add BinaryPB output --format

vorburger · Oct 1, 2023 · 71d8a05 · 71d8a05
1 parent ca8ba60
commit 71d8a05
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/cli/src/main/java/dev/enola/cli/Format.java b/cli/src/main/java/dev/enola/cli/Format.java
@@ -26,7 +26,9 @@ public enum Format {
 
     YAML,
 
-    JSON;
+    JSON,
+
+    BinaryPB;
 
     MediaType toMediaType() {
         switch (this) {
@@ -36,6 +38,8 @@ MediaType toMediaType() {
                 return ProtobufMediaTypes.PROTOBUF_YAML_UTF_8;
             case JSON:
                 return ProtobufMediaTypes.PROTOBUF_JSON_UTF_8;
+            case BinaryPB:
+                return ProtobufMediaTypes.PROTOBUF_BINARY;
             default:
                 throw new IllegalStateException();
         }

diff --git a/cli/src/test/java/dev/enola/cli/EnolaTest.java b/cli/src/test/java/dev/enola/cli/EnolaTest.java
@@ -113,6 +113,21 @@ public void get() {
                                 + "  paths: \"helo\"\n");
     }
 
+    @Test
+    public void getBinary() {
+        var exec =
+                cli(
+                        "-v",
+                        "get",
+                        "--format",
+                        "BinaryPB",
+                        "--model",
+                        "classpath:cli-test-model.textproto",
+                        "test.foobar/helo");
+        assertThat(exec).err().isEmpty();
+        assertThat(exec).hasExitCode(0);
+    }
+
     @Test
     public void serveBothHttpAndGRPC() {
         var exec =