Coverage for tests/publisher/cve/test_cve_paginate_data.py: 100%

1import unittest 

2from webapp.publisher.cve.cve_helper import CveHelper 

3 

4 

5class CvePaginateDataTest(unittest.TestCase): 

6 

7 def setUp(self): 

8 self.cves = [ 

9 { 

10 "id": "CVE-2023-1001", 

11 "cvss_severity": "high", 

12 "ubuntu_priority": "medium", 

13 "cvss_score": 7.5, 

14 }, 

15 { 

16 "id": "CVE-2023-1002", 

17 "cvss_severity": "low", 

18 "ubuntu_priority": "high", 

19 "cvss_score": 4.3, 

20 }, 

21 { 

22 "id": "CVE-2023-1003", 

23 "cvss_severity": "critical", 

24 "ubuntu_priority": "low", 

25 "cvss_score": 9.8, 

26 }, 

27 { 

28 "id": "CVE-2023-1004", 

29 "cvss_severity": "medium", 

30 "ubuntu_priority": "critical", 

31 "cvss_score": 6.1, 

32 }, 

33 ] 

34 

35 def test_paginate_cve_list(self): 

36 paginated_cves = CveHelper.paginate_cve_list( 

37 self.cves, page=1, page_size=2 

38 ) 

39 self.assertEqual(paginated_cves["page"], 1) 

40 self.assertEqual(paginated_cves["page_size"], 2) 

41 self.assertEqual(paginated_cves["total_items"], 4) 

42 self.assertEqual(paginated_cves["total_pages"], 2) 

43 self.assertEqual(len(paginated_cves["data"]), 2) 

44 self.assertEqual( 

45 [cve["id"] for cve in paginated_cves["data"]], 

46 ["CVE-2023-1001", "CVE-2023-1002"], 

47 )