def test_filter_advisories_ips_sig_fields_nested(self):
fields = ["advisory_title"] + list(constants.IPS_SIGNATURES)
expected_output = [{
'advisory_title': '%s' % mock_advisory_title,
constants.IPS_SIGNATURE_LABEL: [],
}]
output = utils.filter_advisories(mock_advisories, fields)
self.assertIsInstance(output, list)
self.assertDictEqual(output[0], expected_output[0])
def test_filter_advisories_general_input(self):
fields = ["advisory_title", "sir", "bug_ids"]
expected_output = [{
"advisory_title": "Mock Advisory Title",
"sir": "Medium",
"bug_ids": "BUGISidf"
}]
output = utils.filter_advisories(mock_advisories, fields)
self.assertListEqual(output, expected_output)
def test_output_csv_succeeds(self, mock_open):
s_empty = ',\n,'
mock_open.side_effect = [
mock.mock_open(read_data=s_empty).return_value,
]
filtered_advisories = utils.filter_advisories(mock_advisories,
constants.API_LABELS)
self.assertIsNone(
utils.output(filtered_advisories, CSV_OUTPUT_FORMAT_TOKEN,
mock_open))
def test_output_trial(self, mock_open):
filtered_advisories = utils.filter_advisories(mock_advisories,
constants.API_LABELS)
s_rep = json.dumps(filtered_advisories)
mock_open.side_effect = [
mock.mock_open(read_data=s_rep).return_value,
]
self.assertIsNone(
utils.output(filtered_advisories, JSON_OUTPUT_FORMAT_TOKEN,
mock_open))
def test_filter_advisories_invalid_fields(self):
fields = ["advisory_title", "v_score"]
expected_output = [{'advisory_title': '%s' % mock_advisory_title}]
output = utils.filter_advisories(mock_advisories, fields)
self.assertIsInstance(output, list)
self.assertDictEqual(output[0], expected_output[0])
def test_filter_advisories_empty_fields(self):
fields = []
expected_output = [{}]
self.assertListEqual(utils.filter_advisories(mock_advisories, fields),
expected_output)
