def _delete_heartbeat(self, instance_id):
agent_heart_beat = agent_models.AgentHeartBeat()
try:
heartbeat = agent_heart_beat.find_by_instance_id(instance_id)
heartbeat.delete()
except exception.ModelNotFoundError as e:
LOG.error(e.message)
def test_save(self, dmb_mock):
utils.utcnow = Mock()
dbmodels.get_db_api = MagicMock(
return_value=dbmodels.DatabaseModelBase)
dbapi.save = Mock()
dbmodels.DatabaseModelBase.is_valid = Mock(return_value=True)
self.heartBeat = models.AgentHeartBeat()
self.heartBeat.save()
self.assertEqual(1, utils.utcnow.call_count)
def test_is_active(self):
models.AGENT_HEARTBEAT = 10000000000
mock = models.AgentHeartBeat()
models.AgentHeartBeat.__setitem__(mock, 'updated_at', datetime.now())
self.assertTrue(models.AgentHeartBeat.is_active(mock))
