def test_extend_to_empty_same(self): """Extend ImpactFuncSet to empty one.""" imp_fun = ImpactFuncSet() imp_fun_add = ImpactFuncSet() vulner_1 = ImpactFunc() vulner_1.id = 1 vulner_1.haz_type = 'TC' imp_fun_add.append(vulner_1) vulner_2 = ImpactFunc() vulner_2.id = 3 vulner_2.haz_type = 'TC' imp_fun_add.append(vulner_2) vulner_3 = ImpactFunc() vulner_3.id = 3 vulner_3.haz_type = 'FL' imp_fun_add.append(vulner_3) imp_fun_add.tag.file_name = 'file1.txt' imp_fun.extend(imp_fun_add) imp_fun.check() self.assertEqual(imp_fun.size(), 3) self.assertEqual(imp_fun.size('TC'), 2) self.assertEqual(imp_fun.size('FL'), 1) self.assertEqual(imp_fun.tag.file_name, imp_fun_add.tag.file_name) self.assertEqual(imp_fun.tag.description, imp_fun_add.tag.description)
def test_extend_equal_same(self): """Extend the same ImpactFuncSet. The inital ImpactFuncSet is obtained.""" imp_fun = ImpactFuncSet() vulner_1 = ImpactFunc() vulner_1.id = 1 vulner_1.haz_type = 'TC' imp_fun.append(vulner_1) imp_fun_add = ImpactFuncSet() imp_fun_add.append(vulner_1) imp_fun.extend(imp_fun_add) imp_fun.check() self.assertEqual(imp_fun.size(), 1) self.assertEqual(imp_fun.size('TC'), 1)
def test_extend_different_extend(self): """Extend ImpactFuncSet with same and new values. The vulnerabilities with repeated id are overwritten.""" imp_fun = ImpactFuncSet() vulner_1 = ImpactFunc() vulner_1.id = 1 vulner_1.haz_type = 'TC' imp_fun.append(vulner_1) vulner_2 = ImpactFunc() vulner_2.id = 3 vulner_2.haz_type = 'TC' imp_fun.append(vulner_2) vulner_3 = ImpactFunc() vulner_3.id = 3 vulner_3.haz_type = 'FL' imp_fun.append(vulner_3) imp_fun_add = ImpactFuncSet() vulner_1 = ImpactFunc() vulner_1.id = 1 vulner_1.haz_type = 'TC' imp_fun_add.append(vulner_1) vulner_2 = ImpactFunc() vulner_2.id = 1 vulner_2.haz_type = 'WS' imp_fun_add.append(vulner_2) vulner_3 = ImpactFunc() vulner_3.id = 3 vulner_3.haz_type = 'FL' imp_fun_add.append(vulner_3) imp_fun.extend(imp_fun_add) imp_fun.check() self.assertEqual(imp_fun.size(), 4) self.assertEqual(imp_fun.size('TC'), 2) self.assertEqual(imp_fun.size('FL'), 1) self.assertEqual(imp_fun.size('WS'), 1)
def test_size_wrong_zero(self): """Test size method with wrong inputs.""" imp_fun = ImpactFuncSet() self.assertEqual(0, imp_fun.size('TC')) self.assertEqual(0, imp_fun.size('TC', 3)) self.assertEqual(0, imp_fun.size(fun_id=3))
def test_size_pass(self): """Test size function.""" imp_fun = ImpactFuncSet() self.assertEqual(0, imp_fun.size()) vulner_1 = ImpactFunc() vulner_1.haz_type = 'WS' vulner_1.id = 56 imp_fun.append(vulner_1) self.assertEqual(1, imp_fun.size()) self.assertEqual(1, imp_fun.size('WS', 56)) self.assertEqual(1, imp_fun.size('WS')) self.assertEqual(1, imp_fun.size(fun_id=56)) imp_fun.append(vulner_1) self.assertEqual(1, imp_fun.size()) self.assertEqual(1, imp_fun.size('WS', 56)) self.assertEqual(1, imp_fun.size('WS')) self.assertEqual(1, imp_fun.size(fun_id=56)) vulner_2 = ImpactFunc() vulner_2.haz_type = 'WS' vulner_2.id = 5 imp_fun.append(vulner_2) self.assertEqual(2, imp_fun.size()) self.assertEqual(1, imp_fun.size('WS', 56)) self.assertEqual(2, imp_fun.size('WS')) self.assertEqual(1, imp_fun.size(fun_id=56)) self.assertEqual(1, imp_fun.size(fun_id=5)) vulner_3 = ImpactFunc() vulner_3.haz_type = 'TC' vulner_3.id = 5 imp_fun.append(vulner_3) self.assertEqual(3, imp_fun.size()) self.assertEqual(1, imp_fun.size('TC', 5)) self.assertEqual(2, imp_fun.size('WS')) self.assertEqual(1, imp_fun.size('TC')) self.assertEqual(1, imp_fun.size(fun_id=56)) self.assertEqual(2, imp_fun.size(fun_id=5))