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'])
