86 lines
2.1 KiB
Python
86 lines
2.1 KiB
Python
from __future__ import print_function
|
|
|
|
import logging
|
|
|
|
import config as config
|
|
from utils.api import Api
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class TestRoleE2E:
|
|
def setup_class(self):
|
|
self.api = Api(config)
|
|
self.api.feat_oas_schema()
|
|
|
|
@classmethod
|
|
def teardown_class(self):
|
|
pass
|
|
|
|
def test_1_create_role(self):
|
|
headers = {
|
|
"x-api-key": config.api_key,
|
|
"username": config.username,
|
|
}
|
|
|
|
data = {
|
|
"role_name": config.role_name,
|
|
"creator": "api",
|
|
"permissions": ['train:all', 'checkpoint:all'],
|
|
}
|
|
|
|
self.api.create_role(headers=headers, data=data)
|
|
|
|
def test_2_create_role_byoc(self):
|
|
headers = {
|
|
"x-api-key": config.api_key,
|
|
"username": config.username,
|
|
}
|
|
|
|
data = {
|
|
"role_name": "byoc",
|
|
"creator": "api",
|
|
"permissions": ['sagemaker_endpoint:all'],
|
|
}
|
|
|
|
resp = self.api.create_role(headers=headers, data=data)
|
|
|
|
def test_3_list_roles_exists(self):
|
|
headers = {
|
|
"x-api-key": config.api_key,
|
|
"username": config.username,
|
|
}
|
|
|
|
resp = self.api.list_roles(headers=headers)
|
|
assert resp.status_code == 200, resp.dumps()
|
|
|
|
assert resp.json()["statusCode"] == 200
|
|
roles = resp.json()['data']["roles"]
|
|
assert config.role_name in [user["role_name"] for user in roles]
|
|
|
|
def test_4_delete_roles_default(self):
|
|
headers = {
|
|
"x-api-key": config.api_key,
|
|
"username": config.username,
|
|
}
|
|
|
|
data = {
|
|
"role_name_list": ['IT Operator'],
|
|
}
|
|
|
|
resp = self.api.delete_roles(headers=headers, data=data)
|
|
assert 'cannot delete default role' in resp.json()["message"]
|
|
|
|
def test_5_delete_roles_succeed(self):
|
|
headers = {
|
|
"x-api-key": config.api_key,
|
|
"username": config.username,
|
|
}
|
|
|
|
data = {
|
|
"role_name_list": [config.role_name],
|
|
}
|
|
|
|
resp = self.api.delete_roles(headers=headers, data=data)
|
|
assert resp.status_code == 204, resp.dumps()
|