def test_user_on_delete_is_admin_delete_ok(client, superuser):
create_user("test-user", "test-password")
resp = client.simulate_delete(
"/api/test/users/test-user",
headers={"Authorization": f"Token {superuser}"})
assert resp.status_code == 200
assert resp.json["status"] == "Success"
def test_iplistresource_on_delete_ok(client, superuser):
ip_list = IPList(name="test-list", created_by=User.get_by_token(superuser))
ip_list.save()
resp = client.simulate_delete(
"/api/test/iplists/test-list", headers={"Authorization": f"Token {superuser}"}
)
assert resp.status_code == 200
assert resp.json["status"] == "Success"
def test_iplistitemresource_on_delete_remove_all(client, superuser):
ip_list = IPList(name="test-list", created_by=User.get_by_token(superuser))
ip_list.save()
json = {"ips": ["1.1.1.1", "2.2.2.2"], "note": "test note"}
resp = client.simulate_post(
"/api/test/iplists/test-list/items",
headers={"Authorization": f"Token {superuser}"},
json=json,
)
assert IPListItem.select().where((IPListItem.ip_list == ip_list)).count() == 2
json = {"ips": ["2.2.2.2", "1.1.1.1"]}
resp = client.simulate_delete(
"/api/test/iplists/test-list/items",
headers={"Authorization": f"Token {superuser}"},
json=json,
)
assert resp.status_code == 200
assert resp.json["count_removed"] == 2
assert IPListItem.select().where((IPListItem.ip_list == ip_list)).count() == 0
def test_user_on_delete_is_admin_delete_self(client, superuser):
resp = client.simulate_delete(
"/api/test/users/superuser",
headers={"Authorization": f"Token {superuser}"})
assert resp.status_code == 400
def test_user_on_delete_is_not_admin(client, superuser):
u = create_user("test-user", "test-password")
resp = client.simulate_delete("/api/test/users/superuser",
headers={"Authorization": f"Token {u}"})
assert resp.status_code == 401