mirror of
https://github.com/wizarrrr/wizarr.git
synced 2026-06-11 23:34:47 -04:00
22d7e5dcc5
- Fix test database isolation by cleaning WizardStep data before tests run - Remove unused imports and fix import ordering - Replace setattr with direct attribute assignment - Use contextlib.suppress for cleaner exception handling - Fix nested if statement in wizard_export_import.py Fixes the failing CI checks for PR #817. 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
78 lines
2.3 KiB
Python
78 lines
2.3 KiB
Python
from __future__ import annotations
|
|
|
|
import contextlib
|
|
from pathlib import Path
|
|
|
|
from app.extensions import db
|
|
from app.models import WizardStep
|
|
|
|
|
|
def _enable_wizard_access(client):
|
|
with client.session_transaction() as sess:
|
|
sess["wizard_access"] = "test-code"
|
|
|
|
|
|
def test_db_step_requires_interaction_renders_next_disabled(app, client):
|
|
# Arrange: create a DB-backed step that requires interaction
|
|
with app.app_context():
|
|
step = WizardStep(
|
|
server_type="plex",
|
|
position=0,
|
|
title="Download",
|
|
markdown="# Please download\n[Click me](https://example.com)",
|
|
require_interaction=True,
|
|
)
|
|
db.session.add(step)
|
|
db.session.commit()
|
|
|
|
_enable_wizard_access(client)
|
|
|
|
# Act
|
|
resp = client.get("/wizard/plex/0", headers={"HX-Request": "true"})
|
|
|
|
# Assert
|
|
assert resp.status_code == 200
|
|
body = resp.data.decode()
|
|
assert 'id="next-btn"' in body
|
|
# Next should be disabled when require_interaction is true
|
|
assert 'data-disabled="1"' in body or 'aria-disabled="true"' in body
|
|
|
|
|
|
def test_file_based_front_matter_require_renders_next_disabled(app, client, tmp_path):
|
|
# Arrange: create a temporary server folder with a markdown step that sets require: true
|
|
wizard_steps_root = (
|
|
Path("app/blueprints/wizard/routes.py").resolve().parent.parent.parent.parent
|
|
/ "wizard_steps"
|
|
)
|
|
test_server_dir = wizard_steps_root / "requiretest"
|
|
test_server_dir.mkdir(parents=True, exist_ok=True)
|
|
md_path = test_server_dir / "01_test.md"
|
|
md_path.write_text(
|
|
"---\n"
|
|
"title: Require Test\n"
|
|
"require: true\n"
|
|
"---\n\n"
|
|
"# Step\n\n"
|
|
"[Do it](https://example.com)\n",
|
|
encoding="utf-8",
|
|
)
|
|
|
|
_enable_wizard_access(client)
|
|
|
|
try:
|
|
# Act
|
|
resp = client.get("/wizard/requiretest/0", headers={"HX-Request": "true"})
|
|
|
|
# Assert
|
|
assert resp.status_code == 200
|
|
body = resp.data.decode()
|
|
assert 'id="next-btn"' in body
|
|
assert 'data-disabled="1"' in body or 'aria-disabled="true"' in body
|
|
finally:
|
|
# Cleanup
|
|
with contextlib.suppress(FileNotFoundError):
|
|
md_path.unlink()
|
|
with contextlib.suppress(OSError):
|
|
# Directory not empty or other issue; ignore in tests
|
|
test_server_dir.rmdir()
|