06d9313a7c
Introduces comprehensive async unit tests to verify input processing merges extra values correctly and output generation returns expected URLs and ports. Adds a basic test to confirm package exports remain intact. Cleans up the .gitignore by removing redundant test file ignores to allow test discovery. Improves code quality and test coverage for the LightRAG app integration. Relates to MLO-469
78 lines
2.6 KiB
Python
78 lines
2.6 KiB
Python
from __future__ import annotations
|
|
|
|
from typing import Any
|
|
|
|
import pytest
|
|
|
|
from apolo_app_types.app_types import AppType
|
|
from apolo_app_types.protocols.common import IngressHttp, Preset
|
|
from apolo_app_types.protocols.postgres import CrunchyPostgresUserCredentials
|
|
|
|
from apolo_apps_lightrag.inputs_processor import LightRAGInputsProcessor
|
|
from apolo_apps_lightrag.types import (
|
|
LightRAGAppInputs,
|
|
LightRAGPersistence,
|
|
OpenAIEmbeddingProvider,
|
|
OpenAILLMProvider,
|
|
)
|
|
|
|
|
|
@pytest.mark.asyncio
|
|
async def test_gen_extra_values_merges_sources(monkeypatch: pytest.MonkeyPatch) -> None:
|
|
client_stub = object()
|
|
processor = LightRAGInputsProcessor(client_stub) # type: ignore[arg-type]
|
|
|
|
captured: dict[str, Any] = {}
|
|
|
|
async def fake_gen_extra_values(**kwargs: Any) -> dict[str, Any]:
|
|
captured.update(kwargs)
|
|
return {"platform": {"ingress": True}}
|
|
|
|
monkeypatch.setattr(
|
|
"apolo_apps_lightrag.inputs_processor.gen_extra_values",
|
|
fake_gen_extra_values,
|
|
)
|
|
|
|
app_inputs = LightRAGAppInputs(
|
|
preset=Preset(name="medium"),
|
|
ingress_http=IngressHttp(),
|
|
pgvector_user=CrunchyPostgresUserCredentials(
|
|
user="rag",
|
|
password="secret",
|
|
host="postgres.internal",
|
|
port=5432,
|
|
pgbouncer_host="pgbouncer.internal",
|
|
pgbouncer_port=6432,
|
|
dbname="lightrag",
|
|
),
|
|
persistence=LightRAGPersistence(
|
|
rag_storage_size=20,
|
|
inputs_storage_size=15,
|
|
),
|
|
llm_config=OpenAILLMProvider(api_key="llm-key", model="gpt-4o-mini"),
|
|
embedding_config=OpenAIEmbeddingProvider(api_key="embed-key"),
|
|
)
|
|
|
|
values = await processor.gen_extra_values(
|
|
app_inputs,
|
|
app_name="lightrag-app",
|
|
namespace="apps",
|
|
app_id="instance-123",
|
|
app_secrets_name="lightrag-secrets",
|
|
)
|
|
|
|
assert values["replicaCount"] == 1
|
|
assert values["fullnameOverride"] == "lightrag-app"
|
|
assert values["env"]["POSTGRES_HOST"] == "pgbouncer.internal"
|
|
assert values["env"]["LLM_BINDING"] == "openai"
|
|
assert values["persistence"]["ragStorage"]["size"] == "20Gi"
|
|
assert values["persistence"]["inputs"]["size"] == "15Gi"
|
|
assert values["platform"] == {"ingress": True}
|
|
|
|
assert captured["apolo_client"] is client_stub
|
|
assert captured["preset_type"] == app_inputs.preset
|
|
assert captured["ingress_http"] == app_inputs.ingress_http
|
|
assert captured["ingress_grpc"] is None
|
|
assert captured["namespace"] == "apps"
|
|
assert captured["app_id"] == "instance-123"
|
|
assert captured["app_type"] == AppType.LightRAG
|