1 files changed, 128 insertions, 0 deletions
diff --git a/vespajlib/src/test/java/ai/vespa/json/JsonTest.java b/vespajlib/src/test/java/ai/vespa/json/JsonTest.java
new file mode 100644
index 00000000000..293e99227a7
--- /dev/null
+++ b/vespajlib/src/test/java/ai/vespa/json/JsonTest.java
@@ -0,0 +1,128 @@
+package ai.vespa.json;
+
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertThrows;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+/**
+ * @author bjorncs
+ */
+class JsonTest {
+
+    @Test
+    void parses_json_correctly() {
+        var text =
+                """
+                {
+                    "string": "bar",
+                    "integer": 42,
+                    "floaty": 8.25,
+                    "bool": true,
+                    "array": [1, 2, 3],
+                    "quux": {
+                        "corge": "grault"
+                    }
+                }
+                """;
+        var json = Json.of(text);
+
+        // Primitive members
+        assertEquals("bar", json.f("string").asString());
+        assertTrue(json.f("string").asOptionalString().isPresent());
+        assertEquals("bar", json.f("string").asOptionalString().get());
+        assertEquals(42L, json.f("integer").asLong());
+        assertEquals(42D, json.f("integer").asDouble());
+        assertEquals(8.25D, json.f("floaty").asDouble());
+        assertEquals(8L, json.f("floaty").asLong());
+        assertTrue(json.f("bool").asBool());
+
+        // Array member
+        assertEquals(3, json.f("array").length());
+        assertEquals(1L, json.f("array").entry(0).asLong());
+        assertEquals(2L, json.f("array").entry(1).asLong());
+        assertEquals(3L, json.f("array").entry(2).asLong());
+        json.f("array").forEachEntry((i, entry) -> assertEquals(i + 1, entry.asLong()));
+        int counter = 0;
+        for (var entry : json.f("array")) {
+            assertEquals(++counter, entry.asLong());
+        }
+
+        // Object member
+        assertEquals("grault", json.f("quux").f("corge").asString());
+        json.f("quux").forEachField((name, child) -> {
+            assertEquals("corge", name);
+            assertEquals("grault", child.asString());
+        });
+    }
+
+    @Test
+    void throws_on_missing_and_invalid_members() {
+        var text =
+                """
+                {
+                    "string": "bar"
+                }
+                """;
+        var json = Json.of(text);
+
+        var exception = assertThrows(InvalidJsonException.class, () -> json.f("unknown").asString());
+        assertEquals("Missing JSON member 'unknown'", exception.getMessage());
+
+        exception = assertThrows(InvalidJsonException.class, () -> json.a(0));
+        assertEquals("Expected JSON to be a 'array' but got 'object'", exception.getMessage());
+
+        exception = assertThrows(InvalidJsonException.class, () -> json.f("string").f("unknown"));
+        assertEquals("Expected JSON member 'string' to be a 'object' but got 'string'", exception.getMessage());
+
+        exception = assertThrows(InvalidJsonException.class, () -> json.f("string").asLong());
+        assertEquals("Expected JSON member 'string' to be a 'integer' or 'float' but got 'string'", exception.getMessage());
+    }
+
+    @Test
+    void fallback_to_default_if_field_missing() {
+        var text =
+                """
+                {
+                    "string": "bar"
+                }
+                """;
+        var json = Json.of(text);
+        assertEquals("foo", json.f("unknown").asString("foo"));
+        assertEquals("foo", json.f("unknown").asOptionalString().orElse("foo"));
+        assertEquals("bar", json.f("string").asString("foo"));
+        assertEquals("bar", json.f("string").asOptionalString().orElse("foo"));
+    }
+
+    @Test
+    void builds_expected_json() {
+        var expected =
+                """
+                {
+                  "string": "bar",
+                  "integer": 42,
+                  "floaty": 8.25,
+                  "bool": true,
+                  "array": [
+                    1,
+                    2,
+                    3
+                  ],
+                  "quux": {
+                    "corge": "grault"
+                  }
+                }
+                """;
+        var json = Json.Builder.newObject()
+                .set("string", "bar")
+                .set("integer", 42)
+                .set("floaty", 8.25)
+                .set("bool", true)
+                .set("array", Json.Builder.newArray().add(1).add(2).add(3))
+                .set("quux", Json.Builder.newObject().set("corge", "grault"))
+                .build()
+                .toJson(true);
+        assertEquals(expected, json);
+    }
+}