Merge pull request #39 from bpillai/feature/make-get-map-to-allow-tes…

…ting add public get map method to allow testing
americanexpress · Oct 12, 2018 · f52628f · f52628f
2 parents 72d6bba + 714c5a8
commit f52628f
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 0 deletions.
diff --git a/nodes/src/main/java/io/aexp/nodes/graphql/InputObject.java b/nodes/src/main/java/io/aexp/nodes/graphql/InputObject.java
@@ -49,6 +49,9 @@ public Builder put(String key, T value) {
         public InputObject<T> build() {
             return new InputObject<T>(this);
         }
+    }
 
+    public Map getMap() {
+        return map;
     }
 }
diff --git a/nodes/src/test/java/io/aexp/nodes/graphql/ArgumentTest.java b/nodes/src/test/java/io/aexp/nodes/graphql/ArgumentTest.java
@@ -52,4 +52,13 @@ public void argumentEnumTest() {
         assertEquals(STATUS.active, argument.getValue());
         assertEquals("Argument{key='test', value=active}", argument.toString());
     }
+    @Test
+    public void argumentInputObjectTest() {
+        Argument argument = new Argument(null, null);
+        argument.setKey("test");
+        InputObject object = new InputObject.Builder().put("key", "value").build();
+        argument.setValue(object);
+        assertEquals("test", argument.getKey());
+        assertEquals("value", object.getMap().get("key"));
+    }
 }