def random_group_admin(server_api, superuser_token_headers):
new_user = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=new_user)
created_user = r.json()
group_users, _ = get_group_users_and_admins(server_api,
superuser_token_headers)
group_id = random.choice(list(group_users.keys()))
if r.status_code == 200:
r = requests.post(
f'{server_api}{config.API_V1_STR}/groups/{group_id}/admin_users/',
headers=superuser_token_headers,
data={'user_id': created_user['id']})
email, password = new_user['email'], new_user['password']
auth = user_authentication_headers(server_api, email, password)
return group_id, auth
else:
Exception('Unable to execute due to possible server error')
def test_create_user_existing_email(superuser_token_headers):
server_api = get_server_api()
post_data = random_user()
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=post_data,
)
created_user = r.json()
if r.status_code == 200:
assert "id" in created_user
assert True
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=post_data,
)
created_user = r.json()
assert r.status_code == 400
assert "id" not in created_user
def test_create_user_by_superuser_any_group(server_api,
superuser_token_headers):
new_group = random_group()
r = requests.post(f'{server_api}{config.API_V1_STR}/groups/',
headers=superuser_token_headers,
data=new_group)
created_group = r.json()
group_id = created_group['id']
new_user = random_user(group_id)
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=new_user)
expected_fields = [
"id",
"is_active",
"created_at",
"email",
"first_name",
"last_name",
"group",
"groups_admin",
"is_superuser",
]
created_user = r.json()
for expected_field in expected_fields:
assert expected_field in created_user
assert r.status_code == 200
def test_assign_group_admin_by_group_admin(superuser_token_headers):
server_api = get_server_api()
_, group_admin_auth = random_group_admin(server_api,
superuser_token_headers)
new_group = random_group()
r = requests.post(
f"{server_api}{config.API_V1_STR}/groups/",
headers=superuser_token_headers,
data=new_group,
)
created_group = r.json()
group_id = created_group["id"]
new_user = random_user()
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=new_user,
)
created_user = r.json()
user_id = created_user["id"]
request_data = {"user_id": user_id}
r = requests.post(
f"{server_api}{config.API_V1_STR}/groups/{group_id}/admin_users/",
headers=group_admin_auth,
data=request_data,
)
assert r.status_code == 400
def test_create_user_new_email(superuser_token_headers):
server_api = get_server_api()
post_data = random_user()
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=post_data,
)
created_user = r.json()
if r.status_code == 200:
assert "id" in created_user
user_id = created_user["id"]
r = requests.get(
f"{server_api}{config.API_V1_STR}/users/{user_id}",
headers=superuser_token_headers,
data=post_data,
)
code, response = r.status_code, r.json()
for key in response:
assert created_user[key] == response[key]
if r.status_code == 400:
assert "id" not in created_user
def test_create_user_by_superuser(superuser_token_headers):
server_api = get_server_api()
new_user = random_user()
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=new_user,
)
expected_fields = [
"id",
"is_active",
"created_at",
"email",
"first_name",
"last_name",
"group",
"groups_admin",
"is_superuser",
]
created_user = r.json()
for expected_field in expected_fields:
assert expected_field in created_user
assert r.status_code == 200
def test_assign_group_admin_by_group_admin(server_api,
superuser_token_headers):
_, group_admin_auth = random_group_admin(server_api,
superuser_token_headers)
new_group = random_group()
r = requests.post(f'{server_api}{config.API_V1_STR}/groups/',
headers=superuser_token_headers,
data=new_group)
created_group = r.json()
group_id = created_group['id']
new_user = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=new_user)
created_user = r.json()
user_id = created_user['id']
request_data = {"user_id": user_id}
r = requests.post(
f'{server_api}{config.API_V1_STR}/groups/{group_id}/admin_users/',
headers=group_admin_auth,
data=request_data)
assert r.status_code == 400
def test_fetch_group_admin_user_groups(server_api, superuser_token_headers):
new_user = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=new_user)
created_user = r.json()
group_users, _ = get_group_users_and_admins(server_api,
superuser_token_headers)
group_id = random.choice(list(group_users.keys()))
if r.status_code == 200:
r = requests.post(
f'{server_api}{config.API_V1_STR}/groups/{group_id}/admin_users/',
headers=superuser_token_headers,
data={'user_id': created_user['id']})
email, password = new_user['email'], new_user['password']
auth = user_authentication_headers(server_api, email, password)
r = requests.get(f'{server_api}{config.API_V1_STR}/groups/',
headers=auth)
groups = r.json()
assert len(groups) == 1
def test_assign_group_admin_by_normal_user(superuser_token_headers):
server_api = get_server_api()
new_user = random_user()
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=new_user,
)
created_user = r.json()
if r.status_code == 200:
email, password = new_user["email"], new_user["password"]
auth = user_authentication_headers(server_api, email, password)
new_group = random_group()
r = requests.post(
f"{server_api}{config.API_V1_STR}/groups/",
headers=superuser_token_headers,
data=new_group,
)
created_group = r.json()
group_id = created_group["id"]
new_user = random_user()
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=new_user,
)
created_user = r.json()
user_id = created_user["id"]
request_data = {"user_id": user_id}
r = requests.post(
f"{server_api}{config.API_V1_STR}/groups/{group_id}/admin_users/",
headers=auth,
data=request_data,
)
assert r.status_code == 400
def test_create_user_by_normal_user(server_api, superuser_token_headers):
new_user = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=new_user)
created_user = r.json()
if r.status_code == 200:
email, password = new_user['email'], new_user['password']
auth = user_authentication_headers(server_api, email, password)
new_user = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=auth,
data=new_user)
assert r.status_code == 400
def test_create_user_by_group_admin(server_api, superuser_token_headers):
_, group_admin_auth = random_group_admin(server_api,
superuser_token_headers)
new_user = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=group_admin_auth,
data=new_user)
assert r.status_code == 400
def test_assign_group_admin_by_normal_user(server_api,
superuser_token_headers):
new_user = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=new_user)
created_user = r.json()
if r.status_code == 200:
email, password = new_user['email'], new_user['password']
auth = user_authentication_headers(server_api, email, password)
new_group = random_group()
r = requests.post(f'{server_api}{config.API_V1_STR}/groups/',
headers=superuser_token_headers,
data=new_group)
created_group = r.json()
group_id = created_group['id']
new_user = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=new_user)
created_user = r.json()
user_id = created_user['id']
request_data = {"user_id": user_id}
r = requests.post(
f'{server_api}{config.API_V1_STR}/groups/{group_id}/admin_users/',
headers=auth,
data=request_data)
assert r.status_code == 400
def test_create_user_by_normal_user(superuser_token_headers):
server_api = get_server_api()
new_user = random_user()
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=new_user,
)
created_user = r.json()
if r.status_code == 200:
email, password = new_user["email"], new_user["password"]
auth = user_authentication_headers(server_api, email, password)
new_user = random_user()
r = requests.post(f"{server_api}{config.API_V1_STR}/users/",
headers=auth,
data=new_user)
assert r.status_code == 400
def test_user_group_permissions(superuser_token_headers):
server_api = get_server_api()
new_user = random_user()
r = requests.post(
f"{server_api}{config.API_V1_STR}/users/",
headers=superuser_token_headers,
data=new_user,
)
created_user = r.json()
group_users, _ = get_group_users_and_admins(server_api,
superuser_token_headers)
group_id = random.choice(list(group_users.keys()))
if r.status_code == 200:
r = requests.post(
f"{server_api}{config.API_V1_STR}/groups/{group_id}/admin_users/",
headers=superuser_token_headers,
data={"user_id": created_user["id"]},
)
if r.status_code == 200:
email, password = new_user["email"], new_user["password"]
auth = user_authentication_headers(server_api, email, password)
allowed_users = group_users[group_id]
other_users = [
u for g in group_users for u in group_users[g] if g != group_id
]
for user in allowed_users:
user_id = user["id"]
r = requests.get(
f"{server_api}{config.API_V1_STR}/users/{user_id}",
headers=auth)
assert r.status_code == 200
for user in other_users:
user_id = user["id"]
r = requests.get(
f"{server_api}{config.API_V1_STR}/users/{user_id}",
headers=auth)
assert r.status_code == 400
def test_create_user_existing_email(server_api, superuser_token_headers):
post_data = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=post_data)
created_user = r.json()
if r.status_code == 200:
assert 'id' in created_user
assert True
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=post_data)
created_user = r.json()
assert r.status_code == 400
assert 'id' not in created_user
def test_create_user_new_email(server_api, superuser_token_headers):
post_data = random_user()
r = requests.post(f'{server_api}{config.API_V1_STR}/users/',
headers=superuser_token_headers,
data=post_data)
created_user = r.json()
if r.status_code == 200:
assert 'id' in created_user
user_id = created_user['id']
r = requests.get(f'{server_api}{config.API_V1_STR}/users/{user_id}',
headers=superuser_token_headers,
data=post_data)
code, response = r.status_code, r.json()
for key in response:
assert created_user[key] == response[key]
if r.status_code == 400:
assert 'id' not in created_user
