398 lines
16 KiB
Python
398 lines
16 KiB
Python
"""
|
|
Tests fuer die CRUD-Vervollstaendigung (Edit/Delete fuer alles, was die
|
|
Admin-API anlegt, siehe 0007_crud_extras.sql) und die Mehrfachauswahl bei
|
|
Rollenvergaben (role_names: list statt role_name: str).
|
|
|
|
Vormals (bis Teil C des Umsetzungsauftrags) enthielt diese Datei zusaetzlich
|
|
Tests fuer die inzwischen vollstaendig entfernte Mandantenfaehigkeit
|
|
(app/tenancy.py, 0006_tenants.sql) -- siehe app/db/migrations/
|
|
0014_drop_tenants.sql fuer den Rueckbau. Diese Datei heisst deshalb nicht
|
|
mehr test_tenants.py.
|
|
|
|
Deckt ab:
|
|
1) Mehrfachauswahl bei Rollenvergabe (role_names) funktioniert fuer
|
|
Einzel-User UND Benutzergruppen.
|
|
2) Update/Delete-Endpunkte fuer User, Hostgruppen, Hosts, SSH-Keys,
|
|
Benutzergruppen funktionieren wie vorgesehen (inkl. Anonymisieren statt
|
|
Hard-Delete bei vorhandener Audit-Historie, Soft-Delete bei Hosts).
|
|
3) GET /admin/hosts/{id} (Grundlage fuer den "Details"-Fix) liefert den
|
|
vollstaendigen aktuellen Datensatz inkl. SSH-Key-Zuordnungen.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import pyotp
|
|
import pytest
|
|
|
|
from cryptography.hazmat.primitives import serialization
|
|
from cryptography.hazmat.primitives.asymmetric import ed25519
|
|
|
|
|
|
def _key_pem() -> str:
|
|
"""Echtes, unverschluesseltes Schluesselmaterial.
|
|
|
|
Seit Phase 10 prueft POST/PUT /admin/ssh-keys das Material sofort gegen
|
|
asyncssh (damit ein unbrauchbarer Schluessel nicht erst beim ersten
|
|
Verbindungsversuch auffaellt). Platzhalter wie "PEM" werden deshalb --
|
|
voellig korrekt -- mit HTTP 400 abgelehnt; diese Tests hier pruefen CRUD
|
|
und brauchen einen gueltigen Schluessel.
|
|
"""
|
|
return ed25519.Ed25519PrivateKey.generate().private_bytes(
|
|
serialization.Encoding.PEM,
|
|
serialization.PrivateFormat.OpenSSH,
|
|
serialization.NoEncryption(),
|
|
).decode()
|
|
|
|
|
|
async def _create_user(conn, username: str, password: str, *, is_admin: bool = False) -> int:
|
|
from app.security.passwords import hash_password
|
|
|
|
cursor = await conn.execute(
|
|
"INSERT INTO users (username, password_hash, is_admin, must_change_password) "
|
|
"VALUES (?, ?, ?, 0)",
|
|
(username, hash_password(password), int(is_admin)),
|
|
)
|
|
await conn.commit()
|
|
return cursor.lastrowid
|
|
|
|
|
|
async def _login_full(client, username: str, password: str) -> str:
|
|
resp = await client.post("/auth/login", json={"username": username, "password": password})
|
|
assert resp.status_code == 200, resp.text
|
|
pending = resp.json()["pending_token"]
|
|
resp = await client.post("/auth/totp/enroll/start", json={"pending_token": pending})
|
|
assert resp.status_code == 200, resp.text
|
|
provisioning_uri = resp.json()["provisioning_uri"]
|
|
secret = dict(part.split("=") for part in provisioning_uri.split("?", 1)[1].split("&"))["secret"]
|
|
code = pyotp.TOTP(secret).now()
|
|
resp = await client.post("/auth/totp/enroll/confirm", json={"pending_token": pending, "code": code})
|
|
assert resp.status_code == 200, resp.text
|
|
return resp.cookies.get("jh_session")
|
|
|
|
|
|
async def _setup_hostgroup_and_host(conn, *, group_name="team-a", hostname="srv-a"):
|
|
cursor = await conn.execute(
|
|
"INSERT INTO host_groups (name) VALUES (?)", (group_name,)
|
|
)
|
|
hg_id = cursor.lastrowid
|
|
cursor = await conn.execute(
|
|
"INSERT INTO hosts (host_group_id, hostname, address, protocol, port, os_type) "
|
|
"VALUES (?, ?, '10.0.0.9', 'ssh', 22, 'linux')",
|
|
(hg_id, hostname),
|
|
)
|
|
await conn.commit()
|
|
return hg_id, cursor.lastrowid
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# 1) Mehrfachauswahl bei Rollenvergabe (role_names)
|
|
# ---------------------------------------------------------------------------
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_direct_multi_role_grant_for_individual_user_is_retired(client):
|
|
"""Teil D Schritt 5: Direktvergabe an einzelne Benutzer (auch mit
|
|
Mehrfachauswahl -- vormals test_multi_role_grant_for_individual_user,
|
|
siehe FORTSETZUNG_Teil_D.md Abschnitt 1c) ist ersatzlos entfallen. Diese
|
|
Faehigkeit existiert nur noch fuer Benutzergruppen, siehe
|
|
test_multi_role_grant_for_user_group() unten -- deshalb hier bewusst
|
|
KEIN Ersatztest fuer den Einzel-User-Pfad, sondern die Bestaetigung,
|
|
dass er tatsaechlich abgeschaltet ist (statt z.B. nur zu vergessen und
|
|
fuer immer gruen zu bleiben, obwohl der Pfad gar nicht mehr existiert)."""
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "mr_admin", "Correct-Horse-Battery-Staple-G1", is_admin=True)
|
|
member_id = await _create_user(conn, "mr_member", "Correct-Horse-Battery-Staple-G2")
|
|
hg_id, host_id = await _setup_hostgroup_and_host(conn, group_name="mr-team", hostname="mr-srv")
|
|
|
|
await _login_full(client, "mr_admin", "Correct-Horse-Battery-Staple-G1")
|
|
resp = await client.post(
|
|
"/admin/roles/grant",
|
|
json={
|
|
"user_id": member_id, "host_group_id": hg_id,
|
|
"role_names": ["ssh_connect", "file_transfer"],
|
|
},
|
|
)
|
|
assert resp.status_code == 410, resp.text
|
|
|
|
resp = await client.post(
|
|
"/admin/roles/revoke",
|
|
json={"user_id": member_id, "host_group_id": hg_id, "role_name": "file_transfer"},
|
|
)
|
|
assert resp.status_code == 410, resp.text
|
|
|
|
# GET /admin/roles (die abgeleitete Effektiv-Sicht, Schritt 5) zeigt
|
|
# entsprechend keine Rolle fuer member_id -- der Grant-Versuch oben ist
|
|
# tatsaechlich folgenlos geblieben, nicht nur mit 410 quittiert worden.
|
|
resp = await client.get("/admin/roles")
|
|
assert resp.status_code == 200, resp.text
|
|
granted_roles = {r["role_name"] for r in resp.json() if r["user_id"] == member_id}
|
|
assert granted_roles == set()
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_multi_role_grant_for_user_group(client):
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "mrg_admin", "Correct-Horse-Battery-Staple-H1", is_admin=True)
|
|
hg_id, host_id = await _setup_hostgroup_and_host(conn, group_name="mrg-team", hostname="mrg-srv")
|
|
|
|
await _login_full(client, "mrg_admin", "Correct-Horse-Battery-Staple-H1")
|
|
resp = await client.post("/admin/user-groups", json={"name": "mrg-group"})
|
|
group_id = resp.json()["id"]
|
|
|
|
resp = await client.post(
|
|
"/admin/group-roles/grant",
|
|
json={
|
|
"user_group_id": group_id, "host_group_id": hg_id,
|
|
"role_names": ["ssh_connect", "file_transfer"],
|
|
},
|
|
)
|
|
assert resp.status_code == 200, resp.text
|
|
assert set(resp.json()["roles_granted"]) == {"ssh_connect", "file_transfer"}
|
|
|
|
resp = await client.get("/admin/group-roles")
|
|
roles = {r["role_name"] for r in resp.json() if r["user_group_id"] == group_id}
|
|
assert {"ssh_connect", "file_transfer"} <= roles
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# 2) CRUD-Vervollstaendigung: Edit/Delete ueberall
|
|
# ---------------------------------------------------------------------------
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_update_deactivate_and_delete_semantics(client):
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "crud_admin", "Correct-Horse-Battery-Staple-I1", is_admin=True)
|
|
await _login_full(client, "crud_admin", "Correct-Horse-Battery-Staple-I1")
|
|
|
|
resp = await client.post(
|
|
"/admin/users",
|
|
json={"username": "crud_target", "initial_password": "Correct-Horse-Battery-Staple-I2"},
|
|
)
|
|
assert resp.status_code == 201, resp.text
|
|
target_id = resp.json()["id"]
|
|
|
|
# Update: Passwort setzen -> must_change_password/session_version bumped.
|
|
resp = await client.put(f"/admin/users/{target_id}", json={"new_password": "Correct-Horse-Battery-Staple-I3"})
|
|
assert resp.status_code == 200, resp.text
|
|
assert resp.json()["changed"] is True
|
|
|
|
# Deactivate.
|
|
resp = await client.post(f"/admin/users/{target_id}/deactivate")
|
|
assert resp.status_code == 200, resp.text
|
|
resp = await client.get("/admin/users")
|
|
row = next(u for u in resp.json() if u["id"] == target_id)
|
|
assert row["is_active"] is False
|
|
|
|
# Reaktivieren ueber update.
|
|
resp = await client.put(f"/admin/users/{target_id}", json={"is_active": True})
|
|
assert resp.status_code == 200, resp.text
|
|
|
|
# Dieses Konto hat inzwischen Audit-Historie (user_updated/deactivated) ->
|
|
# Loeschen anonymisiert statt hart zu entfernen.
|
|
resp = await client.delete(f"/admin/users/{target_id}")
|
|
assert resp.status_code == 200, resp.text
|
|
assert resp.json()["hard_deleted"] is False
|
|
|
|
resp = await client.get("/admin/users")
|
|
assert all(u["id"] != target_id for u in resp.json()), "Anonymisierter User darf nicht mehr gelistet werden"
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_without_audit_history_is_hard_deleted(client):
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "crud_admin2", "Correct-Horse-Battery-Staple-J1", is_admin=True)
|
|
await _login_full(client, "crud_admin2", "Correct-Horse-Battery-Staple-J1")
|
|
|
|
resp = await client.post(
|
|
"/admin/users",
|
|
json={"username": "throwaway", "initial_password": "Correct-Horse-Battery-Staple-J2"},
|
|
)
|
|
target_id = resp.json()["id"]
|
|
|
|
# Das einzige Audit-Ereignis, das diesen User referenziert, waere sein
|
|
# eigenes Login -- er hat sich nie eingeloggt, also existiert keine
|
|
# audit_log-Zeile mit user_id=target_id -> Hard-Delete moeglich.
|
|
resp = await client.delete(f"/admin/users/{target_id}")
|
|
assert resp.status_code == 200, resp.text
|
|
assert resp.json()["hard_deleted"] is True
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_cannot_delete_own_account(client):
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "self_admin", "Correct-Horse-Battery-Staple-K1", is_admin=True)
|
|
await _login_full(client, "self_admin", "Correct-Horse-Battery-Staple-K1")
|
|
resp = await client.get("/auth/me")
|
|
my_id = resp.json()["id"]
|
|
resp = await client.delete(f"/admin/users/{my_id}")
|
|
assert resp.status_code == 400
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_host_group_and_host_update_delete(client):
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "hcrud_admin", "Correct-Horse-Battery-Staple-L1", is_admin=True)
|
|
await _login_full(client, "hcrud_admin", "Correct-Horse-Battery-Staple-L1")
|
|
|
|
resp = await client.post("/admin/host-groups", json={"name": "hcrud-team"})
|
|
hg_id = resp.json()["id"]
|
|
resp = await client.put(f"/admin/host-groups/{hg_id}", json={"description": "Aktualisiert"})
|
|
assert resp.status_code == 200, resp.text
|
|
|
|
resp = await client.post(
|
|
"/admin/hosts",
|
|
json={
|
|
"host_group_id": hg_id, "hostname": "hcrud-srv", "address": "10.2.0.1",
|
|
"protocol": "ssh", "port": 22, "os_type": "linux",
|
|
},
|
|
)
|
|
host_id = resp.json()["id"]
|
|
|
|
resp = await client.put(f"/admin/hosts/{host_id}", json={"hostname": "hcrud-srv-renamed"})
|
|
assert resp.status_code == 200, resp.text
|
|
resp = await client.get(f"/admin/hosts/{host_id}")
|
|
assert resp.json()["hostname"] == "hcrud-srv-renamed"
|
|
|
|
# Hostgruppe kann nicht geloescht werden, solange sie noch (auch inaktive) Hosts enthaelt.
|
|
resp = await client.delete(f"/admin/host-groups/{hg_id}")
|
|
assert resp.status_code == 409, resp.text
|
|
|
|
# Host-Delete ist standardmaessig Soft-Delete.
|
|
resp = await client.delete(f"/admin/hosts/{host_id}")
|
|
assert resp.status_code == 200, resp.text
|
|
assert resp.json()["hard_deleted"] is False
|
|
resp = await client.get(f"/admin/hosts/{host_id}")
|
|
assert resp.json()["is_active"] is False
|
|
|
|
# ?hard=true entfernt ihn tatsaechlich (keine Sitzungshistorie vorhanden).
|
|
resp = await client.delete(f"/admin/hosts/{host_id}?hard=true")
|
|
assert resp.status_code == 200, resp.text
|
|
assert resp.json()["hard_deleted"] is True
|
|
resp = await client.get(f"/admin/hosts/{host_id}")
|
|
assert resp.status_code == 404
|
|
|
|
resp = await client.delete(f"/admin/host-groups/{hg_id}")
|
|
assert resp.status_code == 200, resp.text
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_ssh_key_update_rotate_and_delete(client):
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "kcrud_admin", "Correct-Horse-Battery-Staple-M1", is_admin=True)
|
|
await _login_full(client, "kcrud_admin", "Correct-Horse-Battery-Staple-M1")
|
|
|
|
resp = await client.post(
|
|
"/admin/ssh-keys",
|
|
json={"label": "orig-key", "private_key_pem": _key_pem(), "public_key": "PUB", "key_type": "ed25519"},
|
|
)
|
|
key_id = resp.json()["id"]
|
|
|
|
resp = await client.put(f"/admin/ssh-keys/{key_id}", json={"label": "renamed-key"})
|
|
assert resp.status_code == 200, resp.text
|
|
|
|
# Rotation erfordert alle drei Felder gemeinsam (400 noch vor jeder
|
|
# Materialpruefung -- der Platzhalter hier ist deshalb Absicht).
|
|
resp = await client.put(f"/admin/ssh-keys/{key_id}", json={"private_key_pem": "NEWPEM"})
|
|
assert resp.status_code == 400, resp.text
|
|
resp = await client.put(
|
|
f"/admin/ssh-keys/{key_id}",
|
|
json={"private_key_pem": _key_pem(), "public_key": "NEWPUB", "key_type": "rsa-4096"},
|
|
)
|
|
assert resp.status_code == 200, resp.text
|
|
|
|
resp = await client.delete(f"/admin/ssh-keys/{key_id}")
|
|
assert resp.status_code == 200, resp.text
|
|
resp = await client.get("/admin/ssh-keys")
|
|
assert all(k["id"] != key_id for k in resp.json())
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_user_group_update_and_delete(client):
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "gcrud_admin", "Correct-Horse-Battery-Staple-N1", is_admin=True)
|
|
await _login_full(client, "gcrud_admin", "Correct-Horse-Battery-Staple-N1")
|
|
|
|
resp = await client.post("/admin/user-groups", json={"name": "gcrud-team"})
|
|
group_id = resp.json()["id"]
|
|
resp = await client.put(f"/admin/user-groups/{group_id}", json={"description": "Aktualisiert"})
|
|
assert resp.status_code == 200, resp.text
|
|
|
|
resp = await client.delete(f"/admin/user-groups/{group_id}")
|
|
assert resp.status_code == 200, resp.text
|
|
resp = await client.get("/admin/user-groups")
|
|
assert all(g["id"] != group_id for g in resp.json())
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# 3) GET /admin/hosts/{id} -- Grundlage fuer den "Details"-Fix
|
|
# ---------------------------------------------------------------------------
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_host_detail_endpoint_includes_ssh_keys_and_rdp_flag(client):
|
|
from app.db import get_db
|
|
|
|
conn = get_db()
|
|
await _create_user(conn, "detail_admin", "Correct-Horse-Battery-Staple-O1", is_admin=True)
|
|
await _login_full(client, "detail_admin", "Correct-Horse-Battery-Staple-O1")
|
|
|
|
resp = await client.post("/admin/host-groups", json={"name": "detail-team"})
|
|
hg_id = resp.json()["id"]
|
|
resp = await client.post(
|
|
"/admin/hosts",
|
|
json={
|
|
"host_group_id": hg_id, "hostname": "win-srv", "address": "10.3.0.1",
|
|
"protocol": "rdp", "port": 3389, "os_type": "windows",
|
|
},
|
|
)
|
|
host_id = resp.json()["id"]
|
|
|
|
resp = await client.post(
|
|
"/admin/ssh-keys",
|
|
json={"label": "detail-key", "private_key_pem": _key_pem(), "public_key": "PUB", "key_type": "ed25519"},
|
|
)
|
|
key_id = resp.json()["id"]
|
|
resp = await client.post(f"/admin/hosts/{host_id}/ssh-keys/{key_id}")
|
|
assert resp.status_code == 200, resp.text
|
|
|
|
# Migration 0012: RDP-Zugangsdaten sind ein eigenstaendiges Objekt, das
|
|
# separat angelegt und dann dem Host zugewiesen wird (wie ein SSH-Key).
|
|
resp = await client.post(
|
|
"/admin/rdp-credentials",
|
|
json={"label": "detail-cred", "username": "Administrator",
|
|
"password": "Correct-Horse-Battery-Staple-O2"},
|
|
)
|
|
assert resp.status_code == 201, resp.text
|
|
credential_id = resp.json()["id"]
|
|
resp = await client.post(f"/admin/hosts/{host_id}/rdp-credentials/{credential_id}", json={})
|
|
assert resp.status_code == 200, resp.text
|
|
|
|
resp = await client.get(f"/admin/hosts/{host_id}")
|
|
assert resp.status_code == 200, resp.text
|
|
data = resp.json()
|
|
assert data["hostname"] == "win-srv"
|
|
assert any(k["id"] == key_id for k in data["ssh_keys"])
|
|
assert data["rdp_credentials_set"] is True
|
|
assert data["rdp_credentials_id"] == credential_id
|
|
|
|
resp = await client.get("/admin/rdp-credentials")
|
|
assert resp.status_code == 200, resp.text
|
|
row = next(r for r in resp.json() if r["id"] == credential_id)
|
|
assert any(h["id"] == host_id for h in row["assigned_hosts"])
|
|
|
|
resp = await client.get("/admin/hosts/424242")
|
|
assert resp.status_code == 404
|