Add equals/hashcode to test data class to avoid direct runner warning…

… log (#32883)
apache · Nov 7, 2024 · c7e4db7 · c7e4db7
1 parent 3c8bb54
commit c7e4db7
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/WithKeysTest.java b/sdks/java/core/src/test/java/org/apache/beam/sdk/transforms/WithKeysTest.java
@@ -22,6 +22,7 @@
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.List;
+import java.util.Objects;
 import org.apache.beam.sdk.coders.Coder;
 import org.apache.beam.sdk.coders.KvCoder;
 import org.apache.beam.sdk.coders.StringUtf8Coder;
@@ -226,5 +227,19 @@ public long getNum() {
     public String getStr() {
       return this.str;
     }
+
+    @Override
+    public boolean equals(Object o) {
+      if (!(o instanceof Pojo)) {
+        return false;
+      }
+      Pojo pojo = (Pojo) o;
+      return num == pojo.num && Objects.equals(str, pojo.str);
+    }
+
+    @Override
+    public int hashCode() {
+      return Objects.hash(num, str);
+    }
   }
 }