Skip to content

Commit c1819e2

Browse files
committed
Add CPE VO EQ and hash method.
Close #30 Close #31
1 parent fcf7410 commit c1819e2

File tree

2 files changed

+50
-0
lines changed

2 files changed

+50
-0
lines changed

tests/security/cve_test.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,31 @@
55

66

77
class TestCveValueObject(unittest.TestCase):
8+
9+
def __init__(self, *args):
10+
super().__init__(*args)
11+
self._cls = Cve
12+
13+
def test_vo_equal_hash(self):
14+
original_vo_hash = hash(self._cls('CVE-2023-33000'))
15+
equal_vo_hash = hash(self._cls('CVE-2023-33000'))
16+
self.assertEqual(original_vo_hash, equal_vo_hash)
17+
18+
def test_vo_different_hash(self):
19+
original_vo_hash = hash(self._cls('CVE-2023-33000'))
20+
not_equal_vo_hash = hash(self._cls('CVE-2021-34527'))
21+
self.assertNotEqual(original_vo_hash, not_equal_vo_hash)
22+
23+
def test_vo_equality(self):
24+
original_vo = self._cls('CVE-2023-33000')
25+
equal_vo = self._cls('CVE-2023-33000')
26+
self.assertEqual(original_vo, equal_vo)
27+
28+
def test_vo_different_equality(self):
29+
original_vo = self._cls('CVE-2023-33000')
30+
different_vo = self._cls('CVE-2021-34527')
31+
self.assertNotEqual(original_vo, different_vo)
32+
833
def test_value_return_input_value(self):
934
vo = Cve('CVE-2023-33000')
1035
self.assertEqual('CVE-2023-33000', vo.value())

tests/security/nullable_cve_test.py

Lines changed: 25 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,31 @@
55

66

77
class TestNullableCveValueObject(unittest.TestCase):
8+
9+
def __init__(self, *args):
10+
super().__init__(*args)
11+
self._cls = NullableCve
12+
13+
def test_vo_equal_hash(self):
14+
original_vo_hash = hash(self._cls('CVE-2023-33000'))
15+
equal_vo_hash = hash(self._cls('CVE-2023-33000'))
16+
self.assertEqual(original_vo_hash, equal_vo_hash)
17+
18+
def test_vo_different_hash(self):
19+
original_vo_hash = hash(self._cls('CVE-2023-33000'))
20+
not_equal_vo_hash = hash(self._cls('CVE-2021-34527'))
21+
self.assertNotEqual(original_vo_hash, not_equal_vo_hash)
22+
23+
def test_vo_equality(self):
24+
original_vo = self._cls('CVE-2023-33000')
25+
equal_vo = self._cls('CVE-2023-33000')
26+
self.assertEqual(original_vo, equal_vo)
27+
28+
def test_vo_different_equality(self):
29+
original_vo = self._cls('CVE-2023-33000')
30+
different_vo = self._cls('CVE-2021-34527')
31+
self.assertNotEqual(original_vo, different_vo)
32+
833
def test_value_return_input_value(self):
934
self.assertRaises(ValueObjectError, NullableCve, '')
1035

0 commit comments

Comments
 (0)