flowent 0.0.6 → 0.0.10

package/backend/tests/test_logging.py

@@ -0,0 +1,182 @@
+import logging
+import sys
+
+from flowent.logging import (
+    LITELLM_LOGGER_NAMES,
+    TRACE_LEVEL,
+    configure_litellm_logging,
+    configure_logging,
+    ensure_logging_configured,
+    redact_log_value,
+)
+
+
+def test_logging_creates_run_file_under_data_logs(tmp_path, monkeypatch) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+
+    log_file = configure_logging()
+
+    assert log_file.parent == tmp_path / "logs"
+    assert log_file.name.startswith("flowent-")
+    assert log_file.suffix == ".log"
+    assert log_file.is_file()
+
+
+def test_file_logging_accepts_trace_and_console_uses_mode_levels(
+    tmp_path, monkeypatch
+) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+    monkeypatch.setenv("DEBUG", "False")
+
+    log_file = configure_logging()
+    handlers = logging.getLogger().handlers
+    file_handler = next(
+        handler for handler in handlers if isinstance(handler, logging.FileHandler)
+    )
+    console_handler = next(
+        handler for handler in handlers if not isinstance(handler, logging.FileHandler)
+    )
+
+    assert file_handler.level == TRACE_LEVEL
+    assert console_handler.level == logging.INFO
+
+    logging.getLogger("flowent.test").log(TRACE_LEVEL, "trace-only detail")
+    for handler in logging.getLogger().handlers:
+        handler.flush()
+
+    assert "trace-only detail" in log_file.read_text()
+
+    monkeypatch.setenv("DEBUG", "True")
+    configure_logging()
+    console_handler = next(
+        handler
+        for handler in logging.getLogger().handlers
+        if not isinstance(handler, logging.FileHandler)
+    )
+
+    assert console_handler.level == logging.DEBUG
+
+
+def test_logging_prunes_old_run_logs(tmp_path, monkeypatch) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+    logs = tmp_path / "logs"
+    logs.mkdir(parents=True)
+    for index in range(6):
+        (logs / f"flowent-20260101-00000{index}-1.log").write_text(str(index))
+
+    configure_logging()
+
+    files = sorted(log.name for log in logs.glob("flowent-*.log"))
+    assert len(files) == 5
+    assert "flowent-20260101-000000-1.log" not in files
+
+
+def test_logging_path_follows_flowent_data_dir(tmp_path, monkeypatch) -> None:
+    data_dir = tmp_path / "custom-flowent"
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(data_dir))
+
+    log_file = configure_logging()
+
+    assert log_file.parent == data_dir / "logs"
+
+
+def test_logging_redacts_full_api_key_but_keeps_context(tmp_path, monkeypatch) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+    log_file = configure_logging()
+
+    logging.getLogger("flowent.test").log(
+        TRACE_LEVEL,
+        "provider=OpenAI model=gpt-5 output=hello api_key=sk-secret-value",
+    )
+    for handler in logging.getLogger().handlers:
+        handler.flush()
+
+    rendered = log_file.read_text()
+
+    assert "provider=OpenAI" in rendered
+    assert "model=gpt-5" in rendered
+    assert "output=hello" in rendered
+    assert "sk-secret-value" not in rendered
+    assert "api_key=[REDACTED]" in rendered
+    assert redact_log_value("authorization=Bearer sk-secret-value") == (
+        "authorization=[REDACTED]"
+    )
+
+
+def test_direct_main_app_import_creates_data_log_file(tmp_path, monkeypatch) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+    sys.modules.pop("flowent.main", None)
+
+    try:
+        __import__("flowent.main")
+    finally:
+        sys.modules.pop("flowent.main", None)
+
+    files = sorted((tmp_path / "logs").glob("flowent-*.log"))
+    assert len(files) == 1
+
+
+def test_create_app_creates_data_log_file(tmp_path, monkeypatch) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+
+    from flowent.main import create_app
+
+    create_app(serve_frontend=False)
+
+    files = sorted((tmp_path / "logs").glob("flowent-*.log"))
+    assert len(files) == 1
+
+
+def test_create_app_reuses_logging_handlers(tmp_path, monkeypatch) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+
+    from flowent.main import create_app
+
+    create_app(serve_frontend=False)
+    create_app(serve_frontend=False)
+
+    handlers = logging.getLogger().handlers
+    file_handlers = [
+        handler for handler in handlers if isinstance(handler, logging.FileHandler)
+    ]
+    console_handlers = [
+        handler for handler in handlers if not isinstance(handler, logging.FileHandler)
+    ]
+    files = sorted((tmp_path / "logs").glob("flowent-*.log"))
+
+    assert len(file_handlers) == 1
+    assert len(console_handlers) == 1
+    assert len(files) == 1
+
+
+def test_any_logging_path_follows_flowent_data_dir(tmp_path, monkeypatch) -> None:
+    data_dir = tmp_path / "custom-flowent"
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(data_dir))
+
+    log_file = ensure_logging_configured()
+
+    assert log_file.parent == data_dir / "logs"
+
+
+def test_litellm_debug_logs_use_flowent_handlers(tmp_path, monkeypatch, capsys) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+    monkeypatch.delenv("LITELLM_LOG", raising=False)
+
+    log_file = configure_logging()
+    import litellm  # noqa: F401
+
+    configure_litellm_logging()
+    logging.getLogger("LiteLLM").debug("stream chunk detail")
+    for handler in logging.getLogger().handlers:
+        handler.flush()
+
+    captured = capsys.readouterr()
+    litellm_handlers = [
+        handler
+        for logger_name in LITELLM_LOGGER_NAMES
+        for handler in logging.getLogger(logger_name).handlers
+    ]
+
+    assert litellm_handlers == []
+    assert "stream chunk detail" not in captured.err
+    assert "stream chunk detail" in log_file.read_text()

package/backend/tests/test_persistence.py

@@ -0,0 +1,125 @@
+from fastapi.testclient import TestClient
+
+from flowent.main import create_app
+
+
+def test_app_state_persists_providers_across_app_instances(
+    tmp_path, monkeypatch
+) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+    client = TestClient(create_app(serve_frontend=False))
+
+    response = client.post(
+        "/api/providers",
+        json={
+            "api_key": "sk-local",
+            "base_url": "https://api.example.test/v1",
+            "id": "provider-openai",
+            "models": ["gpt-5.1", "gpt-5.1-mini"],
+            "name": "OpenAI",
+            "type": "openai",
+        },
+    )
+
+    assert response.status_code == 200
+
+    restarted_client = TestClient(create_app(serve_frontend=False))
+    state_response = restarted_client.get("/api/state")
+
+    assert state_response.status_code == 200
+    assert state_response.json()["providers"] == [
+        {
+            "api_key": "sk-local",
+            "base_url": "https://api.example.test/v1",
+            "id": "provider-openai",
+            "models": ["gpt-5.1", "gpt-5.1-mini"],
+            "name": "OpenAI",
+            "type": "openai",
+        }
+    ]
+
+
+def test_app_state_persists_settings_and_workspace_messages(
+    tmp_path, monkeypatch
+) -> None:
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(tmp_path))
+    client = TestClient(create_app(serve_frontend=False))
+    client.post(
+        "/api/providers",
+        json={
+            "api_key": "",
+            "base_url": "",
+            "id": "provider-anthropic",
+            "models": ["claude-sonnet-4-5"],
+            "name": "Anthropic",
+            "type": "anthropic",
+        },
+    )
+
+    settings_response = client.put(
+        "/api/settings",
+        json={
+            "selected_model": "claude-sonnet-4-5",
+            "selected_provider_id": "provider-anthropic",
+        },
+    )
+    messages_response = client.put(
+        "/api/workspace/messages",
+        json={
+            "messages": [
+                {
+                    "author": "assistant",
+                    "content": "Draft a launch checklist",
+                    "id": "message-1",
+                    "tools": [
+                        {
+                            "id": "tool-1",
+                            "name": "read_file",
+                            "status": "success",
+                            "title": "Read notes.txt",
+                        }
+                    ],
+                }
+            ]
+        },
+    )
+
+    assert settings_response.status_code == 200
+    assert messages_response.status_code == 200
+
+    restarted_client = TestClient(create_app(serve_frontend=False))
+    state = restarted_client.get("/api/state").json()
+
+    assert state["settings"] == {
+        "selected_model": "claude-sonnet-4-5",
+        "selected_provider_id": "provider-anthropic",
+    }
+    assert state["messages"] == [
+        {
+            "author": "assistant",
+            "content": "Draft a launch checklist",
+            "id": "message-1",
+            "tools": [
+                {
+                    "arguments": None,
+                    "content": None,
+                    "data": None,
+                    "id": "tool-1",
+                    "name": "read_file",
+                    "status": "success",
+                    "title": "Read notes.txt",
+                }
+            ],
+        }
+    ]
+
+
+def test_data_directory_uses_flowent_data_dir(tmp_path, monkeypatch) -> None:
+    data_dir = tmp_path / "custom-flowent"
+    monkeypatch.setenv("FLOWENT_DATA_DIR", str(data_dir))
+
+    client = TestClient(create_app(serve_frontend=False))
+    response = client.get("/api/state")
+
+    assert response.status_code == 200
+    assert (data_dir / "flowent.db").is_file()