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