mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
phernandez
117 lines
3.8 KiB
Python
117 lines
3.8 KiB
Python
"""Tests for the project_info CLI command."""
|
|
|
|
import json
|
|
from datetime import datetime
|
|
from unittest.mock import patch, AsyncMock
|
|
|
|
from typer.testing import CliRunner
|
|
|
|
from basic_memory.cli.main import app as cli_app
|
|
from basic_memory.schemas.project_info import (
|
|
ProjectInfoResponse,
|
|
ProjectStatistics,
|
|
ActivityMetrics,
|
|
SystemStatus,
|
|
)
|
|
|
|
|
|
def test_info_stats():
|
|
"""Test the 'project info' command with default output."""
|
|
runner = CliRunner()
|
|
|
|
# Create mock project info data
|
|
mock_info = ProjectInfoResponse(
|
|
project_name="test-project",
|
|
project_path="/test/path",
|
|
default_project="test-project",
|
|
statistics=ProjectStatistics(
|
|
total_entities=10,
|
|
total_observations=20,
|
|
total_relations=5,
|
|
total_unresolved_relations=1,
|
|
isolated_entities=2,
|
|
entity_types={"note": 8, "concept": 2},
|
|
observation_categories={"tech": 15, "note": 5},
|
|
relation_types={"connects_to": 3, "references": 2},
|
|
most_connected_entities=[],
|
|
),
|
|
activity=ActivityMetrics(recently_created=[], recently_updated=[], monthly_growth={}),
|
|
system=SystemStatus(
|
|
version="0.13.0",
|
|
database_path="/test/db.sqlite",
|
|
database_size="1.2 MB",
|
|
watch_status=None,
|
|
timestamp=datetime.now(),
|
|
),
|
|
available_projects={"test-project": {"path": "/test/path"}},
|
|
)
|
|
|
|
# Mock the async project_info function
|
|
with patch(
|
|
"basic_memory.cli.commands.project.project_info", new_callable=AsyncMock
|
|
) as mock_func:
|
|
mock_func.return_value = mock_info
|
|
|
|
# Run the command
|
|
result = runner.invoke(cli_app, ["project", "info"])
|
|
|
|
# Verify exit code
|
|
assert result.exit_code == 0
|
|
|
|
# Check that key data is included in the output
|
|
assert "Basic Memory Project Info" in result.stdout
|
|
assert "test-project" in result.stdout
|
|
assert "Statistics" in result.stdout
|
|
|
|
|
|
def test_info_stats_json():
|
|
"""Test the 'project info --json' command for JSON output."""
|
|
runner = CliRunner()
|
|
|
|
# Create mock project info data
|
|
mock_info = ProjectInfoResponse(
|
|
project_name="test-project",
|
|
project_path="/test/path",
|
|
default_project="test-project",
|
|
statistics=ProjectStatistics(
|
|
total_entities=10,
|
|
total_observations=20,
|
|
total_relations=5,
|
|
total_unresolved_relations=1,
|
|
isolated_entities=2,
|
|
entity_types={"note": 8, "concept": 2},
|
|
observation_categories={"tech": 15, "note": 5},
|
|
relation_types={"connects_to": 3, "references": 2},
|
|
most_connected_entities=[],
|
|
),
|
|
activity=ActivityMetrics(recently_created=[], recently_updated=[], monthly_growth={}),
|
|
system=SystemStatus(
|
|
version="0.13.0",
|
|
database_path="/test/db.sqlite",
|
|
database_size="1.2 MB",
|
|
watch_status=None,
|
|
timestamp=datetime.now(),
|
|
),
|
|
available_projects={"test-project": {"path": "/test/path"}},
|
|
)
|
|
|
|
# Mock the async project_info function
|
|
with patch(
|
|
"basic_memory.cli.commands.project.project_info", new_callable=AsyncMock
|
|
) as mock_func:
|
|
mock_func.return_value = mock_info
|
|
|
|
# Run the command with --json flag
|
|
result = runner.invoke(cli_app, ["project", "info", "--json"])
|
|
|
|
# Verify exit code
|
|
assert result.exit_code == 0
|
|
|
|
# Parse JSON output
|
|
output = json.loads(result.stdout)
|
|
|
|
# Verify JSON structure matches our mock data
|
|
assert output["default_project"] == "test-project"
|
|
assert output["project_name"] == "test-project"
|
|
assert output["statistics"]["total_entities"] == 10
|