Move MemoryByteSink to test

Author: misolt

dumper/app/src/main/java/com/google/edwmigration/dumper/application/dumper/task/AbstractTask.java

@@ -28,7 +28,6 @@
 import com.google.edwmigration.dumper.application.dumper.io.OutputHandle.WriteMode;
 import com.google.errorprone.annotations.ForOverride;
 import java.beans.PropertyDescriptor;
-import java.io.ByteArrayOutputStream;
 import java.io.OutputStream;
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -187,28 +186,6 @@ public OutputStream openStream() {
     }
   }
 
-  public static final class MemoryByteSink extends ByteSink {
-
-    private final ByteArrayOutputStream out;
-
-    public MemoryByteSink() {
-      this(new ByteArrayOutputStream());
-    }
-
-    private MemoryByteSink(ByteArrayOutputStream out) {
-      this.out = out;
-    }
-
-    @Override
-    public OutputStream openStream() {
-      return out;
-    }
-
-    public String getContent() {
-      return out.toString();
-    }
-  }
-
   @AutoValue
   @ParametersAreNonnullByDefault
   public abstract static class TaskOptions {

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/connector/hadoop/oozie/OozieWorkflowJobsTaskTest.java

@@ -40,7 +40,7 @@
 import com.github.tomakehurst.wiremock.core.WireMockConfiguration;
 import com.github.tomakehurst.wiremock.extension.responsetemplating.ResponseTemplateTransformer;
 import com.google.edwmigration.dumper.application.dumper.ConnectorArguments;
-import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.MemoryByteSink;
+import com.google.edwmigration.dumper.application.dumper.task.MemoryByteSink;
 import com.google.edwmigration.dumper.application.dumper.task.TaskRunContext;
 import java.io.IOException;
 import java.net.URISyntaxException;

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/connector/ranger/AbstractRangerTaskTest.java

@@ -22,7 +22,7 @@
 
 import com.google.edwmigration.dumper.application.dumper.connector.ranger.RangerClient.ConnectionWrapper;
 import com.google.edwmigration.dumper.application.dumper.connector.ranger.RangerConnector.RangerClientHandle;
-import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.MemoryByteSink;
+import com.google.edwmigration.dumper.application.dumper.task.MemoryByteSink;
 import java.io.IOException;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/connector/redshift/RedshiftClusterNodesTaskTest.java

@@ -26,7 +26,7 @@
 import com.amazonaws.services.redshift.model.Endpoint;
 import com.google.common.collect.ImmutableList;
 import com.google.edwmigration.dumper.application.dumper.handle.RedshiftHandle;
-import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.MemoryByteSink;
+import com.google.edwmigration.dumper.application.dumper.task.MemoryByteSink;
 import java.util.Optional;
 import org.junit.Rule;
 import org.junit.Test;

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/connector/redshift/RedshiftClusterUsageMetricsTaskTest.java

@@ -34,7 +34,7 @@
 import com.google.edwmigration.dumper.application.dumper.connector.redshift.RedshiftClusterUsageMetricsTask.MetricName;
 import com.google.edwmigration.dumper.application.dumper.connector.redshift.RedshiftClusterUsageMetricsTask.MetricType;
 import com.google.edwmigration.dumper.application.dumper.handle.RedshiftHandle;
-import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.MemoryByteSink;
+import com.google.edwmigration.dumper.application.dumper.task.MemoryByteSink;
 import com.google.edwmigration.dumper.plugin.lib.dumper.spi.RedshiftRawLogsDumpFormat;
 import java.time.ZoneId;
 import java.time.ZonedDateTime;

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/task/DumpMetadataTaskTest.java

@@ -20,7 +20,6 @@
 import static org.mockito.Mockito.mock;
 
 import com.google.edwmigration.dumper.application.dumper.ConnectorArguments;
-import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.MemoryByteSink;
 import java.io.IOException;
 import org.junit.Test;
 import org.junit.runner.RunWith;

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/task/FormatTaskTest.java

@@ -19,7 +19,6 @@
 import static org.junit.Assert.assertEquals;
 import static org.mockito.Mockito.mock;
 
-import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.MemoryByteSink;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/task/JdbcSelectTaskTest.java

@@ -22,7 +22,6 @@
 
 import com.google.edwmigration.dumper.application.dumper.handle.JdbcHandle;
 import com.google.edwmigration.dumper.application.dumper.io.OutputHandle.WriteMode;
-import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.MemoryByteSink;
 import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.TaskOptions;
 import com.google.edwmigration.dumper.application.dumper.test.DumperTestUtils;
 import java.io.File;

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/task/MemoryByteSink.java

@@ -0,0 +1,35 @@
+/*
+ * Copyright 2022-2025 Google LLC
+ * Copyright 2013-2021 CompilerWorks
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.google.edwmigration.dumper.application.dumper.task;
+
+import com.google.common.io.ByteSink;
+import java.io.ByteArrayOutputStream;
+import java.io.OutputStream;
+
+public final class MemoryByteSink extends ByteSink {
+
+  private final ByteArrayOutputStream out = new ByteArrayOutputStream();
+
+  @Override
+  public OutputStream openStream() {
+    return out;
+  }
+
+  public String getContent() {
+    return out.toString();
+  }
+}

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/task/VersionTaskTest.java

@@ -19,7 +19,6 @@
 import static org.junit.Assert.assertEquals;
 import static org.mockito.Mockito.mock;
 
-import com.google.edwmigration.dumper.application.dumper.task.AbstractTask.MemoryByteSink;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;