feat: add more tests for generator, patterns, recorder, and search

tests/test_search.py

@@ -0,0 +1,83 @@
+"""Tests for CLI Command Memory search functionality."""
+
+from datetime import datetime, timedelta
+from cli_memory.search import SearchEngine
+from cli_memory.models import Command, CommandType
+
+
+class MockCommand:
+    def __init__(self, id, command, command_type, timestamp=None, project_id=None,
+                 exit_code=0, duration_ms=None, working_directory="/tmp"):
+        self.id = id
+        self.command = command
+        self.command_type = command_type
+        self.timestamp = timestamp or datetime.utcnow()
+        self.project_id = project_id
+        self.exit_code = exit_code
+        self.duration_ms = duration_ms
+        self.working_directory = working_directory
+
+
+def test_search_engine_init():
+    """Test SearchEngine initialization."""
+    engine = SearchEngine()
+    assert engine is not None
+
+
+def test_matches_query_exact():
+    """Test exact query matching."""
+    engine = SearchEngine()
+    assert engine._matches_query("git status", "git status")
+    assert not engine._matches_query("git status", "docker ps")
+
+
+def test_matches_query_partial():
+    """Test partial query matching."""
+    engine = SearchEngine()
+    assert engine._matches_query("git status", "status")
+    assert engine._matches_query("git status", "git")
+
+
+def test_search_commands_by_type():
+    """Test search filtering by command type."""
+    commands = [
+        MockCommand(1, "git status", CommandType.GIT),
+        MockCommand(2, "git add .", CommandType.GIT),
+        MockCommand(3, "docker ps", CommandType.DOCKER),
+    ]
+
+    engine = SearchEngine()
+    results = engine.search_commands(commands, "", command_type="docker")
+    assert len(results) == 1
+
+
+def test_search_recent():
+    """Test recent command search."""
+    now = datetime.utcnow()
+    hour_ago = now - timedelta(hours=1)
+    day_ago = now - timedelta(days=1)
+
+    commands = [
+        MockCommand(1, "recent cmd", CommandType.OTHER, timestamp=hour_ago),
+        MockCommand(2, "old cmd", CommandType.OTHER, timestamp=day_ago),
+    ]
+
+    engine = SearchEngine()
+    results = engine.search_recent(commands, hours=24)
+    assert len(results) == 1
+
+
+def test_get_command_statistics():
+    """Test command statistics calculation."""
+    commands = [
+        MockCommand(1, "git status", CommandType.GIT),
+        MockCommand(2, "git add .", CommandType.GIT),
+        MockCommand(3, "docker ps", CommandType.DOCKER),
+    ]
+
+    engine = SearchEngine()
+    stats = engine.get_command_statistics(commands)
+
+    assert stats["total_commands"] == 3
+    assert "git" in stats["by_type"]
+    assert stats["by_type"]["git"] == 2