def test__expire_leases_error(self, mock_ga, mock_utcnow, mock_expire,
mock_save):
error_lease = lease.Lease(
offer_uuid=self.test_offer.uuid,
name='c',
uuid=uuidutils.generate_uuid(),
project_id='lesseeid',
status=statuses.CREATED,
start_time=datetime.datetime(3000, 7, 16),
end_time=datetime.datetime(4000, 7, 16),
)
mock_ga.return_value = [error_lease]
mock_utcnow.return_value = datetime.datetime(5000, 7, 16)
mock_expire.side_effect = Exception('whoops')
s = ManagerService()
s._expire_leases()
mock_expire.assert_called_once()
mock_ga.assert_called_once_with(
{
'status': [
statuses.ACTIVE, statuses.CREATED, statuses.WAIT_EXPIRE,
statuses.WAIT_FULFILL
]
}, s._context)
self.assertEqual(statuses.ERROR, error_lease.status)
mock_save.assert_called_once()
def test__cancel_leases(self, mock_ga, mock_utcnow, mock_cancel):
mock_ga.return_value = [self.test_lease, self.test_lease]
s = ManagerService()
s._cancel_leases()
assert mock_cancel.call_count == 2
mock_ga.assert_called_once_with({'status': [statuses.WAIT_CANCEL]},
s._context)
def test__fulfill_leases(self, mock_ga, mock_utcnow, mock_fulfill):
mock_ga.return_value = [self.test_lease, self.test_lease]
mock_utcnow.return_value = datetime.datetime(3500, 7, 16)
s = ManagerService()
s._fulfill_leases()
assert mock_fulfill.call_count == 2
mock_ga.assert_called_once_with(
{'status': [statuses.CREATED, statuses.WAIT_FULFILL]}, s._context)
def test__expire_offers(self, mock_ga, mock_utcnow, mock_expire):
mock_ga.return_value = [self.test_offer, self.test_offer]
mock_utcnow.return_value = datetime.datetime(5000, 7, 16)
s = ManagerService()
s._expire_offers()
assert mock_expire.call_count == 2
mock_ga.assert_called_once_with({'status': statuses.OFFER_CAN_DELETE},
s._context)
def test__expire_offers(mock_ga, mock_log, mock_utcnow):
mock_ga.return_value = [test_offer]
mock_utcnow.return_value = datetime.datetime(3500, 7, 16)
s = ManagerService()
with mock.patch.object(test_offer, 'expire') as mock_expire:
s._expire_offers()
mock_expire.assert_called_once()
mock_ga.assert_called_once()
mock_log.assert_called_once()
def test__fulfill_contracts(mock_ga, mock_log, mock_utcnow):
mock_ga.return_value = [test_contract]
mock_utcnow.return_value = datetime.datetime(3500, 7, 16)
s = ManagerService()
with mock.patch.object(test_contract, 'fulfill') as mock_fulfill:
s._fulfill_contracts()
mock_fulfill.assert_called_once()
mock_ga.assert_called_once()
mock_log.assert_called_once()
def test__expire_offers_error(self, mock_ga, mock_utcnow, mock_expire,
mock_save):
error_offer = offer.Offer(resource_type='test_node',
resource_uuid='abc',
name='o',
uuid=uuidutils.generate_uuid(),
status=statuses.AVAILABLE,
start_time=datetime.datetime(3000, 7, 16),
end_time=datetime.datetime(4000, 7, 16),
project_id='ownerid')
mock_ga.return_value = [error_offer]
mock_utcnow.return_value = datetime.datetime(5000, 7, 16)
mock_expire.side_effect = Exception('whoops')
s = ManagerService()
s._expire_offers()
mock_expire.assert_called_once()
mock_ga.assert_called_once_with({'status': statuses.OFFER_CAN_DELETE},
s._context)
self.assertEqual(statuses.ERROR, error_offer.status)
mock_save.assert_called_once()