Skip to content

Commit 33a4502

Browse files
Convert www.security test suite to pytest and remove residuals (apache#18961)
Co-authored-by: Tzu-ping Chung <[email protected]>
1 parent 9a469d8 commit 33a4502

File tree

4 files changed

+691
-567
lines changed

4 files changed

+691
-567
lines changed

Diff for: tests/test_utils/api_connexion_utils.py

+12-4
Original file line numberDiff line numberDiff line change
@@ -80,15 +80,23 @@ def create_role(app, name, permissions=None):
8080
return role
8181

8282

83+
def set_user_single_role(app, username, role_name):
84+
role = create_role(app, role_name)
85+
user = app.appbuilder.sm.find_user(username)
86+
if role not in user.roles:
87+
user.roles = [role]
88+
app.appbuilder.sm.update_user(user)
89+
90+
8391
def delete_role(app, name):
84-
if app.appbuilder.sm.find_role(name):
85-
app.appbuilder.sm.delete_role(name)
92+
if name not in EXISTING_ROLES:
93+
if app.appbuilder.sm.find_role(name):
94+
app.appbuilder.sm.delete_role(name)
8695

8796

8897
def delete_roles(app):
8998
for role in app.appbuilder.sm.get_all_roles():
90-
if role.name not in EXISTING_ROLES:
91-
app.appbuilder.sm.delete_role(role.name)
99+
delete_role(app, role.name)
92100

93101

94102
def delete_user(app, username):

0 commit comments

Comments
 (0)