def test_heartbeat(self): task = FakeTask() fake_node = FakeNode() kwargs = { 'agent_url': 'http://127.0.0.1:9999/bar' } with tests.mock_now(self.fake_datetime): node = self.vendor._heartbeat(task, fake_node, **kwargs) self.assertEqual(self.fake_datetime, node.instance_info['last_heartbeat']) self.assertEqual('http://127.0.0.1:9999/bar', node.instance_info['agent_url'])
def test_heartbeat_no_uuid(self, find_mock): kwargs = { 'hardware': [ { 'id': 'aa:bb:cc:dd:ee:ff', 'type': 'mac_address' }, { 'id': 'ff:ee:dd:cc:bb:aa', 'type': 'mac_address' } ] } expected_node = FakeNode(uuid='heartbeat') find_mock.return_value = expected_node with tests.mock_now(self.fake_datetime): node = self.vendor._heartbeat_no_uuid(FakeTask(), **kwargs) self.assertEqual(expected_node, node['node'])