tests: add customize and data handler tests (35 new)

tests/test_handlers_data.py

@@ -0,0 +1,142 @@
+import json
+from unittest.mock import patch
+
+from tinyweb.db import get_db, return_db
+from tinyweb.handlers import (
+    handle_export, handle_import_form, handle_import_submit,
+    handle_reindex_form, handle_reindex_submit,
+)
+
+
+def _all_urls():
+    db = get_db()
+    try:
+        return {r["url"] for r in db.execute("SELECT url FROM pages").fetchall()}
+    finally:
+        return_db(db)
+
+
+def test_export_returns_json(seeded_db, csrf_session):
+    resp = handle_export({})
+    assert resp["status"] == 200
+    assert resp["content_type"] == "application/json"
+    data = json.loads(resp["body"])
+    assert len(data) == 4
+    assert data[0]["url"] == "https://example.com/rust-intro"
+    assert data[0]["title"] == "Rust Intro"
+
+
+def test_export_paginates(seeded_db, csrf_session):
+    from tinyweb.handlers.data import MAX_EXPORT
+    resp = handle_export({"batch": ["0"]})
+    data = json.loads(resp["body"])
+    assert len(data) <= MAX_EXPORT
+
+
+def test_export_respects_batch_param(seeded_db, csrf_session):
+    resp = handle_export({"batch": ["99"]})
+    data = json.loads(resp["body"])
+    assert data == []
+
+
+def test_export_bad_batch_defaults_to_zero(seeded_db, csrf_session):
+    resp = handle_export({"batch": ["abc"]})
+    assert resp["status"] == 200
+
+
+def test_export_has_content_disposition(seeded_db, csrf_session):
+    resp = handle_export({})
+    assert any(
+        k.lower() == "content-disposition" and "tinyweb-export.json" in v
+        for k, v in resp.get("headers", {}).items()
+    )
+
+
+def test_import_form_renders(temp_db, csrf_session):
+    resp = handle_import_form()
+    assert resp["status"] == 200
+    assert "import" in resp["body"].lower()
+    assert "textarea" in resp["body"]
+
+
+def test_import_submit_empty_rejected(temp_db, csrf_session):
+    resp = handle_import_submit({"data": [""]})
+    assert resp["status"] == 200
+    assert "paste json" in resp["body"].lower()
+
+
+def test_import_submit_invalid_json(temp_db, csrf_session):
+    resp = handle_import_submit({"data": ["not json"]})
+    assert resp["status"] == 200
+    assert "invalid json" in resp["body"].lower()
+
+
+def test_import_submit_non_array_rejected(temp_db, csrf_session):
+    resp = handle_import_submit({"data": ['"string"']})
+    assert resp["status"] == 200
+    assert "expected a json array" in resp["body"].lower()
+
+
+def test_import_submit_too_many_rejected(temp_db, csrf_session):
+    big = [{"url": "https://x.com/y"}] * 101
+    resp = handle_import_submit({"data": [json.dumps(big)]})
+    assert resp["status"] == 200
+    assert "too many" in resp["body"].lower()
+
+
+@patch("tinyweb.handlers.data.index_url")
+def test_import_submit_calls_index_url(mock_index, temp_db, csrf_session):
+    payload = [
+        {"url": "https://example.com/a", "note": "note a"},
+        {"url": "https://example.com/b", "note": "note b"},
+    ]
+    resp = handle_import_submit({"data": [json.dumps(payload)]})
+    assert resp["status"] == 200
+    assert "imported 2" in resp["body"].lower()
+    assert mock_index.call_count == 2
+    mock_index.assert_any_call("https://example.com/a", "note a")
+    mock_index.assert_any_call("https://example.com/b", "note b")
+
+
+@patch("tinyweb.handlers.data.index_url")
+def test_import_submit_skips_empty_url(mock_index, temp_db, csrf_session):
+    payload = [
+        {"url": "", "note": ""},
+        {"url": "https://example.com/real", "note": ""},
+    ]
+    resp = handle_import_submit({"data": [json.dumps(payload)]})
+    assert "imported 1" in resp["body"].lower()
+    assert mock_index.call_count == 1
+
+
+@patch("tinyweb.handlers.data.index_url")
+def test_import_submit_counts_errors(mock_index, temp_db, csrf_session):
+    mock_index.side_effect = [None, ValueError("boom")]
+    payload = [
+        {"url": "https://example.com/a", "note": ""},
+        {"url": "https://example.com/b", "note": ""},
+    ]
+    resp = handle_import_submit({"data": [json.dumps(payload)]})
+    assert "imported 1" in resp["body"].lower()
+    assert "1 error" in resp["body"].lower()
+
+
+def test_reindex_form_shows_disabled_message(temp_db, csrf_session):
+    resp = handle_reindex_form()
+    assert resp["status"] == 200
+    assert "disabled" in resp["body"].lower()
+    assert "settings" in resp["body"]
+
+
+def test_reindex_form_shows_progress_when_running(temp_db, csrf_session):
+    from tinyweb.db import set_setting
+    set_setting("semantic_search", "1")
+    set_setting("reindex_progress", "5/10")
+    resp = handle_reindex_form()
+    assert "reindex in progress" in resp["body"].lower()
+    assert "5/10" in resp["body"]
+
+
+def test_reindex_submit_redirects(temp_db, csrf_session):
+    resp = handle_reindex_submit({})
+    assert resp["status"] in (302, 303)