openstack
/
cyborg
Code Issues Proposed changes

Merge "add ut for get arqs list"

This commit is contained in:
Zuul 2020-06-19 08:27:34 +00:00 committed by Gerrit Code Review
parent abe8100dd9 4153c1fea3
commit 9915d7a5f7
2 changed files with 213 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

62
cyborg/tests/unit/api/controllers/v2/test_arqs.py
View File

@ -34,6 +34,9 @@ class TestARQsController(v2_test.APITestV2):
super(TestARQsController, self).setUp()
self.headers = self.gen_headers(self.context)
self.fake_extarqs = fake_extarq.get_fake_extarq_objs()
self.fake_bind_extarqs = fake_extarq.get_fake_extarq_bind_objs()
self.fake_resolved_extarqs = (
fake_extarq.get_fake_extarq_resolved_objs())
self.arqs_controller = arqs.ARQsController()
def _validate_links(self, links, arq_uuid):
@ -79,6 +82,65 @@ class TestARQsController(v2_test.APITestV2):
for in_extarq, out_arq in zip(self.fake_extarqs, out_arqs):
self._validate_arq(in_extarq.arq, out_arq)
@mock.patch('cyborg.objects.ExtARQ.list')
def test_get_all_with_instance(self, mock_extarqs):
# test get_all with instance
mock_extarqs.return_value = self.fake_bind_extarqs
instance_uuid = self.fake_bind_extarqs[0].arq.instance_uuid
url = '%s?instance=%s' % (self.ARQ_URL, instance_uuid)
data = self.get_json(url, headers=self.headers)
out_arqs = data['arqs']
result = isinstance(out_arqs, list)
self.assertTrue(result)
self.assertTrue(len(out_arqs), len(self.fake_bind_extarqs[:2]))
for in_extarq, out_arq in zip(self.fake_bind_extarqs[:2], out_arqs):
self._validate_arq(in_extarq.arq, out_arq)
@mock.patch('cyborg.objects.ExtARQ.list')
def test_get_all_with_bind_state(self, mock_extarqs):
# test get_all with valid bind_state(resolved)
mock_extarqs.return_value = self.fake_resolved_extarqs
url = '%s?bind_state=resolved' % self.ARQ_URL
data = self.get_json(url, headers=self.headers)
out_arqs = data['arqs']
result = isinstance(out_arqs, list)
self.assertTrue(result)
self.assertTrue(len(out_arqs), len(self.fake_resolved_extarqs[1:]))
for in_extarq, out_arq in zip(self.fake_resolved_extarqs[1:],
out_arqs):
self._validate_arq(in_extarq.arq, out_arq)
@mock.patch('cyborg.objects.ExtARQ.list')
def test_get_all_with_instance_and_bind_state(self, mock_extarqs):
# test get_all with instance and valid bind_state(resolved)
mock_extarqs.return_value = self.fake_bind_extarqs[:3]
instance_uuid = self.fake_bind_extarqs[0].arq.instance_uuid
url = '%s?instance=%s&bind_state=resolved' % (
self.ARQ_URL, instance_uuid)
data = self.get_json(url, headers=self.headers)
out_arqs = data['arqs']
result = isinstance(out_arqs, list)
self.assertTrue(result)
self.assertTrue(len(out_arqs), len(self.fake_bind_extarqs[:2]))
for in_extarq, out_arq in zip(self.fake_bind_extarqs[:2], out_arqs):
self._validate_arq(in_extarq.arq, out_arq)
@mock.patch('cyborg.objects.ExtARQ.list')
def test_get_all_with_http_client_LOCKED(self, mock_extarqs):
# test get_all if not all ARQs are in bound state
mock_extarqs.return_value = self.fake_bind_extarqs
instance_uuid = self.fake_bind_extarqs[0].arq.instance_uuid
url = '%s?instance=%s&bind_state=resolved' % (
self.ARQ_URL, instance_uuid)
try:
self.get_json(url, headers=self.headers)
except Exception as e:
exc = e
self.assertIn('423 Locked', exc.args[0])
@mock.patch('cyborg.objects.ExtARQ.list')
def test_get_all_with_invalid_bind_state(self, mock_extarqs):
# test get_all with bind_state=started

151
cyborg/tests/unit/fake_extarq.py
View File

@ -86,6 +86,145 @@ def _get_arqs_as_dict():
return new_arqs
def _get_arqs_resloved_as_dict():
arqs = [ # Corresponds to 1st device profile in fake_device)profile.py
{"uuid": 'a097fefa-da62-4630-8e8b-424c0e3426dd',
"device_profile_group_id": 0,
"state": "Initial",
"device_profile_name": "afaas_example_1",
"device_rp_uuid": None,
"hostname": None,
"instance_uuid": None,
"attach_handle_type": None,
# attach_handle info should vary across ARQs but ignored for testing
"attach_handle_info": {},
"device_profile_group": {}
},
{"uuid": 'aa140114-4869-45ea-8213-45f530804b0e',
"device_profile_group_id": 1,
"device_rp_uuid": "fbd485e1-40b1-4a7e-84b9-f6b6959114a5",
"state": "Bound",
"device_profile_name": "afaas_example_1",
"hostname": "myhost",
"instance_uuid": "5922a70f-1e06-4cfd-88dd-a332120d7144",
"attach_handle_type": "PCI",
# attach_handle info should vary across ARQs but ignored for testing
"attach_handle_info": {
"bus": "1",
"device": "0",
"domain": "0",
"function": "0"
},
"device_profile_group": {
"trait:CUSTOM_FPGA_INTEL": "required",
"resources:FPGA": "1",
"accel:bitstream_id": "b069d97a-010a-4057-b70d-eca2b337fc9c"}
},
]
new_arqs = []
for idx, new_arq in enumerate(arqs):
new_arq.update(id=idx)
new_arqs.append(new_arq)
return new_arqs
def _get_arqs_bind_as_dict():
common = [
{
"state": "Bound",
"device_profile_name": "afaas_example_1",
"hostname": "myhost1",
"instance_uuid": "5922a70f-1e06-4cfd-88dd-a332120d7144",
"attach_handle_type": "PCI",
# attach_handle info should vary across ARQs but ignored for
# testing
"attach_handle_info": {
"bus": "1",
"device": "0",
"domain": "0",
"function": "0"
},
"device_profile_group": {"resources:GPU": "1"}
}, {
"state": "Deleting",
"device_profile_name": "afaas_example_2",
"hostname": "myhost1",
"instance_uuid": "5922a70f-1e06-4cfd-88dd-a332120d7144",
"attach_handle_type": "PCI",
# attach_handle info should vary across ARQs but ignored for
# testing
"attach_handle_info": {
"bus": "2",
"device": "0",
"domain": "0",
"function": "0"
},
"device_profile_group": {
"trait:CUSTOM_FPGA_INTEL": "required",
"resources:FPGA": "1"}
}, {
"state": "Deleting",
"device_profile_name": "afaas_example_3",
"hostname": "myhost3",
"instance_uuid": "5922a70f-1e06-4cfd-88dd-a332120d7146",
"attach_handle_type": "PCI",
# attach_handle info should vary across ARQs but ignored for
# testing
"attach_handle_info": {
"bus": "3",
"device": "0",
"domain": "0",
"function": "0"
},
"device_profile_group": {
"trait:CUSTOM_FPGA_INTEL": "required",
"resources:FPGA": "1",
"accel:bitstream_id": "b069d97a-010a-4057-b70d-eca2b337fc9e"}
}, {
"state": "Unbound",
"device_profile_name": "afaas_example_2",
"hostname": "myhost1",
"instance_uuid": "5922a70f-1e06-4cfd-88dd-a332120d7144",
"attach_handle_type": "PCI",
# attach_handle info should vary across ARQs but ignored for
# testing
"attach_handle_info": {
"bus": "2",
"device": "0",
"domain": "0",
"function": "0"
},
"device_profile_group": {
"trait:CUSTOM_FPGA_INTEL": "required",
"resources:FPGA": "1"}
},
]
arqs = [ # Corresponds to 1st device profile in fake_device)profile.py
{"uuid": "a097fefa-da62-4630-8e8b-424c0e3426de",
"device_profile_group_id": 0,
"device_rp_uuid": "8787595e-9954-49f8-b5c1-cdb55b59062e",
},
{"uuid": "aa140114-4869-45ea-8213-45f530804b0d",
"device_profile_group_id": 0,
"device_rp_uuid": "a1ec17f2-0051-4737-bac4-f074d8a01a9d",
},
{"uuid": "292b2fa2-0831-484c-aeac-09c794428a5e",
"device_profile_group_id": 0,
"device_rp_uuid": "57455a49-bde4-490e-9179-9aa84a3870bb",
},
{"uuid": "292b2fa2-0831-484c-aeac-09c794428a5d",
"device_profile_group_id": 0,
"device_rp_uuid": "57455a49-bde4-490e-9179-9aa84a3870bc",
}
]
new_arqs = []
for idx, new_arq in enumerate(arqs):
new_arq.update(common[idx])
new_arq.update(id=idx)
new_arqs.append(new_arq)
return new_arqs
def _convert_from_dict_to_obj(arq_dict):
obj_arq = arq.ARQ()
for field in arq_dict.keys():
@ -102,6 +241,18 @@ def get_fake_extarq_objs():
return obj_extarqs
def get_fake_extarq_resolved_objs():
arq_list = _get_arqs_resloved_as_dict()
obj_extarqs = list(map(_convert_from_dict_to_obj, arq_list))
return obj_extarqs
def get_fake_extarq_bind_objs():
arq_list = _get_arqs_bind_as_dict()
obj_extarqs = list(map(_convert_from_dict_to_obj, arq_list))
return obj_extarqs
def _convert_from_dict_to_fpga_obj(arq_dict):
obj_arq = arq.ARQ()
for field in arq_dict.keys():