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))
