def test_update_lease_reservations(self, lease_update, lease_get): lease = self.leases.get(name='lease-1') form_data = { 'lease_id': lease['id'], 'reservations': '[{"id": "087bc740-6d2d-410b-9d47-c7b2b55a9d36",' ' "max": 3}]' } lease_get.return_value = lease res = self.client.post(reverse(UPDATE_URL_BASE, args=[lease['id']]), form_data) lease_get.assert_called_once_with(test.IsHttpRequest(), lease['id']) lease_update.assert_called_once_with( test.IsHttpRequest(), lease_id=lease['id'], reservations=[{ "id": "087bc740-6d2d-410b-9d47-c7b2b55a9d36", "max": 3 }]) self.assertNoFormErrors(res) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(res, INDEX_URL)
def test_create_hosts_with_extra_caps(self, hypervisor_list, host_create, host_list): hv_hostnames = [ hv.hypervisor_hostname for hv in self.hypervisors.list() ] calls = [] for host_name in hv_hostnames: calls.append( mock.call(test.IsHttpRequest(), name=host_name, extracap="strong")) form_data = { 'select_hosts_role_member': hv_hostnames, 'extra_caps': '{"extracap": "strong"}' } host_list.return_value = [] host_create.return_value = [] hypervisor_list.return_value = self.hypervisors.list() res = self.client.post(CREATE_URL, form_data) host_list.assert_called_once_with(test.IsHttpRequest()) host_create.assert_has_calls(calls) self.assertEqual(len(hv_hostnames), host_create.call_count) hypervisor_list.assert_called_once_with(test.IsHttpRequest()) self.assertNoFormErrors(res) self.assertMessageCount(success=(len(hv_hostnames) + 1)) self.assertRedirectsNoFollow(res, INDEX_URL)
def test_delete_lease(self, lease_delete, lease_list): leases = self.leases.list() lease = self.leases.get(name='lease-1') action = 'leases__delete__%s' % lease['id'] form_data = {'action': action} lease_list.return_value = leases res = self.client.post(INDEX_URL, form_data) lease_list.assert_called_once_with(test.IsHttpRequest()) lease_delete.assert_called_once_with(test.IsHttpRequest(), lease['id']) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(res, INDEX_URL)
def test_delete_host(self, host_delete, host_list): hosts = self.hosts.list() host = self.hosts.get(hypervisor_hostname='compute-1') action = 'hosts__delete__%s' % host['id'] form_data = {'action': action} host_list.return_value = hosts res = self.client.post(INDEX_URL, form_data) host_list.assert_called_once_with(test.IsHttpRequest()) host_delete.assert_called_once_with(test.IsHttpRequest(), host['id']) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(res, INDEX_URL)
def test_update_host_error(self, host_update, host_get): host = self.hosts.get(hypervisor_hostname='compute-1') form_data = {'host_id': host['id'], 'values': '{"key": "updated"}'} host_get.return_value = host host_update.side_effect = self.exceptions.blazar res = self.client.post(reverse(UPDATE_URL_BASE, args=[host['id']]), form_data) host_get.assert_called_once_with(test.IsHttpRequest(), host['id']) host_update.assert_called_once_with(test.IsHttpRequest(), host_id=host['id'], values={"key": "updated"}) self.assertNoFormErrors(res) self.assertContains(res, 'An error occurred while updating')
def test_update_host(self, host_update, host_get): host = self.hosts.get(hypervisor_hostname='compute-1') form_data = {'host_id': host['id'], 'values': '{"key": "updated"}'} host_get.return_value = host host_update.return_value = [] res = self.client.post(reverse(UPDATE_URL_BASE, args=[host['id']]), form_data) host_get.assert_called_once_with(test.IsHttpRequest(), host['id']) host_update.assert_called_once_with(test.IsHttpRequest(), host_id=host['id'], values={"key": "updated"}) self.assertNoFormErrors(res) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(res, INDEX_URL)
def test_create_lease_client_error(self, lease_create): start_date = datetime(2030, 6, 27, 18, 0, tzinfo=pytz.utc) end_date = datetime(2030, 6, 30, 18, 0, tzinfo=pytz.utc) form_data = { 'name': 'lease-1', 'start_date': start_date.strftime('%Y-%m-%d %H:%M'), 'end_date': end_date.strftime('%Y-%m-%d %H:%M'), 'resource_type': 'host', 'min_hosts': 1, 'max_hosts': 1, } lease_create.side_effect = self.exceptions.blazar res = self.client.post(CREATE_URL, form_data) lease_create.assert_called_once_with( test.IsHttpRequest(), 'lease-1', start_date.strftime('%Y-%m-%d %H:%M'), end_date.strftime('%Y-%m-%d %H:%M'), [{ 'min': 1, 'max': 1, 'hypervisor_properties': '', 'resource_properties': '', 'resource_type': 'physical:host', }], []) self.assertTemplateUsed(res, CREATE_TEMPLATE) self.assertNoFormErrors(res) self.assertContains(res, 'An error occurred while creating')
def test_create_lease_host_reservation(self, lease_create): start_date = datetime(2030, 6, 27, 18, 0, tzinfo=pytz.utc) end_date = datetime(2030, 6, 30, 18, 0, tzinfo=pytz.utc) new_lease = self.leases.get(name='lease-1') form_data = { 'name': 'lease-1', 'start_date': start_date.strftime('%Y-%m-%d %H:%M'), 'end_date': end_date.strftime('%Y-%m-%d %H:%M'), 'resource_type': 'host', 'min_hosts': 1, 'max_hosts': 1, 'hypervisor_properties': '[">=", "$vcpus", "2"]' } lease_create.return_value = new_lease res = self.client.post(CREATE_URL, form_data) lease_create.assert_called_once_with( test.IsHttpRequest(), 'lease-1', start_date.strftime('%Y-%m-%d %H:%M'), end_date.strftime('%Y-%m-%d %H:%M'), [{ 'min': 1, 'max': 1, 'hypervisor_properties': '[">=", "$vcpus", "2"]', 'resource_properties': '', 'resource_type': 'physical:host', }], []) self.assertNoFormErrors(res) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(res, INDEX_URL)
def test_index_error(self, lease_list): lease_list.side_effect = self.exceptions.blazar res = self.client.get(INDEX_URL) lease_list.assert_called_once_with(test.IsHttpRequest()) self.assertTemplateUsed(res, INDEX_TEMPLATE) self.assertMessageCount(res, error=1)
def test_lease_detail_error(self, lease_get): lease_get.side_effect = self.exceptions.blazar res = self.client.get(reverse(DETAIL_URL_BASE, args=['invalid'])) lease_get.assert_called_once_with(test.IsHttpRequest(), 'invalid') self.assertTemplateNotUsed(res, DETAIL_TEMPLATE) self.assertMessageCount(error=1) self.assertRedirectsNoFollow(res, INDEX_URL)
def test_lease_detail(self, lease_get): lease = self.leases.get(name='lease-1') lease_get.return_value = lease res = self.client.get(reverse(DETAIL_URL_BASE, args=[lease['id']])) lease_get.assert_called_once_with(test.IsHttpRequest(), lease['id']) self.assertTemplateUsed(res, DETAIL_TEMPLATE) self.assertContains(res, 'lease-1')
def test_index_no_leases(self, lease_list): leases = [] lease_list.return_value = leases res = self.client.get(INDEX_URL) lease_list.assert_called_once_with(test.IsHttpRequest()) self.assertTemplateUsed(res, INDEX_TEMPLATE) self.assertNoMessages(res) self.assertContains(res, 'No items to display')
def test_host_detail(self, host_get): host = self.hosts.get(hypervisor_hostname='compute-1') host_get.return_value = host res = self.client.get(reverse(DETAIL_URL_BASE, args=[host['id']])) host_get.assert_called_once_with(test.IsHttpRequest(), host['id']) self.assertTemplateUsed(res, DETAIL_TEMPLATE) self.assertContains(res, 'compute-1') self.assertContains(res, 'ex1')
def test_index(self, lease_list): leases = self.leases.list() lease_list.return_value = leases res = self.client.get(INDEX_URL) lease_list.assert_called_once_with(test.IsHttpRequest()) self.assertTemplateUsed(res, INDEX_TEMPLATE) self.assertNoMessages(res) self.assertContains(res, 'lease-2') self.assertContains(res, 'lease-1')
def test_index(self, host_list): hosts = self.hosts.list() host_list.return_value = hosts res = self.client.get(INDEX_URL) host_list.assert_called_once_with(test.IsHttpRequest()) self.assertTemplateUsed(res, INDEX_TEMPLATE) self.assertNoMessages(res) self.assertContains(res, 'compute-1') self.assertContains(res, 'compute-2')
def test_update_lease_name_and_date(self, lease_update, lease_get): lease = self.leases.get(name='lease-1') form_data = { 'lease_id': lease['id'], 'lease_name': 'newname', 'end_time': '+1h' } lease_get.return_value = lease res = self.client.post(reverse(UPDATE_URL_BASE, args=[lease['id']]), form_data) lease_get.assert_called_once_with(test.IsHttpRequest(), lease['id']) lease_update.assert_called_once_with(test.IsHttpRequest(), lease_id=lease['id'], name='newname', prolong_for='1h') self.assertNoFormErrors(res) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(res, INDEX_URL)
def test_update_lease_error(self, lease_update, lease_get): lease = self.leases.get(name='lease-1') form_data = { 'lease_id': lease['id'], 'lease_name': 'newname', 'end_time': '+1h' } lease_get.return_value = lease lease_update.side_effect = self.exceptions.blazar res = self.client.post(reverse(UPDATE_URL_BASE, args=[lease['id']]), form_data) lease_get.assert_called_once_with(test.IsHttpRequest(), lease['id']) lease_update.assert_called_once_with(test.IsHttpRequest(), lease_id=lease['id'], name='newname', prolong_for='1h') self.assertTemplateUsed(UPDATE_TEMPLATE) self.assertNoFormErrors(res) self.assertContains(res, 'An error occurred while updating')
def test_create_lease_instance_reservation(self, lease_create): start_date = datetime(2030, 6, 27, 18, 0, tzinfo=pytz.utc) end_date = datetime(2030, 6, 30, 18, 0, tzinfo=pytz.utc) dummy_lease = {} form_data = { 'name': 'lease-1', 'start_date': start_date.strftime('%Y-%m-%d %H:%M'), 'end_date': end_date.strftime('%Y-%m-%d %H:%M'), 'resource_type': 'instance', 'amount': 3, 'vcpus': 2, 'memory_mb': 4096, 'disk_gb': 128, 'affinity': False, 'resource_properties': '["==", "$energy", "clean"]' } lease_create.return_value = dummy_lease res = self.client.post(CREATE_URL, form_data) lease_create.assert_called_once_with( test.IsHttpRequest(), 'lease-1', start_date.strftime('%Y-%m-%d %H:%M'), end_date.strftime('%Y-%m-%d %H:%M'), [ { 'resource_type': 'virtual:instance', 'amount': 3, 'vcpus': 2, 'memory_mb': 4096, 'disk_gb': 128, 'affinity': 'False', 'resource_properties': '["==", "$energy", "clean"]' } ], []) self.assertNoFormErrors(res) self.assertMessageCount(success=1) self.assertRedirectsNoFollow(res, INDEX_URL)