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)