Testing FlinkKinesisConsumer how to sync data creation and stream consuming

Question

I have a simple method that creates a stream of X records. but when i try to test it, the stream ends before it is consumed (Test A). However, if i call that method after th