mirror of
https://github.com/basicmachines-co/basic-memory
synced 2026-06-21 13:47:35 +00:00
bm-claudeai
020957cd76
Signed-off-by: phernandez <paul@basicmachines.co> Co-authored-by: phernandez <paul@basicmachines.co> Co-authored-by: Claude <noreply@anthropic.com>
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
"""Tests for the ProjectInfoRepository."""
|
|
|
|
import pytest
|
|
from sqlalchemy import text
|
|
|
|
from basic_memory.repository.project_info_repository import ProjectInfoRepository
|
|
from basic_memory.models.project import Project # Add a model reference
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_project_info_repository_init(session_maker):
|
|
"""Test ProjectInfoRepository initialization."""
|
|
# Create a ProjectInfoRepository
|
|
repository = ProjectInfoRepository(session_maker)
|
|
|
|
# Verify it was initialized properly
|
|
assert repository is not None
|
|
assert repository.session_maker == session_maker
|
|
# Model is set to a dummy value (Project is used as a reference here)
|
|
assert repository.Model is Project
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_project_info_repository_execute_query(session_maker):
|
|
"""Test ProjectInfoRepository execute_query method."""
|
|
# Create a ProjectInfoRepository
|
|
repository = ProjectInfoRepository(session_maker)
|
|
|
|
# Execute a simple query
|
|
result = await repository.execute_query(text("SELECT 1 as test"))
|
|
|
|
# Verify the result
|
|
assert result is not None
|
|
row = result.fetchone()
|
|
assert row is not None
|
|
assert row[0] == 1
|