Skip to content

Commit 83e265d

Browse files
committed
fix tests
Signed-off-by: phernandez <paul@basicmachines.co>
1 parent 70bb10b commit 83e265d

2 files changed

Lines changed: 3 additions & 2 deletions

File tree

src/basic_memory/repository/relation_repository.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -140,7 +140,8 @@ async def add_all_ignore_duplicates(self, relations: List[Relation]) -> int:
140140
stmt = stmt.on_conflict_do_nothing()
141141

142142
result = await session.execute(stmt)
143-
return result.rowcount if result.rowcount else 0
143+
# rowcount can be -1 for ON CONFLICT DO NOTHING (undefined affected rows)
144+
return result.rowcount if result.rowcount > 0 else 0
144145

145146
def get_load_options(self) -> List[LoaderOption]:
146147
return [selectinload(Relation.from_entity), selectinload(Relation.to_entity)]

tests/services/test_initialization.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -182,7 +182,7 @@ async def test_initialize_file_sync_background_tasks(
182182

183183
@pytest.mark.asyncio
184184
@patch("basic_memory.services.initialization.db.get_or_create_db")
185-
@patch("basic_memory.cli.commands.sync.get_sync_service")
185+
@patch("basic_memory.sync.sync_service.get_sync_service")
186186
@patch("basic_memory.sync.WatchService")
187187
@patch("basic_memory.services.initialization.asyncio.create_task")
188188
@patch.dict("os.environ", {"BASIC_MEMORY_MCP_PROJECT": "project1"})

0 commit comments

Comments
 (0)