@Test public void testStatusManagerHeartbeats() throws IOException { StatusManager stats = new StatusManager(); FlumeConfiguration cfg = FlumeConfiguration.createTestableConfiguration(); Clock.resetDefault(); cfg.set(FlumeConfiguration.WEBAPPS_PATH, "build/webapps"); cfg.set(FlumeConfiguration.MASTER_STORE, "memory"); // avoiding gossip ack manager until it shuts down cleanly. ConfigStore cfgStore = FlumeMaster.createConfigStore(cfg); FlumeMaster fm = new FlumeMaster( new CommandManager(), new ConfigManager(cfgStore), stats, new MasterAckManager(), cfg); fm.getSpecMan().addLogicalNode("physnode", "foo"); // foo is in state HELLO and has configuration numbered 0 long prev = Clock.unixTime(); boolean needsRefresh = stats.updateHeartbeatStatus(NetUtils.localhost(), "physnode", "foo", NodeState.HELLO, 0); LOG.info(stats.getNodeStatuses()); assertTrue(needsRefresh); StatusManager.NodeStatus ns = stats.getNodeStatuses().get("foo"); assertEquals(0, ns.version); assertTrue(prev <= ns.lastseen); assertEquals(NodeState.HELLO, ns.state); prev = ns.lastseen; needsRefresh = stats.updateHeartbeatStatus(NetUtils.localhost(), "physnode", "foo", NodeState.IDLE, 0); LOG.info(stats.getNodeStatuses()); assertFalse(needsRefresh); // no new data flow version assertEquals(0, ns.version); assertTrue(prev <= ns.lastseen); assertEquals(NodeState.IDLE, ns.state); prev = ns.lastseen; needsRefresh = stats.updateHeartbeatStatus(NetUtils.localhost(), "physnode", "foo", NodeState.ACTIVE, 10); LOG.info(stats.getNodeStatuses()); assertFalse(needsRefresh); // node has updated version number, but master // has not assertEquals(10, ns.version); assertTrue(prev <= ns.lastseen); assertEquals(NodeState.ACTIVE, ns.state); prev = ns.lastseen; // same message, no refresh needsRefresh = stats.updateHeartbeatStatus(NetUtils.localhost(), "physnode", "foo", NodeState.ACTIVE, 10); LOG.info(stats.getNodeStatuses()); assertFalse(needsRefresh); // this is wierd, is this right? assertEquals(10, ns.version); assertTrue(prev <= ns.lastseen); assertEquals(NodeState.ACTIVE, ns.state); prev = ns.lastseen; // regress to an older version, needsRefresh = stats.updateHeartbeatStatus(NetUtils.localhost(), "physnode", "foo", NodeState.ACTIVE, 5); LOG.info(stats.getNodeStatuses()); assertFalse(needsRefresh); assertEquals(5, ns.version); assertTrue(prev <= ns.lastseen); assertEquals(NodeState.ACTIVE, ns.state); prev = ns.lastseen; LOG.info(stats.getReport()); LOG.info(stats.getName()); }