mirror of
https://github.com/open-jarvis/OpenJarvis.git
synced 2026-07-28 14:07:55 +00:00
krypticmouseandClaude Sonnet 4.6
138fb8a3d0
Wire up ObsidianConnector and GmailConnector via auto-import in the connectors __init__.py so they register on package import. Add a full end-to-end integration test covering the Obsidian vault → SyncEngine → KnowledgeStore → knowledge_search pipeline. Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
76 lines
2.7 KiB
Python
76 lines
2.7 KiB
Python
"""Integration test: Obsidian vault → SyncEngine → KnowledgeStore → knowledge_search."""
|
|
|
|
from __future__ import annotations
|
|
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from openjarvis.connectors.obsidian import ObsidianConnector
|
|
from openjarvis.connectors.pipeline import IngestionPipeline
|
|
from openjarvis.connectors.store import KnowledgeStore
|
|
from openjarvis.connectors.sync_engine import SyncEngine
|
|
from openjarvis.tools.knowledge_search import KnowledgeSearchTool
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Fixtures
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
@pytest.fixture
|
|
def vault(tmp_path: Path) -> Path:
|
|
vault_dir = tmp_path / "test-vault"
|
|
vault_dir.mkdir()
|
|
(vault_dir / "project-notes.md").write_text(
|
|
"# Project Alpha\n\nWe decided to migrate to Kubernetes in March.\n\n"
|
|
"## Cost Analysis\n\n"
|
|
"Estimated 40% increase in cloud spend during transition.\n\n"
|
|
"## Timeline\n\nSix-week migration window starting April 1st."
|
|
)
|
|
(vault_dir / "meeting.md").write_text(
|
|
"# Sprint Review\n\nDiscussed budget concerns with Mike and Sarah.\n"
|
|
"Action item: Sarah to prepare cost comparison document."
|
|
)
|
|
return vault_dir
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# Tests
|
|
# ---------------------------------------------------------------------------
|
|
|
|
|
|
def test_full_pipeline_obsidian_to_search(vault: Path, tmp_path: Path) -> None:
|
|
"""End-to-end: Obsidian vault → SyncEngine → KnowledgeStore → knowledge_search."""
|
|
# 1. Setup
|
|
store = KnowledgeStore(db_path=str(tmp_path / "integration.db"))
|
|
pipeline = IngestionPipeline(store=store)
|
|
engine = SyncEngine(pipeline=pipeline, state_db=str(tmp_path / "state.db"))
|
|
connector = ObsidianConnector(vault_path=str(vault))
|
|
|
|
# 2. Sync
|
|
engine.sync(connector)
|
|
|
|
# 3. Verify checkpoint
|
|
cp = engine.get_checkpoint("obsidian")
|
|
assert cp is not None
|
|
# SyncEngine checkpoint records total chunks ingested (not document count).
|
|
# The two vault files produce 6 chunks via SemanticChunker (note strategy).
|
|
assert cp["items_synced"] == 6 # 6 chunks from 2 md files
|
|
|
|
# 4. Search via knowledge_search tool
|
|
tool = KnowledgeSearchTool(store=store)
|
|
|
|
result = tool.execute(query="Kubernetes migration")
|
|
assert result.success
|
|
assert "Kubernetes" in result.content
|
|
|
|
result = tool.execute(query="budget concerns")
|
|
assert result.success
|
|
assert "budget" in result.content.lower()
|
|
|
|
result = tool.execute(query="migration", source="obsidian")
|
|
assert result.success
|
|
|
|
result = tool.execute(query="migration", source="gmail")
|
|
assert "No relevant results" in result.content
|