本文整理汇总了Python中w3af.core.data.kb.info_set.InfoSet类的典型用法代码示例。如果您正苦于以下问题：Python InfoSet类的具体用法？Python InfoSet怎么用？Python InfoSet使用的例子？那么, 这里精选的类代码示例或许可以为您提供帮助。
在下文中一共展示了InfoSet类的14个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: test_deepcopy
def test_deepcopy(self):
i = MockInfo()
iset1 = InfoSet([i])
iset1_copy = copy.deepcopy(iset1)
self.assertEqual(iset1.get_uniq_id(), iset1_copy.get_uniq_id())示例2: test_pickle
def test_pickle(self):
i = MockInfo()
iset1 = InfoSet([i])
pickled_iset1 = cpickle_dumps(iset1)
iset1_clone = loads(pickled_iset1)
self.assertEqual(iset1.get_uniq_id(), iset1_clone.get_uniq_id())示例3: test_add
def test_add(self):
i1 = MockInfo(ids=1)
i2 = MockInfo(ids=2)
iset = InfoSet([i1])
added = iset.add(i2)
self.assertEqual(iset.get_id(), [1, 2])
self.assertTrue(added)示例4: test_get_desc_template_info_attr_access
def test_get_desc_template_info_attr_access(self):
value = 'Yuuup!'
i = MockInfo()
i['tag'] = value
iset = InfoSet([i])
iset.TEMPLATE = '{{ tag }}'
self.assertEqual(iset.get_desc(), value)示例5: test_add_more_than_max
def test_add_more_than_max(self):
i1 = MockInfo(ids=1)
i2 = MockInfo(ids=2)
iset = InfoSet([i1])
iset.MAX_INFO_INSTANCES = 2
added = iset.add(i1)
self.assertTrue(added)
added = iset.add(i2)
self.assertFalse(added)示例6: test_match_same_itag
def test_match_same_itag(self):
"""
https://github.com/andresriancho/w3af/issues/10286
"""
itag_1 = 'hello'
i1 = MockInfo(ids=1)
i1[itag_1] = 1
iset_1 = InfoSet([i1])
iset_1.ITAG = itag_1
i2 = MockInfo(ids=2)
i2[itag_1] = 1
self.assertTrue(iset_1.match(i2))示例7: test_match_different_itag
def test_match_different_itag(self):
"""
https://github.com/andresriancho/w3af/issues/10286
"""
itag_1 = 'hello'
i1 = MockInfo(ids=1)
i1[itag_1] = 1
iset_1 = InfoSet([i1])
iset_1.ITAG = itag_1
itag_2 = 'world'
i2 = MockInfo(ids=2)
i2[itag_2] = 2
self.assertFalse(iset_1.match(i2))示例8: test_all_of_info_exclude_ids
def test_all_of_info_exclude_ids(self):
i1 = MockInfo()
i2 = MockInfo()
v1 = MockVuln()
v2 = MockVuln()
iset = InfoSet([i2])
vset = InfoSet([v2])
kb.append('a', 'b', i1)
kb.append('w', 'z', iset)
kb.append('x', 'y', v1)
kb.append('4', '2', vset)
all_findings = kb.get_all_findings()
all_findings_except_v1 = kb.get_all_findings(exclude_ids=(v1.get_uniq_id(),))
all_findings_except_v1_v2 = kb.get_all_findings(exclude_ids=(v1.get_uniq_id(), vset.get_uniq_id()))
self.assertEqual(all_findings, [i1, iset, v1, vset])
self.assertEqual(all_findings_except_v1, [i1, iset, vset])
self.assertEqual(all_findings_except_v1_v2, [i1, iset])示例9: test_get_uniq_id
def test_get_uniq_id(self):
i = MockInfo()
iset = InfoSet([i])
self.assertIsNotNone(iset.get_uniq_id())示例10: test_get_plugin_name
def test_get_plugin_name(self):
i = MockInfo()
iset = InfoSet([i])
self.assertEqual(iset.get_plugin_name(), 'plugin_name')示例11: test_get_id
def test_get_id(self):
i1 = MockInfo(ids=1)
i2 = MockInfo(ids=2)
iset = InfoSet([i2, i1])
self.assertEqual(iset.get_id(), [1, 2])示例12: test_get_desc_no_template
def test_get_desc_no_template(self):
i = MockInfo()
iset = InfoSet([i])
self.assertEqual(iset.get_desc(), MockInfo.LONG_DESC)示例13: test_get_name
def test_get_name(self):
i = MockInfo()
iset = InfoSet([i])
self.assertEqual(iset.get_name(), 'TestCase')示例14: test_to_json
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')
iset = InfoSet([i])
jd = iset.to_json()
json_string = json.dumps(jd)
jd = json.loads(json_string)
self.assertEqual(jd['name'], iset.get_name())
self.assertEqual(jd['url'], str(iset.get_url()))
self.assertEqual(jd['var'], iset.get_token_name())
self.assertEqual(jd['response_ids'], iset.get_id())
self.assertEqual(jd['vulndb_id'], iset.get_vulndb_id())
self.assertEqual(jd['desc'], iset.get_desc(with_id=False))
self.assertEqual(jd['long_description'], iset.get_long_description())
self.assertEqual(jd['fix_guidance'], iset.get_fix_guidance())
self.assertEqual(jd['fix_effort'], iset.get_fix_effort())
self.assertEqual(jd['tags'], iset.get_tags())
self.assertEqual(jd['wasc_ids'], iset.get_wasc_ids())
self.assertEqual(jd['wasc_urls'], list(iset.get_wasc_urls()))
self.assertEqual(jd['cwe_urls'], list(iset.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'], iset.get_plugin_name())
self.assertEqual(jd['severity'], iset.get_severity())
self.assertEqual(jd['attributes'], iset.first_info.copy())
self.assertEqual(jd['highlight'], list(iset.get_to_highlight()))