@Test public void testLocalityManager() {
LocalityManager localityManager = new LocalityManager(config, new MetricsRegistryMap());
localityManager.writeContainerToHostMapping("0", "localhost");
Map<String, Map<String, String>> localMap = localityManager.readContainerLocality();
Map<String, Map<String, String>> expectedMap =
new HashMap<String, Map<String, String>>() {
{
this.put("0",
new HashMap<String, String>() {
{
this.put(SetContainerHostMapping.HOST_KEY, "localhost");
}
});
}
};
assertEquals(expectedMap, localMap);
localityManager.close();
MockCoordinatorStreamSystemProducer producer = mockCoordinatorStreamSystemFactory.getCoordinatorStreamSystemProducer(config, null);
MockCoordinatorStreamSystemConsumer consumer = mockCoordinatorStreamSystemFactory.getCoordinatorStreamSystemConsumer(config, null);
assertTrue(producer.isStopped());
assertTrue(consumer.isStopped());
}