def test_cli_add_user_role(self):
args = self.parser.parse_args([
'users',
'create',
'--username',
'test4',
'--lastname',
'doe',
'--firstname',
'jon',
'--email',
TEST_USER1_EMAIL,
'--role',
'Viewer',
'--use-random-password',
])
user_command.users_create(args)
self.assertFalse(
_does_user_belong_to_role(appbuilder=self.appbuilder,
email=TEST_USER1_EMAIL,
rolename='Op'),
"User should not yet be a member of role 'Op'",
)
args = self.parser.parse_args(
['users', 'add-role', '--username', 'test4', '--role', 'Op'])
user_command.users_manage_role(args, remove=False)
self.assertTrue(
_does_user_belong_to_role(appbuilder=self.appbuilder,
email=TEST_USER1_EMAIL,
rolename='Op'),
"User should have been added to role 'Op'",
)
def test_cli_remove_user_role(self):
args = self.parser.parse_args(
[
'users',
'create',
'--username',
'test4',
'--lastname',
'doe',
'--firstname',
'jon',
'--email',
TEST_USER1_EMAIL,
'--role',
'Viewer',
'--use-random-password',
]
)
user_command.users_create(args)
assert _does_user_belong_to_role(
appbuilder=self.appbuilder, email=TEST_USER1_EMAIL, rolename='Viewer'
), "User should have been created with role 'Viewer'"
args = self.parser.parse_args(['users', 'remove-role', '--username', 'test4', '--role', 'Viewer'])
user_command.users_manage_role(args, remove=True)
assert not _does_user_belong_to_role(
appbuilder=self.appbuilder, email=TEST_USER1_EMAIL, rolename='Viewer'
), "User should have been removed from role 'Viewer'"
