wehub-resource-sync
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
"""Shared helpers for the Google ADK integration test suite.
|
|
|
|
Defines a ``trace_test(schema_name)`` decorator factory that resolves
|
|
schema files relative to ``schemas/`` next to this conftest, dispatching
|
|
to ``generate_trace_json`` (when ``GENERATE_SCHEMAS=true``) or
|
|
``assert_trace_json``. Mirrors the per-file definition in the AgentCore
|
|
suite, lifted into conftest so the four test modules don't duplicate
|
|
the same five lines.
|
|
|
|
The Google ADK test suite is split into:
|
|
- ``test_span_interceptor.py`` — synthetic OTel-span unit tests for
|
|
``OpenInferenceSpanInterceptor`` (no live ADK / Gemini calls).
|
|
- ``test_sync.py`` / ``test_async.py`` — end-to-end traces via real
|
|
ADK agents, schema-asserted. Skipped without ``GOOGLE_API_KEY``.
|
|
- ``test_evaluate_agent.py`` — component-level evals through
|
|
``dataset.evals_iterator``. Skipped without ``GOOGLE_API_KEY``
|
|
+ ``OPENAI_API_KEY`` (the metric scorer).
|
|
|
|
Skip markers live on the integration test modules themselves
|
|
(``pytestmark = pytest.mark.skipif(...)``) — defining them here would
|
|
also skip the synthetic interceptor tests, which don't need any keys.
|
|
"""
|
|
|
|
import os
|
|
|
|
from tests.test_integrations.utils import (
|
|
assert_trace_json,
|
|
generate_trace_json,
|
|
is_generate_mode,
|
|
)
|
|
|
|
|
|
_current_dir = os.path.dirname(os.path.abspath(__file__))
|
|
_schemas_dir = os.path.join(_current_dir, "schemas")
|
|
|
|
|
|
def trace_test(schema_name: str):
|
|
"""Resolve to ``generate_trace_json`` or ``assert_trace_json``."""
|
|
schema_path = os.path.join(_schemas_dir, schema_name)
|
|
if is_generate_mode():
|
|
return generate_trace_json(schema_path)
|
|
else:
|
|
return assert_trace_json(schema_path)
|