Timmy-time-dashboard/tests/functional/test_setup_prod.py

import os
import shutil
import subprocess
import time
from pathlib import Path

import pytest

# Production-like paths for functional testing
PROD_PROJECT_DIR = Path("/home/ubuntu/prod-sovereign-stack")
PROD_VAULT_DIR = PROD_PROJECT_DIR / "TimmyVault"
SETUP_SCRIPT_PATH = Path("/home/ubuntu/setup_timmy.sh")

pytestmark = pytest.mark.skipif(
    not SETUP_SCRIPT_PATH.exists(),
    reason=f"Setup script not found at {SETUP_SCRIPT_PATH}",
)


@pytest.fixture(scope="module", autouse=True)
def setup_prod_env():
    """Ensure a clean environment and run the full installation."""
    if PROD_PROJECT_DIR.exists():
        shutil.rmtree(PROD_PROJECT_DIR)

    # Run the actual install command
    env = os.environ.copy()
    env["PROJECT_DIR"] = str(PROD_PROJECT_DIR)
    env["VAULT_DIR"] = str(PROD_VAULT_DIR)

    result = subprocess.run(
        [str(SETUP_SCRIPT_PATH), "install"], capture_output=True, text=True, env=env
    )

    assert result.returncode == 0, f"Install failed: {result.stderr}"
    yield
    # Cleanup after all tests in module
    # shutil.rmtree(PROD_PROJECT_DIR)


def test_prod_directory_structure():
    """Verify the directory structure matches production expectations."""
    assert PROD_PROJECT_DIR.exists()
    assert (PROD_PROJECT_DIR / "paperclip").exists()
    assert (PROD_PROJECT_DIR / "agents/hello-timmy").exists()
    assert PROD_VAULT_DIR.exists()
    assert (PROD_PROJECT_DIR / "logs").exists()
    assert (PROD_PROJECT_DIR / "pids").exists()


def test_prod_paperclip_dependencies():
    """Verify that Paperclip dependencies were actually installed (node_modules exists)."""
    node_modules = PROD_PROJECT_DIR / "paperclip/node_modules"
    assert node_modules.exists(), "Paperclip node_modules should exist after installation"
    # Check for a common package to ensure it's not just an empty dir
    assert (
        (node_modules / "typescript").exists()
        or (node_modules / "vite").exists()
        or (node_modules / "next").exists()
        or any(node_modules.iterdir())
    )


def test_prod_openfang_config():
    """Verify OpenFang agent configuration."""
    agent_toml = PROD_PROJECT_DIR / "agents/hello-timmy/agent.toml"
    assert agent_toml.exists()
    with open(agent_toml) as f:
        content = f.read()
        assert 'name = "hello-timmy"' in content
        assert 'model = "default"' in content


def test_prod_obsidian_vault_content():
    """Verify the initial content of the Obsidian vault."""
    hello_note = PROD_VAULT_DIR / "Hello World.md"
    soul_note = PROD_VAULT_DIR / "SOUL.md"

    assert hello_note.exists()
    assert soul_note.exists()

    with open(hello_note) as f:
        content = f.read()
        assert "# Hello World" in content
        assert "Paperclip" in content
        assert "OpenFang" in content

    with open(soul_note) as f:
        content = f.read()
        assert "I am Timmy" in content
        assert "sovereign AI agent" in content


def test_prod_service_lifecycle():
    """Verify that services can be started, checked, and stopped."""
    env = os.environ.copy()
    env["PROJECT_DIR"] = str(PROD_PROJECT_DIR)
    env["VAULT_DIR"] = str(PROD_VAULT_DIR)

    # Start services
    start_result = subprocess.run(
        [str(SETUP_SCRIPT_PATH), "start"], capture_output=True, text=True, env=env
    )
    assert start_result.returncode == 0

    # Wait a moment for processes to initialize
    time.sleep(2)

    # Check status
    status_result = subprocess.run(
        [str(SETUP_SCRIPT_PATH), "status"], capture_output=True, text=True, env=env
    )
    assert "running" in status_result.stdout

    # Stop services
    stop_result = subprocess.run(
        [str(SETUP_SCRIPT_PATH), "stop"], capture_output=True, text=True, env=env
    )
    assert stop_result.returncode == 0

    # Final status check
    final_status = subprocess.run(
        [str(SETUP_SCRIPT_PATH), "status"], capture_output=True, text=True, env=env
    )
    assert "stopped" in final_status.stdout