@Override SamzaContainer createSamzaContainer(String processorId, JobModel jobModel) { if (container == null) { RunLoop mockRunLoop = mock(RunLoop.class); doAnswer(invocation -> { try { runLoopStartForMain.countDown(); containerStop.await(); } catch (InterruptedException e) { System.out.println("In exception" + e); e.printStackTrace(); } return null; }).when(mockRunLoop).run(); Mockito.doAnswer(invocation -> { containerStop.countDown(); return null; }).when(mockRunLoop).shutdown(); container = StreamProcessorTestUtils.getDummyContainer(mockRunLoop, Mockito.mock(StreamTask.class)); } return container; }