mirror of
https://github.com/goharbor/harbor.git
synced 2024-09-05 18:48:04 +02:00
* Refactor scan all api This commit is to let scan all api using admin job to handle schedule management. After the PR, GC and scan all share unified code path. Signed-off-by: wang yan <wangyan@vmware.com> * update admin job api code according to review comments Signed-off-by: wang yan <wangyan@vmware.com> * Update test code and comments per review Signed-off-by: wang yan <wangyan@vmware.com>
100 lines
4.4 KiB
Python
100 lines
4.4 KiB
Python
from __future__ import absolute_import
|
|
import unittest
|
|
|
|
from testutils import harbor_server
|
|
from testutils import TEARDOWN
|
|
from testutils import ADMIN_CLIENT
|
|
from library.system import System
|
|
from library.project import Project
|
|
from library.user import User
|
|
from library.repository import Repository
|
|
from library.repository import push_image_to_project
|
|
|
|
class TestProjects(unittest.TestCase):
|
|
@classmethod
|
|
def setUp(self):
|
|
system = System()
|
|
self.system= system
|
|
|
|
project = Project()
|
|
self.project= project
|
|
|
|
user = User()
|
|
self.user= user
|
|
|
|
repo = Repository()
|
|
self.repo= repo
|
|
|
|
@classmethod
|
|
def tearDown(self):
|
|
print "Case completed"
|
|
|
|
@unittest.skipIf(TEARDOWN == False, "Test data won't be erased.")
|
|
def test_ClearData(self):
|
|
#1. Delete Alice's repository and Luca's repository;
|
|
self.repo.delete_repoitory(TestProjects.repo_Alice_name, **ADMIN_CLIENT)
|
|
self.repo.delete_repoitory(TestProjects.repo_Luca_name, **ADMIN_CLIENT)
|
|
|
|
#2. Delete Alice's project and Luca's project;
|
|
self.project.delete_project(TestProjects.project_Alice_id, **ADMIN_CLIENT)
|
|
self.project.delete_project(TestProjects.project_Luca_id, **ADMIN_CLIENT)
|
|
|
|
#3. Delete user Alice and Luca.
|
|
self.user.delete_user(TestProjects.user_Alice_id, **ADMIN_CLIENT)
|
|
self.user.delete_user(TestProjects.user_Luca_id, **ADMIN_CLIENT)
|
|
|
|
def testScanImage(self):
|
|
"""
|
|
Test case:
|
|
Scan All Image
|
|
Test step and expected result:
|
|
1. Create user Alice and Luca;
|
|
2. Create 2 new private projects project_Alice and project_Luca;
|
|
3. Push a image to project_Alice and push another image to project_Luca;
|
|
4. Trigger scan all event;
|
|
5. Check if image in project_Alice and another image in project_Luca were both scanned.
|
|
Tear down:
|
|
1. Delete Alice's repository and Luca's repository;
|
|
2. Delete Alice's project and Luca's project;
|
|
3. Delete user Alice and Luca.
|
|
"""
|
|
url = ADMIN_CLIENT["endpoint"]
|
|
user_common_password = "Aa123456"
|
|
|
|
#1. Create user Alice and Luca;
|
|
TestProjects.user_Alice_id, user_Alice_name = self.user.create_user(user_password = user_common_password, **ADMIN_CLIENT)
|
|
TestProjects.user_Luca_id, user_Luca_name = self.user.create_user(user_password = user_common_password, **ADMIN_CLIENT)
|
|
|
|
USER_ALICE_CLIENT=dict(endpoint = url, username = user_Alice_name, password = user_common_password)
|
|
USER_LUCA_CLIENT=dict(endpoint = url, username = user_Luca_name, password = user_common_password)
|
|
|
|
#2. Create 2 new private projects project_Alice and project_Luca;
|
|
TestProjects.project_Alice_id, project_Alice_name = self.project.create_project(metadata = {"public": "false"}, **USER_ALICE_CLIENT)
|
|
TestProjects.project_Luca_id, project_Luca_name = self.project.create_project(metadata = {"public": "false"}, **USER_LUCA_CLIENT)
|
|
|
|
#3. Push a image to project_Alice and push another image to project_Luca;
|
|
|
|
#Note: Please make sure that this Image has never been pulled before by any other cases,
|
|
# so it is a not-scanned image rigth after repository creation.
|
|
#image = "tomcat"
|
|
image = "mariadb"
|
|
src_tag = "latest"
|
|
#3.1 Push a image to project_Alice;
|
|
TestProjects.repo_Alice_name, tag_Alice = push_image_to_project(project_Alice_name, harbor_server, user_Alice_name, user_common_password, image, src_tag)
|
|
|
|
#Note: Please make sure that this Image has never been pulled before by any other cases,
|
|
# so it is a not-scanned image rigth after repository creation.
|
|
image = "mysql"
|
|
src_tag = "latest"
|
|
#3.2 push another image to project_Luca;
|
|
TestProjects.repo_Luca_name, tag_Luca = push_image_to_project(project_Luca_name, harbor_server, user_Luca_name, user_common_password, image, src_tag)
|
|
|
|
#4. Trigger scan all event;
|
|
self.system.scan_now(**ADMIN_CLIENT)
|
|
|
|
#5. Check if image in project_Alice and another image in project_Luca were both scanned.
|
|
self.repo.check_image_scan_result(TestProjects.repo_Alice_name, tag_Alice, expected_scan_status = "finished", **USER_ALICE_CLIENT)
|
|
self.repo.check_image_scan_result(TestProjects.repo_Luca_name, tag_Luca, expected_scan_status = "finished", **USER_LUCA_CLIENT)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |