mirror of
https://github.com/goharbor/harbor.git
synced 2024-07-03 10:15:00 +02:00
78 lines
3.6 KiB
Python
78 lines
3.6 KiB
Python
from __future__ import absolute_import
|
|
|
|
import unittest
|
|
import time
|
|
|
|
from testutils import ADMIN_CLIENT, TEARDOWN, suppress_urllib3_warning
|
|
from library.user import User
|
|
from library.system import System
|
|
from library.system_cve_allowlist import SystemCVEAllowlist
|
|
|
|
import v2_swagger_client
|
|
|
|
class TestSysCVEAllowlist(unittest.TestCase):
|
|
"""
|
|
Test case:
|
|
System Level CVE Allowlist
|
|
Setup:
|
|
Create user(RA)
|
|
Test Steps:
|
|
1. User(RA) reads the system level CVE allowlist and it's empty.
|
|
2. User(RA) updates the system level CVE allowlist, verify it's failed.
|
|
3. Update user(RA) to system admin
|
|
4. User(RA) updates the system level CVE allowlist, verify it's successful.
|
|
5. User(RA) reads the system level CVE allowlist, verify the CVE list is updated.
|
|
6. User(RA) updates the expiration date of system level CVE allowlist.
|
|
7. User(RA) reads the system level CVE allowlist, verify the expiration date is updated.
|
|
Tear Down:
|
|
1. Clear the system level CVE allowlist.
|
|
2. Delete User(RA)
|
|
"""
|
|
@suppress_urllib3_warning
|
|
def setUp(self):
|
|
self.user = User()
|
|
self.system = System()
|
|
self.system_cve_allowlist = SystemCVEAllowlist()
|
|
|
|
user_ra_password = "Aa123456"
|
|
print("Setup: Creating user for test")
|
|
user_ra_id, user_ra_name = self.user.create_user(user_password=user_ra_password, **ADMIN_CLIENT)
|
|
print("Created user: %s, id: %s" % (user_ra_name, user_ra_id))
|
|
self.USER_RA_CLIENT = dict(endpoint=ADMIN_CLIENT["endpoint"],
|
|
username=user_ra_name,
|
|
password=user_ra_password)
|
|
self.user_ra_id = int(user_ra_id)
|
|
|
|
@unittest.skipIf(TEARDOWN == False, "Test data won't be erased.")
|
|
def tearDown(self):
|
|
print("TearDown: Clearing the Allowlist")
|
|
self.system_cve_allowlist.set_cve_allowlist(**ADMIN_CLIENT)
|
|
print("TearDown: Deleting user: %d" % self.user_ra_id)
|
|
self.user.delete_user(self.user_ra_id, **ADMIN_CLIENT)
|
|
|
|
def testSysCVEAllowlist(self):
|
|
# 1. User(RA) reads the system level CVE allowlist and it's empty.
|
|
wl = self.system_cve_allowlist.get_cve_allowlist(**self.USER_RA_CLIENT)
|
|
self.assertEqual(0, len(wl.items), "The initial system level CVE allowlist is not empty: %s" % wl.items)
|
|
# 2. User(RA) updates the system level CVE allowlist, verify it's failed.
|
|
cves = ['CVE-2019-12310']
|
|
self.system_cve_allowlist.set_cve_allowlist(None, 403, *cves, **self.USER_RA_CLIENT)
|
|
# 3. Update user(RA) to system admin
|
|
self.user.update_user_role_as_sysadmin(self.user_ra_id, True, **ADMIN_CLIENT)
|
|
# 4. User(RA) updates the system level CVE allowlist, verify it's successful.
|
|
self.system_cve_allowlist.set_cve_allowlist(None, 200, *cves, **self.USER_RA_CLIENT)
|
|
# 5. User(RA) reads the system level CVE allowlist, verify the CVE list is updated.
|
|
expect_wl = [v2_swagger_client.CVEAllowlistItem(cve_id='CVE-2019-12310')]
|
|
wl = self.system_cve_allowlist.get_cve_allowlist(**self.USER_RA_CLIENT)
|
|
self.assertIsNone(wl.expires_at)
|
|
self.assertEqual(expect_wl, wl.items)
|
|
# 6. User(RA) updates the expiration date of system level CVE allowlist.
|
|
exp = int(time.time()) + 3600
|
|
self.system_cve_allowlist.set_cve_allowlist(exp, 200, *cves, **self.USER_RA_CLIENT)
|
|
# 7. User(RA) reads the system level CVE allowlist, verify the expiration date is updated.
|
|
wl = self.system_cve_allowlist.get_cve_allowlist(**self.USER_RA_CLIENT)
|
|
self.assertEqual(exp, wl.expires_at)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|