def test_send_beat_without_creating_service(self, mock_save, mock_list):
mock_list.return_value = [objects.Service(mock.Mock(),
name='watcher-service',
host='controller')]
service.ServiceHeartbeat(service_name='watcher-service')
self.assertEqual(1, mock_save.call_count)
def test_send_beat_with_creating_service(self, mock_create,
mock_list):
CONF.set_default('host', 'fake-fqdn')
mock_list.return_value = []
service.ServiceHeartbeat(service_name='watcher-service')
mock_list.assert_called_once_with(mock.ANY,
filters={'name': 'watcher-service',
'host': 'fake-fqdn'})
self.assertEqual(1, mock_create.call_count)