from __future__ import absolute_import import unittest import urllib import library.singularity from library.sign import sign_image from testutils import ADMIN_CLIENT from testutils import harbor_server from testutils import TEARDOWN from library.user import User from library.project import Project from library.repository import Repository from library.artifact import Artifact class TestProjects(unittest.TestCase): @classmethod def setUp(self): self.project = Project() self.user = User() self.artifact = Artifact() self.repo = Repository() self.repo_name = "busybox" self.tag = "1.28" @classmethod def tearDown(self): print "Case completed" @unittest.skipIf(TEARDOWN == False, "Test data won't be erased.") def test_ClearData(self): #1. Delete user(UA); self.user.delete_user(TestProjects.user_sign_image_id, **ADMIN_CLIENT) def testPushSingularity(self): """ Test case: Push Singularity file With Singularity CLI Test step and expected result: 1. Create user-001 2. Create a new private project(PA) by user(UA); 3. Push a sif file to harbor by singularity; 4. Get repository from Harbor successfully, and verfiy repository name is repo pushed by singularity CLI; 5. Get and verify artifacts by tag; 6. Pull sif file from harbor by singularity; Tear down: NA """ url = ADMIN_CLIENT["endpoint"] user_001_password = "Aa123456" #1. Create user-001 TestProjects.user_sign_image_id, user_name = self.user.create_user(user_password = user_001_password, **ADMIN_CLIENT) TestProjects.USER_CLIENT=dict(with_signature = True, endpoint = url, username = user_name, password = user_001_password) #2. Create a new private project(PA) by user(UA); TestProjects.project_id, TestProjects.project_name = self.project.create_project(metadata = {"public": "false"}, **TestProjects.USER_CLIENT) #3. Push a sif file to harbor by singularity; library.singularity.push_singularity_to_harbor("library:", "library/default/", harbor_server, user_name, user_001_password, TestProjects.project_name, self.repo_name, self.tag) #4. Get repository from Harbor successfully, and verfiy repository name is repo pushed by singularity CLI; repo_data = self.repo.get_repository(TestProjects.project_name, self.repo_name, **TestProjects.USER_CLIENT) print "repo_data:", repo_data self.assertEqual(repo_data.name, TestProjects.project_name + "/" + self.repo_name) #5. Get and verify artifacts by tag; artifact = self.artifact.get_reference_info(TestProjects.project_name, self.repo_name, self.tag, **TestProjects.USER_CLIENT) print "artifact:", artifact self.assertEqual(artifact[0].tags[0].name, self.tag) #6. Pull sif file from harbor by singularity; library.singularity.singularity_pull(TestProjects.project_name + ".sif", "oras://"+harbor_server + "/" + TestProjects.project_name + "/" + self.repo_name+":"+ self.tag) if __name__ == '__main__': unittest.main()