import pytest
from genesis.services.rag_service import RagService, CannedRagService


def test_canned_rag_returns_rules():
    svc = CannedRagService(samples_dir="samples")
    rules = svc.retrieve_write_rules("db_design")
    assert isinstance(rules, list)
    rules2 = svc.retrieve_design_rules("db_design")
    assert isinstance(rules2, list)


def test_rag_service_is_protocol():
    assert isinstance(CannedRagService("samples"), RagService)