You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tinkerpop.apache.org by sp...@apache.org on 2015/03/11 18:28:35 UTC

[09/19] incubator-tinkerpop git commit: Add test for IteratorUtils.collectMap.

Add test for IteratorUtils.collectMap.


Project: http://git-wip-us.apache.org/repos/asf/incubator-tinkerpop/repo
Commit: http://git-wip-us.apache.org/repos/asf/incubator-tinkerpop/commit/56b5e352
Tree: http://git-wip-us.apache.org/repos/asf/incubator-tinkerpop/tree/56b5e352
Diff: http://git-wip-us.apache.org/repos/asf/incubator-tinkerpop/diff/56b5e352

Branch: refs/heads/newapi
Commit: 56b5e352cdc57f00d390e4d366921c3ce817fbaa
Parents: b2c673c
Author: Stephen Mallette <sp...@apache.org>
Authored: Tue Mar 10 11:13:53 2015 -0400
Committer: Stephen Mallette <sp...@apache.org>
Committed: Tue Mar 10 11:13:53 2015 -0400

----------------------------------------------------------------------
 .../util/iterator/IteratorUtilsTest.java        | 29 ++++++++++++++++++--
 1 file changed, 27 insertions(+), 2 deletions(-)
----------------------------------------------------------------------


http://git-wip-us.apache.org/repos/asf/incubator-tinkerpop/blob/56b5e352/gremlin-test/src/test/java/org/apache/tinkerpop/gremlin/util/iterator/IteratorUtilsTest.java
----------------------------------------------------------------------
diff --git a/gremlin-test/src/test/java/org/apache/tinkerpop/gremlin/util/iterator/IteratorUtilsTest.java b/gremlin-test/src/test/java/org/apache/tinkerpop/gremlin/util/iterator/IteratorUtilsTest.java
index f039c4e..0e2c2f0 100644
--- a/gremlin-test/src/test/java/org/apache/tinkerpop/gremlin/util/iterator/IteratorUtilsTest.java
+++ b/gremlin-test/src/test/java/org/apache/tinkerpop/gremlin/util/iterator/IteratorUtilsTest.java
@@ -169,8 +169,7 @@ public class IteratorUtilsTest {
 
     @Test
     public void shouldCountEmpty() {
-        final List<String> iterable = new ArrayList<>();
-        assertEquals(0, IteratorUtils.count(iterable.iterator()));
+        assertEquals(0, IteratorUtils.count(new ArrayList<>().iterator()));
     }
 
     @Test
@@ -246,6 +245,32 @@ public class IteratorUtilsTest {
         assertFalse(IteratorUtils.noneMatch(iterable.iterator(), s -> s.startsWith("test")));
     }
 
+    @Test
+    public void shouldProduceMapFromIteratorUsingIdentityForValue() {
+        final List<String> iterable = new ArrayList<>();
+        iterable.add("test1");
+        iterable.add("test2");
+        iterable.add("test3");
+
+        final Map<String,String> m = IteratorUtils.collectMap(iterable.iterator(), k -> k.substring(4));
+        assertEquals("test1", m.get("1"));
+        assertEquals("test2", m.get("2"));
+        assertEquals("test3", m.get("3"));
+    }
+
+    @Test
+    public void shouldProduceMapFromIterator() {
+        final List<String> iterable = new ArrayList<>();
+        iterable.add("test1");
+        iterable.add("test2");
+        iterable.add("test3");
+
+        final Map<String,String> m = IteratorUtils.collectMap(iterable.iterator(), k -> k.substring(4), v -> v.substring(0, 4));
+        assertEquals("test", m.get("1"));
+        assertEquals("test", m.get("2"));
+        assertEquals("test", m.get("3"));
+    }
+
     public <S> void assertIterator(final Iterator<S> itty, final int size) {
         for (int ix = 0; ix < size; ix++) {
             assertEquals("test" + (ix + 1), itty.next());