taskAssignmentManager.close();
taskAssignmentManager.close();
taskAssignmentManager.close();
taskAssignmentManager.close();
@Test public void testTaskAssignmentManager() { TaskAssignmentManager taskAssignmentManager = new TaskAssignmentManager(config, new MetricsRegistryMap()); Map<String, String> expectedMap = ImmutableMap.of("Task0", "0", "Task1", "1", "Task2", "2", "Task3", "0", "Task4", "1"); for (Map.Entry<String, String> entry : expectedMap.entrySet()) { taskAssignmentManager.writeTaskContainerMapping(entry.getKey(), entry.getValue(), TaskMode.Active); } Map<String, String> localMap = taskAssignmentManager.readTaskAssignment(); assertEquals(expectedMap, localMap); taskAssignmentManager.close(); }
@Test public void testDeleteMappings() { TaskAssignmentManager taskAssignmentManager = new TaskAssignmentManager(config, new MetricsRegistryMap()); Map<String, String> expectedMap = ImmutableMap.of("Task0", "0", "Task1", "1"); for (Map.Entry<String, String> entry : expectedMap.entrySet()) { taskAssignmentManager.writeTaskContainerMapping(entry.getKey(), entry.getValue(), TaskMode.Active); } Map<String, String> localMap = taskAssignmentManager.readTaskAssignment(); assertEquals(expectedMap, localMap); taskAssignmentManager.deleteTaskContainerMappings(localMap.keySet()); Map<String, String> deletedMap = taskAssignmentManager.readTaskAssignment(); assertTrue(deletedMap.isEmpty()); taskAssignmentManager.close(); }
@Test public void testTaskAssignmentManagerEmptyCoordinatorStream() { TaskAssignmentManager taskAssignmentManager = new TaskAssignmentManager(config, new MetricsRegistryMap()); Map<String, String> expectedMap = new HashMap<>(); Map<String, String> localMap = taskAssignmentManager.readTaskAssignment(); assertEquals(expectedMap, localMap); taskAssignmentManager.close(); } }