AHeise commented on a change in pull request #11515: [FLINK-16744][task]
implement channel state persistence for unaligned checkpoints
URL: https://github.com/apache/flink/pull/11515#discussion_r400931309
##########
File path:
flink-table/flink-table-runtime-blink/src/test/java/org/apache/flink/table/runtime/operators/over/BufferDataOverWindowOperatorTest.java
##########
@@ -205,9 +206,19 @@ public StreamConfig getOperatorConfig() {
@Override
public StreamTask<?, ?> getContainingTask() {
- StreamTask task = mock(StreamTask.class);
+ class MockableStreamTask extends
StreamTask<Object, StreamOperator<Object>> implements EnvironmentSupport {
+ MockableStreamTask() throws Exception {
+ super(new
MockEnvironmentBuilder().build());
+ }
+
+ @Override
+ protected void init() {
+ }
+ }
+
+ StreamTask<?, ?> task =
mock(MockableStreamTask.class);
Environment env = mock(Environment.class);
- when(task.getEnvironment()).thenReturn(env);
+ when(task.getEnvironment()).thenReturn(new
MockEnvironmentBuilder().build());
when(env.getMemoryManager()).thenReturn(memoryManager);
Review comment:
Couldn't you just override the method on the `MockableStreamTask` to get rid
of mockito?
----------------------------------------------------------------
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
For queries about this service, please contact Infrastructure at:
us...@infra.apache.org
With regards,
Apache Git Services
