def test_reservation_get_all_by_values(self):
"""Create 2 reservations and check find abilities
Create two reservations and verify that we can find reservation per
resource_id or resource_type.
"""
db_api.reservation_create(
_get_fake_phys_reservation_values(id='1', resource_id='1234'))
db_api.reservation_create(
_get_fake_phys_reservation_values(id='2', resource_id='5678'))
self.assertEqual(2, len(db_api.reservation_get_all_by_values()))
self.assertEqual(
1, len(db_api.reservation_get_all_by_values(resource_id='5678')))
self.assertEqual(
2,
len(
db_api.reservation_get_all_by_values(
resource_type=host_plugin.RESOURCE_TYPE)))
def test_get_reservations_by_host_id_with_multi_reservation(self):
self._setup_leases()
fake_lease = _get_fake_phys_lease_values(
id='lease-4',
name='fake_phys_lease_r4',
start_date=_get_datetime('2030-01-01 15:00'),
end_date=_get_datetime('2030-01-01 16:00'),
resource_id='r4-1')
fake_lease['reservations'].append(
_get_fake_phys_reservation_values(lease_id='lease-4',
resource_id='r1'))
_create_physical_lease(values=fake_lease)
expected = db_api.reservation_get_all_by_values(
**{'resource_id': 'r1'})
self.assertEqual(3, len(expected))
self.check_reservation(expected, 'r1', '2030-01-01 08:00',
'2030-01-01 17:00')