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