[beam] branch spark-runner_structured-streaming updated: Fix for test elements container in GroupByKeyTest

[aromanenko]

This is an automated email from the ASF dual-hosted git repository.

aromanenko pushed a commit to branch spark-runner_structured-streaming
in repository https://gitbox.apache.org/repos/asf/beam.git


The following commit(s) were added to 
refs/heads/spark-runner_structured-streaming by this push:
     new dff0931  Fix for test elements container in GroupByKeyTest
dff0931 is described below

commit dff0931950c04f086d3e2d2d7bf33d3d54958c79
Author: Alexey Romanenko <aromanenko....@gmail.com>
AuthorDate: Fri Feb 8 18:44:06 2019 +0100

    Fix for test elements container in GroupByKeyTest
---
 .../translation/batch/GroupByKeyTest.java            | 20 ++++++++++----------
 1 file changed, 10 insertions(+), 10 deletions(-)

diff --git 
a/runners/spark-structured-streaming/src/test/java/org/apache/beam/runners/spark/structuredstreaming/translation/batch/GroupByKeyTest.java
 
b/runners/spark-structured-streaming/src/test/java/org/apache/beam/runners/spark/structuredstreaming/translation/batch/GroupByKeyTest.java
index a069534..b772070 100644
--- 
a/runners/spark-structured-streaming/src/test/java/org/apache/beam/runners/spark/structuredstreaming/translation/batch/GroupByKeyTest.java
+++ 
b/runners/spark-structured-streaming/src/test/java/org/apache/beam/runners/spark/structuredstreaming/translation/batch/GroupByKeyTest.java
@@ -18,8 +18,8 @@
 package org.apache.beam.runners.spark.structuredstreaming.translation.batch;
 
 import java.io.Serializable;
-import java.util.HashMap;
-import java.util.Map;
+import java.util.ArrayList;
+import java.util.List;
 import org.apache.beam.runners.spark.structuredstreaming.SparkPipelineOptions;
 import org.apache.beam.runners.spark.structuredstreaming.SparkRunner;
 import org.apache.beam.sdk.Pipeline;
@@ -31,7 +31,6 @@ import org.apache.beam.sdk.transforms.ParDo;
 import org.apache.beam.sdk.values.KV;
 import org.apache.beam.sdk.values.PCollection;
 import org.junit.BeforeClass;
-import org.junit.Ignore;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
@@ -50,13 +49,14 @@ public class GroupByKeyTest implements Serializable {
 
   @Test
   public void testGroupByKey() {
-    Map<Integer, Integer> elems = new HashMap<>();
-    elems.put(1, 1);
-    elems.put(1, 3);
-    elems.put(1, 5);
-    elems.put(2, 2);
-    elems.put(2, 4);
-    elems.put(2, 6);
+    List<KV<Integer, Integer>> elems = new ArrayList<>();
+    elems.add(KV.of(1, 1));
+    elems.add(KV.of(1, 3));
+    elems.add(KV.of(1, 5));
+    elems.add(KV.of(2, 2));
+    elems.add(KV.of(2, 4));
+    elems.add(KV.of(2, 6));
+
     PCollection<KV<Integer, Integer>> input = pipeline.apply(Create.of(elems));
     input.apply(GroupByKey.create());
     pipeline.run();