def test_host_reservation_get_all(self): """Check that we return 2 hosts.""" db_api.host_reservation_create(_get_fake_host_reservation_values(id=1)) db_api.host_reservation_create(_get_fake_host_reservation_values(id=2)) hosts_reservations = db_api.host_reservation_get_all() self.assertEqual(['1', '2'], [x['id'] for x in hosts_reservations])
def test_host_reservation_get_by_reservation_id(self): """Check that we return 2 hosts.""" db_api.host_reservation_create( _get_fake_host_reservation_values(id=1, reservation_id=1)) db_api.host_reservation_create( _get_fake_host_reservation_values(id=2, reservation_id=2)) res = db_api.host_reservation_get_by_reservation_id(2) self.assertEqual('2', res['id'])
def test_create_duplicate_host_reservation(self): """Create duplicated host reservation Create a duplicated host reservation and verify that an exception is raised. """ db_api.host_reservation_create( _get_fake_host_reservation_values(id='1')) self.assertRaises(db_exceptions.BlazarDBDuplicateEntry, db_api.host_reservation_create, _get_fake_host_reservation_values(id='1'))
def test_delete_host_reservation(self): """Check deletion for host reservation Check all deletion cases for host reservation, including cascade deletion from reservations table. """ self.assertRaises(db_exceptions.BlazarDBNotFound, db_api.host_reservation_destroy, 'fake_id') result = db_api.host_reservation_create( _get_fake_host_reservation_values()) db_api.host_reservation_destroy(result['id']) self.assertIsNone(db_api.host_reservation_get(result['id'])) reserv = db_api.reservation_create(_get_fake_phys_reservation_values()) result = db_api.host_reservation_create( _get_fake_host_reservation_values(reservation_id=reserv['id'])) db_api.reservation_destroy(reserv['id']) self.assertIsNone(db_api.host_reservation_get(result['id']))
def test_create_host_reservation(self): """Create host reservation Create a host reservation and verify that all tables have been populated. """ result = db_api.host_reservation_create( _get_fake_host_reservation_values(id='1')) self.assertEqual(result['id'], _get_fake_host_reservation_values(id='1')['id'])
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 test_update_host_reservation(self): db_api.host_reservation_create(_get_fake_host_reservation_values(id=1)) db_api.host_reservation_update(1, {'resource_properties': 'updated'}) res = db_api.host_reservation_get(1) self.assertEqual('updated', res['resource_properties'])