def test_update_vuln(self): vuln = MockVuln() kb.append('a', 'b', vuln) update_vuln = copy.deepcopy(vuln) update_vuln.set_name('a') update_uniq_id = update_vuln.get_uniq_id() kb.update(vuln, update_vuln) self.assertNotEqual(update_vuln, vuln) self.assertEqual(update_vuln, kb.get_by_uniq_id(update_uniq_id))
def test_update_info(self): info = MockInfo() kb.append('a', 'b', info) update_info = copy.deepcopy(info) update_info.set_name('a') update_uniq_id = update_info.get_uniq_id() kb.update(info, update_info) self.assertNotEqual(update_info, info) self.assertEqual(update_info, kb.get_by_uniq_id(update_uniq_id))
def test_observer_update(self): observer1 = Mock() info = MockInfo() kb.add_observer(observer1) kb.append('a', 'b', info) old_info = copy.deepcopy(info) info.set_name('new name') kb.update(old_info, info) observer1.update.assert_called_once_with(old_info, info)