From 629d5f6ebea50cd7f9861a9d5988b0f969d969d1 Mon Sep 17 00:00:00 2001
From: Pawas Chhokra <pchhokra@linkedin.com>
Date: Tue, 17 Aug 2021 15:53:20 -0700
Subject: [PATCH] Remove test implementation of SamzaIORegistrar

---
 .../samza/util/PipelineJsonRendererTest.java  | 21 -------------------
 .../samza/src/test/resources/ExpectedDag.json |  1 -
 2 files changed, 22 deletions(-)

diff --git a/runners/samza/src/test/java/org/apache/beam/runners/samza/util/PipelineJsonRendererTest.java b/runners/samza/src/test/java/org/apache/beam/runners/samza/util/PipelineJsonRendererTest.java
index f428fa633035..70812d69fe6d 100644
--- a/runners/samza/src/test/java/org/apache/beam/runners/samza/util/PipelineJsonRendererTest.java
+++ b/runners/samza/src/test/java/org/apache/beam/runners/samza/util/PipelineJsonRendererTest.java
@@ -19,18 +19,15 @@
 
 import static org.junit.Assert.assertEquals;
 
-import com.google.auto.service.AutoService;
 import com.google.gson.JsonParser;
 import java.io.IOException;
 import java.nio.charset.StandardCharsets;
 import java.nio.file.Files;
 import java.nio.file.Paths;
-import java.util.Optional;
 import org.apache.beam.runners.samza.SamzaPipelineOptions;
 import org.apache.beam.runners.samza.SamzaRunner;
 import org.apache.beam.sdk.Pipeline;
 import org.apache.beam.sdk.options.PipelineOptionsFactory;
-import org.apache.beam.sdk.runners.TransformHierarchy;
 import org.apache.beam.sdk.transforms.Create;
 import org.apache.beam.sdk.transforms.Sum;
 import org.apache.beam.sdk.transforms.windowing.FixedWindows;
@@ -54,7 +51,6 @@ public void testEmptyPipeline() {
     String jsonDag =
         "{  \"RootNode\": ["
             + "    { \"fullName\":\"OuterMostNode\","
-            + "      \"ioInfo\":\"TestTopic\","
             + "      \"ChildNodes\":[    ]}],\"graphLinks\": []"
             + "}";
 
@@ -85,21 +81,4 @@ public void testCompositePipeline() throws IOException {
         JsonParser.parseString(
             PipelineJsonRenderer.toJsonString(p).replaceAll(System.lineSeparator(), "")));
   }
-
-  @AutoService(PipelineJsonRenderer.SamzaIORegistrar.class)
-  public static class Registrar implements PipelineJsonRenderer.SamzaIORegistrar {
-
-    @Override
-    public PipelineJsonRenderer.SamzaIOInfo getSamzaIO() {
-      return new PipelineJsonRenderer.SamzaIOInfo() {
-        @Override
-        public Optional<String> getIOInfo(TransformHierarchy.Node node) {
-          if (node.isRootNode()) {
-            return Optional.of("TestTopic");
-          }
-          return Optional.empty();
-        }
-      };
-    }
-  }
 }
diff --git a/runners/samza/src/test/resources/ExpectedDag.json b/runners/samza/src/test/resources/ExpectedDag.json
index 8f45ec8252ee..966aaf85a541 100644
--- a/runners/samza/src/test/resources/ExpectedDag.json
+++ b/runners/samza/src/test/resources/ExpectedDag.json
@@ -1,7 +1,6 @@
 {
   "RootNode": [
     { "fullName":"OuterMostNode",
-      "ioInfo":"TestTopic",
       "ChildNodes":[
         { "fullName":"Create.TimestampedValues",
           "enclosingNode":"OuterMostNode",