@Override protected void handleClose() { this.transformFn.close(); }
@Override protected void handleClose() { this.transformFn.close(); }
@Override protected void handleClose() { this.transformFn.close(); }
@Override protected void handleClose() { this.transformFn.close(); }
@Override protected void handleClose() { this.transformFn.close(); }
@Test public void testStreamOperatorClose() { StreamOperatorSpec<TestMessageEnvelope, TestOutputMessageEnvelope> mockOp = mock(StreamOperatorSpec.class); FlatMapFunction<TestMessageEnvelope, TestOutputMessageEnvelope> txfmFn = mock(FlatMapFunction.class); when(mockOp.getTransformFn()).thenReturn(txfmFn); StreamOperatorImpl<TestMessageEnvelope, TestOutputMessageEnvelope> opImpl = new StreamOperatorImpl<>(mockOp); // ensure that close is not called yet verify(txfmFn, times(0)).close(); opImpl.handleClose(); // ensure that close is called once inside handleClose() verify(txfmFn, times(1)).close(); } }