make TestSamzaRunner always use same ConfigMap

Signed-off-by: Alan Zhang <[email protected]>
linkedin · Apr 3, 2024 · 9f558e2 · 9f558e2
1 parent 24c33a2
commit 9f558e2
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/runners/samza/src/main/java/org/apache/beam/runners/samza/TestSamzaRunner.java b/runners/samza/src/main/java/org/apache/beam/runners/samza/TestSamzaRunner.java
@@ -54,10 +54,10 @@ public static SamzaPipelineOptions createSamzaPipelineOptions(
       config.put(JOB_NON_LOGGED_STORE_BASE_DIR, storeDir.getAbsolutePath());
       config.put(JOB_JMX_ENABLED, "false");
 
-      if (samzaOptions.getConfigOverride() != null) {
-        config.putAll(samzaOptions.getConfigOverride());
+      if (samzaOptions.getConfigOverride() == null) {
+        samzaOptions.setConfigOverride(new HashMap<>());
       }
-      samzaOptions.setConfigOverride(config);
+      samzaOptions.getConfigOverride().putAll(config);
       return samzaOptions;
     } catch (Exception e) {
       throw new RuntimeException(e);