property on list shortcut

Author: MykolaGolubyev

webtau-docs/webtau/REST/data-node.md

@@ -3,7 +3,7 @@
 :include-groovy: com/twosigma/webtau/http/HttpTest.groovy {entry: "use groovy closure as validation", bodyOnly: true}
 
 Values that you access inside validation block are special values of `DataNode` type. When you assert them using `should` statement
-they act as proxies that record every assertion you do. 
+they act as proxies that record every assertion that you do. 
 
 
 # Extracting Values
@@ -17,10 +17,16 @@ When you return a value from a validation block, it automatically gets converted
 Note: asserting that value after returning will not track and associated assertions with the call anymore. Use it only
 to get values required for consequent test calls.  
 
+# Properties On Lists
+
+:include-json: objectTestResponse.json
+
+If you have a list of objects like `complexList` above, you can access all its children property value with `complexList.k2`.
+
+:include-groovy: com/twosigma/webtau/http/HttpTest.groovy {entry: "groovy children key shortcut", bodyOnly: true}
 
 # Find
 
-:include-json: objectTestResponse.json
 
 Special values inside assertion block have convenient methods
 
@@ -42,3 +48,9 @@ Note: While values inside a predicate are normal values, the result of `find` an
 Use `collect` to transform a collection of items
 
 :include-groovy: com/twosigma/webtau/http/HttpTest.groovy {entry: "groovy transform list", bodyOnly: true}
+
+# Combine
+
+Methods `find` and `collect` can be chained
+ 
+:include-groovy: com/twosigma/webtau/http/HttpTest.groovy {entry: "groovy findAll, collect, and sum", bodyOnly: true}

webtau-rest-groovy/src/test/groovy/com/twosigma/webtau/http/HttpTest.groovy

@@ -381,18 +381,6 @@ class HttpTest {
         assert found[0].getClass() == Integer
     }
 
-    @Test
-    void "groovy findAlll, collect, and sum"() {
-        def sum = http.get("/end-point") {
-            return complexList
-                    .findAll { it.k1.startsWith('v1') }
-                    .collect { it.k2 }
-                    .sum()
-        }
-
-        assert sum == 70
-    }
-
     @Test
     void "groovy findAll on body that is not a list"() {
         def found = http.get("/end-point") {
@@ -445,6 +433,25 @@ class HttpTest {
         assert transformed == []
     }
 
+    @Test
+    void "groovy findAll, collect, and sum"() {
+        def sum = http.get("/end-point") {
+            return complexList
+                    .findAll { k1.startsWith('v1') }
+                    .collect { k2 }
+                    .sum()
+        }
+
+        assert sum == 70
+    }
+
+    @Test
+    void "groovy children key shortcut"() {
+        http.get("/end-point") {
+            complexList.k2.should == [30, 40]
+        }
+    }
+
     @Test
     void "send form data"() {
         byte[] content = [0, 1, 2, 101, 102, 103, 0] as byte[]

webtau-rest/src/main/java/com/twosigma/webtau/http/datanode/StructuredDataNode.java

@@ -21,6 +21,7 @@
 import java.util.Collections;
 import java.util.List;
 import java.util.Map;
+import java.util.stream.Collectors;
 
 import static java.util.stream.Collectors.joining;
 
@@ -56,7 +57,13 @@ public DataNodeId id() {
 
     @Override
     public DataNode get(String name) {
-        return (children != null && children.containsKey(name)) ? children.get(name) : new NullDataNode(id.child(name));
+        if (isList()) {
+            return getAsCollectFromList(name);
+        }
+
+        return (children != null && children.containsKey(name)) ?
+                children.get(name):
+                new NullDataNode(id.child(name));
     }
 
     @Override
@@ -117,4 +124,11 @@ public String toString() {
 
         return "{" + children.entrySet().stream().map(e -> e.getKey() + ": " + e.getValue()).collect(joining(", "))  + "}";
     }
+
+    private DataNode getAsCollectFromList(String name) {
+        return new StructuredDataNode(id.child(name),
+                values.stream()
+                        .map(n -> n.get(name))
+                        .collect(Collectors.toList()));
+    }
 }

webtau-rest/src/test/groovy/com/twosigma/webtau/http/datanode/StructuredDataNodeTest.groovy

@@ -62,4 +62,19 @@ class StructuredDataNodeTest {
 
         node.get().checkLevel.should == CheckLevel.ExplicitFailed
     }
+
+    @Test
+    void "shortcut for children props"() {
+        def node = DataNodeBuilder.fromList(new DataNodeId("body"), [
+                [name: 'name1', score: 10],
+                [name: 'name2', score: 20],
+        ])
+
+        node.name.should(equal(['name1', 'name2']))
+
+        node.get(0).get('name').get().checkLevel.should == CheckLevel.ExplicitPassed
+        node.get(1).get('name').get().checkLevel.should == CheckLevel.ExplicitPassed
+        node.get(0).get('score').get().checkLevel.should == CheckLevel.None
+        node.get(1).get('score').get().checkLevel.should == CheckLevel.None
+    }
 }