[BEAM-14081] Make BatchContextImpl class abstract

apache · May 11, 2022 · 2f77cc4 · 2f77cc4
1 parent 9e31a72
commit 2f77cc4
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/sdks/java/io/cdap/src/main/java/org/apache/beam/sdk/io/cdap/context/BatchContextImpl.java b/sdks/java/io/cdap/src/main/java/org/apache/beam/sdk/io/cdap/context/BatchContextImpl.java
@@ -43,7 +43,7 @@
 
 /** Class for Batch, Sink and Stream CDAP wrapper classes that use it to provide common details. */
 @SuppressWarnings({"TypeParameterUnusedInFormals", "nullness"})
-public class BatchContextImpl implements BatchContext {
+public abstract class BatchContextImpl implements BatchContext {
 
   private final FailureCollectorWrapper failureCollector = new FailureCollectorWrapper();
 

diff --git a/.../java/io/cdap/src/test/java/org/apache/beam/sdk/io/cdap/context/BatchContextImplTest.java b/.../java/io/cdap/src/test/java/org/apache/beam/sdk/io/cdap/context/BatchContextImplTest.java
@@ -35,7 +35,7 @@ public class BatchContextImplTest {
   public void getLogicalStartTime() {
     /** arrange */
     Timestamp expectedStartTime = new Timestamp(System.currentTimeMillis());
-    BatchContextImpl context = new BatchContextImpl();
+    BatchContextImpl context = new BatchSourceContextImpl();
 
     /** act */
     long actualStartTime = context.getLogicalStartTime();
@@ -47,7 +47,7 @@ public void getLogicalStartTime() {
   @Test
   public void getFailureCollector() {
     /** arrange */
-    BatchContextImpl context = new BatchContextImpl();
+    BatchContextImpl context = new BatchSinkContextImpl();
 
     /** act */
     FailureCollector failureCollector = context.getFailureCollector();