88 lines
3.2 KiB
Python
88 lines
3.2 KiB
Python
import os
|
|
import subprocess
|
|
import shutil
|
|
import pytest
|
|
from pathlib import Path
|
|
|
|
# Constants for testing
|
|
TEST_PROJECT_DIR = Path("/home/ubuntu/test-sovereign-stack")
|
|
TEST_VAULT_DIR = TEST_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 cleanup_test_env():
|
|
"""Ensure a clean environment before and after tests."""
|
|
if TEST_PROJECT_DIR.exists():
|
|
shutil.rmtree(TEST_PROJECT_DIR)
|
|
yield
|
|
# We keep the test env for manual inspection if needed, or cleanup
|
|
# shutil.rmtree(TEST_PROJECT_DIR)
|
|
|
|
def run_setup_command(args):
|
|
"""Helper to run the setup script with arguments."""
|
|
result = subprocess.run(
|
|
[str(SETUP_SCRIPT_PATH)] + args,
|
|
capture_output=True,
|
|
text=True,
|
|
cwd="/home/ubuntu"
|
|
)
|
|
return result
|
|
|
|
def test_setup_install_creates_directories():
|
|
"""Test that './setup_timmy.sh install' creates the expected directory structure."""
|
|
# Note: We expect the script to be present at SETUP_SCRIPT_PATH
|
|
assert SETUP_SCRIPT_PATH.exists(), "Setup script must exist before testing"
|
|
|
|
result = run_setup_command(["install"])
|
|
|
|
# Check if command succeeded
|
|
assert result.returncode == 0, f"Setup install failed: {result.stderr}"
|
|
|
|
# Check directory structure
|
|
assert TEST_PROJECT_DIR.exists()
|
|
assert (TEST_PROJECT_DIR / "paperclip").exists()
|
|
assert (TEST_PROJECT_DIR / "agents/hello-timmy").exists()
|
|
assert TEST_VAULT_DIR.exists()
|
|
assert (TEST_PROJECT_DIR / "logs").exists()
|
|
|
|
def test_setup_install_creates_files():
|
|
"""Test that './setup_timmy.sh install' creates the expected configuration and notes."""
|
|
# Check Agent config
|
|
agent_toml = TEST_PROJECT_DIR / "agents/hello-timmy/agent.toml"
|
|
assert agent_toml.exists()
|
|
with open(agent_toml, "r") as f:
|
|
content = f.read()
|
|
assert 'name = "hello-timmy"' in content
|
|
|
|
# Check Obsidian notes
|
|
hello_note = TEST_VAULT_DIR / "Hello World.md"
|
|
soul_note = TEST_VAULT_DIR / "SOUL.md"
|
|
assert hello_note.exists()
|
|
assert soul_note.exists()
|
|
|
|
with open(soul_note, "r") as f:
|
|
content = f.read()
|
|
assert "I am Timmy" in content
|
|
|
|
def test_setup_install_dependencies():
|
|
"""Test that dependencies are correctly handled (OpenFang, Paperclip deps)."""
|
|
# Check if Paperclip node_modules exists (implies pnpm install ran)
|
|
# Note: In a real TDD we might mock pnpm, but here we want to verify the actual setup
|
|
node_modules = TEST_PROJECT_DIR / "paperclip/node_modules"
|
|
assert node_modules.exists()
|
|
|
|
def test_setup_start_stop_logic():
|
|
"""Test the start/stop command logic (simulated)."""
|
|
# This is harder to test fully without actually running the services,
|
|
# but we can check if the script handles the commands without crashing.
|
|
|
|
# Mocking start (it might fail if ports are taken, so we check return code)
|
|
# For the sake of this test, we just check if the script recognizes the command
|
|
result = run_setup_command(["status"])
|
|
assert "Status" in result.stdout or result.returncode == 0
|