mirror of
https://github.com/goharbor/harbor
synced 2024-09-21 15:17:47 +00:00
5f8acc3896
When the registry shifts from token auth to basic auth, we'll use the middleware to check permission. This commit add middlewares for populate the artifact info and check permission based on request to /v2/* api via security context Signed-off-by: Daniel Jiang <jiangd@vmware.com>
88 lines
3.4 KiB
Python
88 lines
3.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.project import Project
|
|
from library.user import User
|
|
from library.repository import Repository
|
|
from library.repository import push_image_to_project
|
|
from library.system import System
|
|
|
|
class TestProjects(unittest.TestCase):
|
|
@classmethod
|
|
def setUp(self):
|
|
project = Project()
|
|
self.project= project
|
|
|
|
user = User()
|
|
self.user= user
|
|
|
|
repo = Repository()
|
|
self.repo= repo
|
|
|
|
self.system = System()
|
|
|
|
@classmethod
|
|
def tearDown(self):
|
|
print "Case completed"
|
|
|
|
@unittest.skipIf(TEARDOWN == False, "Test data won't be erased.")
|
|
def test_ClearData(self):
|
|
#1. Delete project(PA);
|
|
self.project.delete_project(TestProjects.project_test_quota_id, **ADMIN_CLIENT)
|
|
|
|
#2. Delete user(UA);
|
|
self.user.delete_user(TestProjects.user_test_quota_id, **ADMIN_CLIENT)
|
|
|
|
def testProjectQuota(self):
|
|
"""
|
|
Test case:
|
|
Project Quota
|
|
Test step and expected result:
|
|
1. Create a new user(UA);
|
|
2. Create a new private project(PA) by user(UA);
|
|
3. Add user(UA) as a member of project(PA) with project-admin role;
|
|
4. Push an image to project(PA) by user(UA), then check the project quota usage;
|
|
5. Check quota change
|
|
6. Delete image, the quota should be changed to 0.
|
|
Tear down:
|
|
1. Delete repository(RA) by user(UA);
|
|
2. Delete project(PA);
|
|
3. Delete user(UA);
|
|
"""
|
|
url = ADMIN_CLIENT["endpoint"]
|
|
user_001_password = "Aa123456"
|
|
|
|
#1. Create user-001
|
|
TestProjects.user_test_quota_id, user_test_quota_name = self.user.create_user(user_password = user_001_password, **ADMIN_CLIENT)
|
|
TestProjects.USER_TEST_QUOTA_CLIENT=dict(endpoint = url, username = user_test_quota_name, password = user_001_password)
|
|
|
|
#2. Create a new private project(PA) by user(UA);
|
|
TestProjects.project_test_quota_id, project_test_quota_name = self.project.create_project(metadata = {"public": "false"}, **ADMIN_CLIENT)
|
|
|
|
#3. Add user(UA) as a member of project(PA) with project-admin role;
|
|
self.project.add_project_members(TestProjects.project_test_quota_id, TestProjects.user_test_quota_id, **ADMIN_CLIENT)
|
|
|
|
#4.Push an image to project(PA) by user(UA), then check the project quota usage; -- {"count": 1, "storage": 2791709}
|
|
image = "alpine"
|
|
src_tag = "3.10"
|
|
TestProjects.repo_name, _ = push_image_to_project(project_test_quota_name, harbor_server, user_test_quota_name, user_001_password, image, src_tag)
|
|
|
|
#5. Get project quota
|
|
quota = self.system.get_project_quota("project", TestProjects.project_test_quota_id, **ADMIN_CLIENT)
|
|
self.assertEqual(quota[0].used["count"], 1)
|
|
self.assertEqual(quota[0].used["storage"], 2789002)
|
|
|
|
#6. Delete repository(RA) by user(UA);
|
|
self.repo.delete_repoitory(TestProjects.repo_name, **ADMIN_CLIENT)
|
|
|
|
#6. Quota should be 0
|
|
quota = self.system.get_project_quota("project", TestProjects.project_test_quota_id, **ADMIN_CLIENT)
|
|
self.assertEqual(quota[0].used["count"], 0)
|
|
self.assertEqual(quota[0].used["storage"], 0)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main() |