68 lines
2.9 KiB
Python
68 lines
2.9 KiB
Python
"""公网部署安全回归测试。"""
|
|
import pytest
|
|
from fastapi.testclient import TestClient
|
|
from starlette.websockets import WebSocketDisconnect
|
|
|
|
from backend import config
|
|
from backend.main import app
|
|
from backend.security import hash_password, verify_password
|
|
|
|
|
|
def test_passwords_use_bcrypt_and_verify():
|
|
password_hash = hash_password("a-strong-password")
|
|
|
|
assert password_hash.startswith("$2")
|
|
assert verify_password("a-strong-password", password_hash) == (True, False)
|
|
assert verify_password("wrong-password", password_hash) == (False, False)
|
|
|
|
|
|
def test_websocket_rejects_connection_without_jwt():
|
|
with TestClient(app) as unauthenticated_client:
|
|
with pytest.raises(WebSocketDisconnect) as exc_info:
|
|
with unauthenticated_client.websocket_connect("/ws/tasks/task-1"):
|
|
pass
|
|
|
|
assert exc_info.value.code == 1008
|
|
|
|
|
|
def test_production_rejects_default_security_settings(monkeypatch):
|
|
monkeypatch.setattr(config, "APP_ENV", "production")
|
|
monkeypatch.setattr(config, "JWT_SECRET", "ios-build-server-secret-key-change-in-production")
|
|
monkeypatch.setattr(config, "ADMIN_PASSWORD", "admin123")
|
|
monkeypatch.setattr(config, "CORS_ALLOWED_ORIGINS", ["*"])
|
|
monkeypatch.setattr(config, "TRUSTED_HOSTS", ["*"])
|
|
monkeypatch.setenv("TRUSTED_HOSTS", "*")
|
|
|
|
with pytest.raises(RuntimeError, match="生产环境安全配置不完整"):
|
|
config.validate_production_security()
|
|
|
|
|
|
def test_config_routes_limit_regular_users_to_apps_and_versions(client, tmp_config):
|
|
"""普通账号可管理 Apps 与版本号，不能读取或修改管理员配置。"""
|
|
response = client.post("/api/users", json={
|
|
"username": "builder",
|
|
"password": "builder-password-123",
|
|
"is_admin": False,
|
|
})
|
|
assert response.status_code == 200
|
|
|
|
login = client.post("/api/auth/login", json={
|
|
"username": "builder",
|
|
"password": "builder-password-123",
|
|
})
|
|
token = login.json()["token"]
|
|
headers = {"Authorization": f"Bearer {token}"}
|
|
|
|
assert client.get("/api/config/apps", headers=headers).status_code == 200
|
|
assert client.post("/api/config/apps", json={
|
|
"name": "普通用户 App", "server": "测试环境",
|
|
}, headers=headers).status_code == 200
|
|
assert client.get("/api/config/versions", headers=headers).status_code == 200
|
|
versions = client.put("/api/config/versions", json={"app_ver": "2.196.0"}, headers=headers)
|
|
assert versions.status_code == 200
|
|
assert versions.json()["build_ver"] == "2.196.0.0"
|
|
assert client.get("/api/config", headers=headers).status_code == 403
|
|
assert client.get("/api/config/servers", headers=headers).status_code == 403
|
|
assert client.put("/api/config/build", json={"max_concurrent_builds": 1}, headers=headers).status_code == 403
|
|
assert client.put("/api/config/upload", json={"mode": "oss"}, headers=headers).status_code == 403
|