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())
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())
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())
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])
def test_get_uniq_id(self):
i = MockInfo()
iset = InfoSet([i])
self.assertIsNotNone(iset.get_uniq_id())
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])
