How to use

close

method

in

org.apache.samza.operators.functions.FlatMapFunction

@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();
 }
}