369 lines
13 KiB
Python
369 lines
13 KiB
Python
import json
|
|
|
|
from . import IntegrationApiManagement
|
|
|
|
|
|
class TestIntegrationApiManagementAdminUser(IntegrationApiManagement):
|
|
|
|
def test_accounts_empty_get(
|
|
self, client, initial_data, # noqa: F811
|
|
basic_auth_admin_headers): # noqa: F811
|
|
res = client.get("/api/v1/pdnsadmin/accounts",
|
|
headers=basic_auth_admin_headers)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 200
|
|
assert data == []
|
|
|
|
def test_users_empty_get(
|
|
self, client, initial_data, # noqa: F811
|
|
test_admin_user, test_user, # noqa: F811
|
|
basic_auth_admin_headers): # noqa: F811
|
|
res = client.get("/api/v1/pdnsadmin/users",
|
|
headers=basic_auth_admin_headers)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 200
|
|
# Initally contains 2 records
|
|
assert len(data) == 2
|
|
for user in data:
|
|
assert user["username"] in (test_admin_user, test_user)
|
|
|
|
def test_accounts(
|
|
self, client, initial_data, # noqa: F811
|
|
account_data, # noqa: F811
|
|
basic_auth_admin_headers): # noqa: F811
|
|
account_name = account_data["name"]
|
|
self.client = client
|
|
self.basic_auth_admin_headers = basic_auth_admin_headers
|
|
|
|
# Create account
|
|
res = client.post(
|
|
"/api/v1/pdnsadmin/accounts",
|
|
headers=basic_auth_admin_headers,
|
|
data=json.dumps(account_data),
|
|
content_type="application/json",
|
|
)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 201
|
|
|
|
# Check account
|
|
data = self.check_account(account_data)
|
|
account_id = data["id"]
|
|
|
|
updated = account_data.copy()
|
|
# Update and check values
|
|
for upd_key in ["description", "contact", "mail"]:
|
|
upd_value = "upd-{}".format(account_data[upd_key])
|
|
|
|
# Update
|
|
data = {"name": account_name, upd_key: upd_value}
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/accounts/{}".format(account_id),
|
|
data=json.dumps(data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
updated[upd_key] = upd_value
|
|
|
|
# Check
|
|
data = self.check_account(updated)
|
|
|
|
# Update to defaults
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/accounts/{}".format(account_id),
|
|
data=json.dumps(account_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
|
|
# Check account
|
|
res = client.get(
|
|
"/api/v1/pdnsadmin/accounts/{}".format(account_name),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 200
|
|
assert isinstance(data, dict)
|
|
assert len(data) == 7
|
|
assert data.get('id', None)
|
|
account_id = data["id"]
|
|
for key, value in account_data.items():
|
|
assert data[key] == value
|
|
|
|
# Cleanup (delete account)
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/accounts/{}".format(account_id),
|
|
data=json.dumps(account_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
|
|
# Get non-existing account (should fail)
|
|
data = self.get_account(account_name, status_code=404)
|
|
|
|
# Update non-existing account (should fail)
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/accounts/{}".format(account_id),
|
|
data=json.dumps(account_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
# Delete non-existing account (should fail)
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/accounts/{}".format(account_id),
|
|
data=json.dumps(account_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
def test_users(
|
|
self, client, initial_data, # noqa: F811
|
|
user1_data, # noqa: F811
|
|
basic_auth_admin_headers): # noqa: F811
|
|
user1name = user1_data["username"]
|
|
self.client = client
|
|
self.basic_auth_admin_headers = basic_auth_admin_headers
|
|
|
|
# Create user (user1)
|
|
res = client.post(
|
|
"/api/v1/pdnsadmin/users",
|
|
headers=basic_auth_admin_headers,
|
|
data=json.dumps(user1_data),
|
|
content_type="application/json",
|
|
)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 201
|
|
assert isinstance(data, dict)
|
|
assert len(data) == 6
|
|
assert data.get('id', None)
|
|
|
|
# Check user
|
|
user1 = self.check_user(user1_data, data)
|
|
user1_id = user1["id"]
|
|
|
|
updated = user1_data.copy()
|
|
# Update and check values
|
|
for upd_key in ["firstname", "lastname", "email"]:
|
|
upd_value = "upd-{}".format(user1_data[upd_key])
|
|
|
|
# Update
|
|
data = {"username": user1name, upd_key: upd_value}
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/users/{}".format(user1_id),
|
|
data=json.dumps(data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
updated[upd_key] = upd_value
|
|
|
|
# Check
|
|
data = self.check_user(updated)
|
|
|
|
# Update to defaults
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/users/{}".format(user1_id),
|
|
data=json.dumps(user1_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
|
|
# Check user
|
|
self.check_user(user1_data)
|
|
|
|
# Cleanup (delete user)
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/users/{}".format(user1_id),
|
|
data=json.dumps(user1_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
|
|
# Get non-existing user (should fail)
|
|
data = self.get_user(user1name, status_code=404)
|
|
|
|
# Update non-existing user (should fail)
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/users/{}".format(user1_id),
|
|
data=json.dumps(user1_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
# Delete non-existing user (should fail)
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/users/{}".format(user1_id),
|
|
data=json.dumps(user1_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
def test_account_users(
|
|
self, client, initial_data, # noqa: F811
|
|
test_user, account_data, user1_data, # noqa: F811
|
|
basic_auth_admin_headers): # noqa: F811
|
|
self.client = client
|
|
self.basic_auth_admin_headers = basic_auth_admin_headers
|
|
test_user_id = self.get_user(test_user)["id"]
|
|
|
|
# Create account
|
|
res = client.post(
|
|
"/api/v1/pdnsadmin/accounts",
|
|
headers=basic_auth_admin_headers,
|
|
data=json.dumps(account_data),
|
|
content_type="application/json",
|
|
)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 201
|
|
|
|
# Check account
|
|
data = self.check_account(account_data)
|
|
account_id = data["id"]
|
|
|
|
# Create user1
|
|
res = client.post(
|
|
"/api/v1/pdnsadmin/users",
|
|
headers=basic_auth_admin_headers,
|
|
data=json.dumps(user1_data),
|
|
content_type="application/json",
|
|
)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 201
|
|
assert isinstance(data, dict)
|
|
assert len(data) == 6
|
|
assert data.get('id', None)
|
|
|
|
# Check user
|
|
user1 = self.check_user(user1_data, data)
|
|
user1_id = user1["id"]
|
|
|
|
# Assert test account has no users
|
|
res = client.get(
|
|
"/api/v1/pdnsadmin/accounts/users/{}".format(account_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 200
|
|
assert data == []
|
|
|
|
# Assert unlinking an unlinked account fails
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/accounts/users/{}/{}".format(
|
|
account_id, user1_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
# Link user to account
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/accounts/users/{}/{}".format(
|
|
account_id, user1_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
|
|
# Check user is linked to account
|
|
res = client.get(
|
|
"/api/v1/pdnsadmin/accounts/users/{}".format(account_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 200
|
|
assert len(data) == 1
|
|
self.check_user(user1_data, data[0])
|
|
|
|
# Unlink user from account
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/accounts/users/{}/{}".format(
|
|
account_id, user1_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
|
|
# Check user is unlinked from account
|
|
res = client.get(
|
|
"/api/v1/pdnsadmin/accounts/users/{}".format(account_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
data = res.get_json(force=True)
|
|
assert res.status_code == 200
|
|
assert data == []
|
|
|
|
# Unlink unlinked user from account (should fail)
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/accounts/users/{}/{}".format(
|
|
account_id, user1_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
# Cleanup (delete user)
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/users/{}".format(user1_id),
|
|
data=json.dumps(user1_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
|
|
# Link non-existing user to account (should fail)
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/accounts/users/{}/{}".format(
|
|
account_id, user1_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
# Unlink non-exiting user from account (should fail)
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/accounts/users/{}/{}".format(
|
|
account_id, user1_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
# Cleanup (delete account)
|
|
res = client.delete(
|
|
"/api/v1/pdnsadmin/accounts/{}".format(account_id),
|
|
data=json.dumps(account_data),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 204
|
|
|
|
# List users in non-existing account (should fail)
|
|
res = client.get(
|
|
"/api/v1/pdnsadmin/accounts/users/{}".format(account_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|
|
|
|
# Link existing user to non-existing account (should fail)
|
|
res = client.put(
|
|
"/api/v1/pdnsadmin/accounts/users/{}/{}".format(
|
|
account_id, test_user_id),
|
|
headers=basic_auth_admin_headers,
|
|
content_type="application/json",
|
|
)
|
|
assert res.status_code == 404
|