53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
from shell_history_search.core import SearchEngine, SearchResult
|
|
from shell_history_search.db import init_database
|
|
|
|
|
|
class TestSearchEngine:
|
|
def test_init_with_db_path(self, temp_db_path):
|
|
conn = init_database(temp_db_path)
|
|
engine = SearchEngine(db_path=conn)
|
|
stats = engine.get_stats()
|
|
|
|
assert stats["total_commands"] == 0
|
|
assert stats["total_embeddings"] == 0
|
|
conn.close()
|
|
|
|
def test_get_stats_empty(self, temp_db_path):
|
|
conn = init_database(temp_db_path)
|
|
engine = SearchEngine(db_path=conn)
|
|
stats = engine.get_stats()
|
|
|
|
assert stats["total_commands"] == 0
|
|
assert stats["total_embeddings"] == 0
|
|
assert stats["shell_counts"] == {}
|
|
assert stats["embedding_model"] == "all-MiniLM-L6-v2"
|
|
assert stats["embedding_dim"] == 384
|
|
conn.close()
|
|
|
|
def test_clear_all(self, temp_db_path):
|
|
conn = init_database(temp_db_path)
|
|
engine = SearchEngine(db_path=conn)
|
|
engine.clear_all()
|
|
|
|
stats = engine.get_stats()
|
|
assert stats["total_commands"] == 0
|
|
assert stats["total_embeddings"] == 0
|
|
conn.close()
|
|
|
|
|
|
class TestSearchResult:
|
|
def test_search_result_creation(self):
|
|
result = SearchResult(
|
|
command="git commit",
|
|
shell_type="bash",
|
|
timestamp=1700000000,
|
|
similarity=0.95,
|
|
command_id=1,
|
|
)
|
|
|
|
assert result.command == "git commit"
|
|
assert result.shell_type == "bash"
|
|
assert result.timestamp == 1700000000
|
|
assert result.similarity == 0.95
|
|
assert result.command_id == 1
|