164 lines
6.5 KiB
Python
164 lines
6.5 KiB
Python
from __future__ import annotations
|
|
|
|
import re
|
|
from uuid import uuid4
|
|
|
|
from fastapi.testclient import TestClient
|
|
|
|
from app.main import app
|
|
from app.models.workspace import WorkspaceRepository
|
|
|
|
UUID4_RE = re.compile(
|
|
r"^[0-9a-f]{8}-[0-9a-f]{4}-4[0-9a-f]{3}-[89ab][0-9a-f]{3}-[0-9a-f]{12}$",
|
|
re.IGNORECASE,
|
|
)
|
|
|
|
|
|
def _install_workspace_repo(tmp_path, monkeypatch) -> WorkspaceRepository:
|
|
from app.models import workspace as workspace_module
|
|
|
|
repo = WorkspaceRepository(base_path=tmp_path / "workspaces")
|
|
monkeypatch.setattr(workspace_module, "_workspace_repo", repo)
|
|
return repo
|
|
|
|
|
|
def test_workspace_repository_persists_workspace_specific_portfolio_config(tmp_path) -> None:
|
|
repo = WorkspaceRepository(base_path=tmp_path / "workspaces")
|
|
workspace_id = str(uuid4())
|
|
|
|
created = repo.create_workspace(workspace_id)
|
|
created.loan_amount = 123_456.0
|
|
repo.save_portfolio_config(workspace_id, created)
|
|
|
|
reloaded = repo.load_portfolio_config(workspace_id)
|
|
|
|
assert repo.workspace_exists(workspace_id)
|
|
assert reloaded.loan_amount == 123_456.0
|
|
assert reloaded.gold_value == created.gold_value
|
|
|
|
|
|
def test_root_without_workspace_cookie_shows_welcome_page(tmp_path, monkeypatch) -> None:
|
|
_install_workspace_repo(tmp_path, monkeypatch)
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get("/")
|
|
|
|
assert response.status_code == 200
|
|
assert "Create a private workspace URL" in response.text
|
|
assert "Get started" in response.text
|
|
|
|
|
|
def test_bootstrap_endpoint_creates_workspace_cookie_and_redirects(tmp_path, monkeypatch) -> None:
|
|
repo = _install_workspace_repo(tmp_path, monkeypatch)
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get("/workspaces/bootstrap", follow_redirects=False)
|
|
|
|
assert response.status_code in {302, 303, 307}
|
|
location = response.headers["location"]
|
|
workspace_id = location.strip("/")
|
|
assert UUID4_RE.match(workspace_id)
|
|
assert repo.workspace_exists(workspace_id)
|
|
assert response.cookies.get("workspace_id") == workspace_id
|
|
|
|
|
|
def test_root_with_valid_workspace_cookie_redirects_to_workspace(tmp_path, monkeypatch) -> None:
|
|
repo = _install_workspace_repo(tmp_path, monkeypatch)
|
|
workspace_id = str(uuid4())
|
|
repo.create_workspace(workspace_id)
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get("/", cookies={"workspace_id": workspace_id}, follow_redirects=False)
|
|
|
|
assert response.status_code in {302, 303, 307}
|
|
assert response.headers["location"] == f"/{workspace_id}"
|
|
|
|
|
|
def test_unknown_workspace_route_shows_friendly_recovery_path(tmp_path, monkeypatch) -> None:
|
|
_install_workspace_repo(tmp_path, monkeypatch)
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get("/00000000-0000-4000-8000-000000000000")
|
|
|
|
assert response.status_code == 200
|
|
assert "Workspace not found" in response.text
|
|
assert "Get started" in response.text
|
|
|
|
|
|
def test_workspace_settings_round_trip_uses_workspace_storage(tmp_path, monkeypatch) -> None:
|
|
repo = _install_workspace_repo(tmp_path, monkeypatch)
|
|
workspace_id = str(uuid4())
|
|
config = repo.create_workspace(workspace_id)
|
|
config.monthly_budget = 9_999.0
|
|
repo.save_portfolio_config(workspace_id, config)
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get(f"/{workspace_id}/settings")
|
|
|
|
assert response.status_code == 200
|
|
assert "Settings" in response.text
|
|
assert "9,999" in response.text or "9999" in response.text
|
|
|
|
|
|
def test_workspace_pages_use_workspace_scoped_navigation_links(tmp_path, monkeypatch) -> None:
|
|
repo = _install_workspace_repo(tmp_path, monkeypatch)
|
|
workspace_id = str(uuid4())
|
|
repo.create_workspace(workspace_id)
|
|
|
|
with TestClient(app) as client:
|
|
response = client.get(f"/{workspace_id}")
|
|
if f"/{workspace_id}/hedge" not in response.text:
|
|
response = client.get(f"/{workspace_id}")
|
|
|
|
assert response.status_code == 200
|
|
assert f"/{workspace_id}/hedge" in response.text
|
|
assert f"/{workspace_id}/backtests" in response.text
|
|
assert f"/{workspace_id}/event-comparison" in response.text
|
|
assert f"/{workspace_id}/settings" in response.text
|
|
|
|
|
|
def test_workspace_routes_seed_page_defaults_from_workspace_portfolio_config(tmp_path, monkeypatch) -> None:
|
|
repo = _install_workspace_repo(tmp_path, monkeypatch)
|
|
workspace_id = str(uuid4())
|
|
config = repo.create_workspace(workspace_id)
|
|
config.entry_basis_mode = "weight"
|
|
config.entry_price = 4_400.0
|
|
config.gold_ounces = 220.0
|
|
config.gold_value = 968_000.0
|
|
config.loan_amount = 222_000.0
|
|
config.margin_threshold = 0.80
|
|
config.monthly_budget = 12_345.0
|
|
repo.save_portfolio_config(workspace_id, config)
|
|
|
|
with TestClient(app) as client:
|
|
hedge_response = client.get(f"/{workspace_id}/hedge")
|
|
backtests_response = client.get(f"/{workspace_id}/backtests")
|
|
event_response = client.get(f"/{workspace_id}/event-comparison")
|
|
redirect_response = client.get("/backtests", cookies={"workspace_id": workspace_id}, follow_redirects=False)
|
|
|
|
assert hedge_response.status_code == 200
|
|
assert "Monthly hedge budget" in hedge_response.text
|
|
assert "12,345" in hedge_response.text or "12345" in hedge_response.text
|
|
assert "968,000" in hedge_response.text or "968000" in hedge_response.text
|
|
assert "4,400.00/oz" in hedge_response.text or "4400.00/oz" in hedge_response.text
|
|
assert "220 oz" in hedge_response.text
|
|
assert "222,000" in hedge_response.text or "222000" in hedge_response.text
|
|
assert "80.0%" in hedge_response.text
|
|
|
|
assert backtests_response.status_code == 200
|
|
assert "Workspace defaults seed underlying units, loan amount, and margin threshold." in backtests_response.text
|
|
assert "9680" in backtests_response.text or "9,680" in backtests_response.text
|
|
assert "222000" in backtests_response.text or "222,000" in backtests_response.text
|
|
assert "0.8" in backtests_response.text or "80.0%" in backtests_response.text
|
|
|
|
assert event_response.status_code == 200
|
|
assert "Workspace defaults seed underlying units, loan amount, and margin threshold." in event_response.text
|
|
assert "Underlying units" in event_response.text
|
|
assert "Loan amount" in event_response.text
|
|
assert "222,000" in event_response.text or "222000" in event_response.text
|
|
assert "9,680" in event_response.text or "9680" in event_response.text
|
|
assert "80.0%" in event_response.text
|
|
|
|
assert redirect_response.status_code in {302, 303, 307}
|
|
assert redirect_response.headers["location"] == f"/{workspace_id}/backtests"
|