def test_get_agent_cve(socket_mock, send_mock, params, field_name, expected_items):
"""Check if expected items are returned when using filters on get_agent_cve()
Parameters
----------
params : dict
Parameters to be used in the tested function.
field_name : str
Result's field to be checked.
expected_items : list
Items which should be returned in the response.
"""
result = get_agent_cve(agent_list=['001'], **params).render()
if expected_items:
for idx, expected_value in enumerate(expected_items):
assert result['data']['affected_items'][idx][field_name] == expected_value
else:
assert result['data']['affected_items'] == []
def test_get_agent_cve_select(socket_mock, send_mock, params, expected_fields):
"""Check if expected fields are returned when using select in get_agent_cve()"""
result = get_agent_cve(agent_list=['001'], **params).render()
for item in result['data']['affected_items']:
assert expected_fields.symmetric_difference(item.keys()) == set(
), '"select" param did not return expected fields.'
