@Test public void testStop() { final String storeName = "test-stop-store"; final String taskName = "test-stop-task"; TaskSideInputStorageManager testSideInputStorageManager = new MockTaskSideInputStorageManagerBuilder(taskName, NON_LOGGED_STORE_DIR) .addInMemoryStore(storeName, ImmutableSet.of()) .build(); initializeSideInputStorageManager(testSideInputStorageManager); testSideInputStorageManager.stop(); verify(testSideInputStorageManager.getStore(storeName)).stop(); verify(testSideInputStorageManager).writeOffsetFiles(); }