Merge pull request #17 from fauna/feature/BT-4418-utf8faunareader_rea…

…d_boolean [BT-4418] FaunaParser Boolean values
fauna · Jan 30, 2024 · 545a041 · 545a041
2 parents c300f11 + 7efe46d
commit 545a041
Show file tree

Hide file tree

Showing 2 changed files with 42 additions and 0 deletions.
diff --git a/faunaJava/src/main/java/com/fauna/serialization/FaunaParser.java b/faunaJava/src/main/java/com/fauna/serialization/FaunaParser.java
@@ -93,6 +93,12 @@ public boolean read() throws IOException {
                 case START_OBJECT:
                     handleStartObject();
                     break;
+                case VALUE_TRUE:
+                    currentFaunaTokenType = FaunaTokenType.TRUE;
+                    break;
+                case VALUE_FALSE:
+                    currentFaunaTokenType = FaunaTokenType.FALSE;
+                    break;
                 default:
                     throw new SerializationException(
                         "Unhandled JSON token type " + currentToken + ".");
@@ -186,4 +192,12 @@ public Integer getValueAsInt() {
             throw new RuntimeException("Error getting the current token as Integer", e);
         }
     }
+
+    public Boolean getValueAsBoolean() {
+        try {
+            return jsonParser.getValueAsBoolean();
+        } catch (IOException e) {
+            throw new SerializationException("Error getting the current token as Boolean", e);
+        }
+    }
 }
diff --git a/faunaJava/src/test/java/com/fauna/serialization/FaunaParserTest.java b/faunaJava/src/test/java/com/fauna/serialization/FaunaParserTest.java
@@ -59,6 +59,30 @@ public void testUnexpectedEndDuringAdvance() throws IOException {
         assertThrows(SerializationException.class, reader::read);
     }
 
+    @Test
+    public void testGetValueAsBooleanTrue() throws IOException {
+        InputStream inputStream = new ByteArrayInputStream("true".getBytes());
+        FaunaParser reader = new FaunaParser(inputStream);
+
+        List<Map.Entry<FaunaTokenType, Object>> expectedTokens = List.of(
+            Map.entry(FaunaTokenType.TRUE, true)
+        );
+
+        assertReader(reader, expectedTokens);
+    }
+
+    @Test
+    public void testGetValueAsBooleanFalse() throws IOException {
+        InputStream inputStream = new ByteArrayInputStream("false".getBytes());
+        FaunaParser reader = new FaunaParser(inputStream);
+
+        List<Map.Entry<FaunaTokenType, Object>> expectedTokens = List.of(
+            Map.entry(FaunaTokenType.FALSE, false)
+        );
+
+        assertReader(reader, expectedTokens);
+    }
+
     private static void assertReader(FaunaParser reader,
         List<Map.Entry<FaunaTokenType, Object>> tokens) throws IOException {
         for (Map.Entry<FaunaTokenType, Object> entry : tokens) {
@@ -75,6 +99,10 @@ private static void assertReader(FaunaParser reader,
                 case INT:
                     assertEquals(entry.getValue(), reader.getValueAsInt());
                     break;
+                case TRUE:
+                case FALSE:
+                    assertEquals(entry.getValue(), reader.getValueAsBoolean());
+                    break;
                 default:
                     assertNull(entry.getValue() == null);
                     break;