def test_to_json(self): i = Info('Blind SQL injection vulnerability', MockInfo.LONG_DESC, 1, 'plugin_name') i['test'] = 'foo' i.add_to_highlight('abc', 'def') jd = i.to_json() json_string = json.dumps(jd) jd = json.loads(json_string) self.assertEqual(jd['name'], i.get_name()) self.assertEqual(jd['url'], str(i.get_url())) self.assertEqual(jd['var'], i.get_token_name()) self.assertEqual(jd['response_ids'], i.get_id()) self.assertEqual(jd['vulndb_id'], i.get_vulndb_id()) self.assertEqual(jd['desc'], i.get_desc(with_id=False)) self.assertEqual(jd['long_description'], i.get_long_description()) self.assertEqual(jd['fix_guidance'], i.get_fix_guidance()) self.assertEqual(jd['fix_effort'], i.get_fix_effort()) self.assertEqual(jd['tags'], i.get_tags()) self.assertEqual(jd['wasc_ids'], i.get_wasc_ids()) self.assertEqual(jd['wasc_urls'], list(i.get_wasc_urls())) self.assertEqual(jd['cwe_urls'], list(i.get_cwe_urls())) self.assertEqual(jd['references'], BLIND_SQLI_REFS) self.assertEqual(jd['owasp_top_10_references'], BLIND_SQLI_TOP10_REFS) self.assertEqual(jd['plugin_name'], i.get_plugin_name()) self.assertEqual(jd['severity'], i.get_severity()) self.assertEqual(jd['attributes'], i.copy()) self.assertEqual(jd['highlight'], list(i.get_to_highlight()))
def test_to_json(self): i = Info("Blind SQL injection vulnerability", MockInfo.LONG_DESC, 1, "plugin_name") i["test"] = "foo" i.add_to_highlight("abc", "def") jd = i.to_json() json_string = json.dumps(jd) jd = json.loads(json_string) self.assertEqual(jd["name"], i.get_name()) self.assertEqual(jd["url"], str(i.get_url())) self.assertEqual(jd["var"], i.get_token_name()) self.assertEqual(jd["response_ids"], i.get_id()) self.assertEqual(jd["vulndb_id"], i.get_vulndb_id()) self.assertEqual(jd["desc"], i.get_desc(with_id=False)) self.assertEqual(jd["long_description"], i.get_long_description()) self.assertEqual(jd["fix_guidance"], i.get_fix_guidance()) self.assertEqual(jd["fix_effort"], i.get_fix_effort()) self.assertEqual(jd["tags"], i.get_tags()) self.assertEqual(jd["wasc_ids"], i.get_wasc_ids()) self.assertEqual(jd["wasc_urls"], list(i.get_wasc_urls())) self.assertEqual(jd["cwe_urls"], list(i.get_cwe_urls())) self.assertEqual(jd["references"], BLIND_SQLI_REFS) self.assertEqual(jd["owasp_top_10_references"], BLIND_SQLI_TOP10_REFS) self.assertEqual(jd["plugin_name"], i.get_plugin_name()) self.assertEqual(jd["severity"], i.get_severity()) self.assertEqual(jd["attributes"], i.copy()) self.assertEqual(jd["highlight"], list(i.get_to_highlight()))