def test_get_reservations_by_host_id(self):
self._setup_leases()
self.check_reservation([], 'r1', '2030-01-01 07:00',
'2030-01-01 08:59')
ret = db_api.reservation_get_all_by_lease_id('lease1')
self.check_reservation(ret, 'r1', '2030-01-01 08:00',
'2030-01-01 10:00')
ret = db_api.reservation_get_all_by_lease_id('lease1')
ret.extend(db_api.reservation_get_all_by_lease_id('lease3'))
self.check_reservation(ret, 'r1', '2030-01-01 08:00',
'2030-01-01 15:30')
self.check_reservation([], 'r4', '2030-01-01 07:00',
'2030-01-01 15:00')
def _create_physical_lease(values=_get_fake_phys_lease_values(), random=False):
"""Creating fake lease having a single physical resource."""
if random is True:
values = _get_fake_phys_lease_values(id=_get_fake_random_uuid(),
name=_get_fake_random_uuid())
lease = db_api.lease_create(values)
for reservation in db_api.reservation_get_all_by_lease_id(lease['id']):
allocation_values = {
'id': _get_fake_random_uuid(),
'compute_host_id': reservation['resource_id'],
'reservation_id': reservation['id']
}
db_api.host_allocation_create(allocation_values)
return lease
def test_get_reservation_allocations_by_host_ids_with_reservation_id(self):
self._setup_leases()
reservation1 = db_api.reservation_get_all_by_lease_id('lease1')[0]
# query allocations of lease1
expected = [
self._create_allocation_tuple('lease1'),
]
ret = db_utils.get_reservation_allocations_by_host_ids(
['r1', 'r2'],
'2030-01-01 08:00',
'2030-01-01 15:00',
reservation_id=reservation1['id'])
self.assertListEqual(expected, ret)
def test_get_reservation_allocations_by_host_ids_with_reservation_id(self):
def create_allocation_tuple(lease_id):
reservation = db_api.reservation_get_all_by_lease_id(lease_id)[0]
allocation = db_api.host_allocation_get_all_by_values(
reservation_id=reservation['id'])[0]
return (reservation['id'], allocation['id'])
self._setup_leases()
reservation1 = db_api.reservation_get_all_by_lease_id('lease1')[0]
# query allocations of lease1
expected = [
create_allocation_tuple('lease1'),
]
ret = db_utils.get_reservation_allocations_by_host_ids(
['r1', 'r2'],
'2030-01-01 08:00',
'2030-01-01 15:00',
reservation_id=reservation1['id'])
self.assertListEqual(expected, [(r['id'], a['id']) for r, a in ret])
def _create_physical_lease(values=_get_fake_phys_lease_values(), random=False):
"""Creating fake lease having a single physical resource."""
if random is True:
values = _get_fake_phys_lease_values(id=_get_fake_random_uuid(),
name=_get_fake_random_uuid())
lease = db_api.lease_create(values)
phys_res = _get_fake_phys_reservation_values()
for reservation in db_api.reservation_get_all_by_lease_id(lease['id']):
allocation_values = {
'id': _get_fake_random_uuid(),
'compute_host_id': values['reservations'][0]['resource_id'],
'reservation_id': reservation['id']
}
db_api.host_allocation_create(allocation_values)
computehost_reservations = {
'id': _get_fake_random_uuid(),
'reservation_id': values['reservations'][0]['id'],
'resource_properties': phys_res['resource_properties'],
'hypervisor_properties': phys_res['hypervisor_properties'],
'count_range': "{0} - {1}".format(phys_res['min'], phys_res['max'])
}
db_api.host_reservation_create(computehost_reservations)
return lease
def _create_allocation_tuple(self, lease_id):
reservation = db_api.reservation_get_all_by_lease_id(lease_id)[0]
allocation = db_api.host_allocation_get_all_by_values(
reservation_id=reservation['id'])[0]
return (reservation['id'], reservation['lease_id'],
allocation['compute_host_id'])
def create_allocation_tuple(lease_id):
reservation = db_api.reservation_get_all_by_lease_id(lease_id)[0]
allocation = db_api.host_allocation_get_all_by_values(
reservation_id=reservation['id'])[0]
return (reservation['id'], allocation['id'])
