ilia/nanobot
1
0
Fork 0
Code Issues Pull Requests 1 Actions Packages Projects Releases Wiki Activity

tests: update test_commands.py

Browse Source
This commit is contained in:
Re-bin 2026-02-14 01:02:58 +00:00
parent 12540ba8cb
commit 3b580fd6c8
1 changed files with 29 additions and 71 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
tests/test_commands.py
View File

@ -1,7 +1,6 @@
import os
import shutil
from pathlib import Path
from unittest.mock import MagicMock, patch
from unittest.mock import patch
import pytest
from typer.testing import CliRunner
@ -13,122 +12,81 @@ runner = CliRunner()
@pytest.fixture
def mock_paths():
"""Mock configuration and workspace paths for isolation."""
with patch("nanobot.config.loader.get_config_path") as mock_config_path, \
patch("nanobot.config.loader.save_config") as mock_save_config, \
patch("nanobot.utils.helpers.get_workspace_path") as mock_ws_path:
# Create temporary paths
"""Mock config/workspace paths for test isolation."""
with patch("nanobot.config.loader.get_config_path") as mock_cp, \
patch("nanobot.config.loader.save_config") as mock_sc, \
patch("nanobot.config.loader.load_config") as mock_lc, \
patch("nanobot.utils.helpers.get_workspace_path") as mock_ws:
base_dir = Path("./test_onboard_data")
if base_dir.exists():
shutil.rmtree(base_dir)
base_dir.mkdir()
config_file = base_dir / "config.json"
workspace_dir = base_dir / "workspace"
mock_config_path.return_value = config_file
mock_ws_path.return_value = workspace_dir
# We need save_config to actually write the file for existence checks to work
def side_effect_save_config(config):
with open(config_file, "w") as f:
f.write("{}")
mock_save_config.side_effect = side_effect_save_config
mock_cp.return_value = config_file
mock_ws.return_value = workspace_dir
mock_sc.side_effect = lambda config: config_file.write_text("{}")
yield config_file, workspace_dir
# Cleanup
if base_dir.exists():
shutil.rmtree(base_dir)
def test_onboard_fresh_install(mock_paths):
"""Test onboarding with no existing files."""
"""No existing config — should create from scratch."""
config_file, workspace_dir = mock_paths
result = runner.invoke(app, ["onboard"])
assert result.exit_code == 0
assert "Created config" in result.stdout
assert "Created workspace" in result.stdout
assert "nanobot is ready" in result.stdout
assert config_file.exists()
assert workspace_dir.exists()
assert (workspace_dir / "AGENTS.md").exists()
assert (workspace_dir / "memory" / "MEMORY.md").exists()
def test_onboard_existing_config_no_overwrite(mock_paths):
"""Test onboarding with existing config, user declines overwrite."""
def test_onboard_existing_config_refresh(mock_paths):
"""Config exists, user declines overwrite — should refresh (load-merge-save)."""
config_file, workspace_dir = mock_paths
# Pre-create config
config_file.write_text('{"existing": true}')
# Input "n" for overwrite prompt
result = runner.invoke(app, ["onboard"], input="n\n")
assert result.exit_code == 0
assert "Config already exists" in result.stdout
# Verify config was NOT changed
assert '{"existing": true}' in config_file.read_text()
# Verify workspace was still created
assert "Created workspace" in result.stdout
assert "existing values preserved" in result.stdout
assert workspace_dir.exists()
assert (workspace_dir / "AGENTS.md").exists()
def test_onboard_existing_config_overwrite(mock_paths):
"""Test onboarding with existing config, user checks overwrite."""
"""Config exists, user confirms overwrite — should reset to defaults."""
config_file, workspace_dir = mock_paths
# Pre-create config
config_file.write_text('{"existing": true}')
# Input "y" for overwrite prompt
result = runner.invoke(app, ["onboard"], input="y\n")
assert result.exit_code == 0
assert "Config already exists" in result.stdout
assert "Created config" in result.stdout
# Verify config WAS changed (our mock writes "{}")
test_content = config_file.read_text()
assert test_content == "{}" or test_content == ""
assert "Config reset to defaults" in result.stdout
assert workspace_dir.exists()
def test_onboard_existing_workspace_safe_create(mock_paths):
"""Test onboarding with existing workspace safely creates templates without prompting."""
"""Workspace exists — should not recreate, but still add missing templates."""
config_file, workspace_dir = mock_paths
# Pre-create workspace
workspace_dir.mkdir(parents=True)
# Scenario: Config exists (keep it), Workspace exists (add templates automatically)
config_file.write_text("{}")
inputs = "n\n" # No overwrite config
result = runner.invoke(app, ["onboard"], input=inputs)
result = runner.invoke(app, ["onboard"], input="n\n")
assert result.exit_code == 0
# Workspace exists message
# Depending on implementation, it might say "Workspace already exists" or just proceed.
# Code in commands.py Line 180: if not workspace.exists(): ...
# It does NOT print "Workspace already exists" if it exists.
# It only prints "Created workspace" if it created it.
assert "Created workspace" not in result.stdout
# Should NOT prompt for templates
assert "Create missing default templates?" not in result.stdout
# But SHOULD create them (since _create_workspace_templates is called unconditionally)
assert "Created AGENTS.md" in result.stdout
assert (workspace_dir / "AGENTS.md").exists()