pytest-adk 0.0.1__tar.gz

pytest_adk-0.0.1/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 nikkie
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

pytest_adk-0.0.1/PKG-INFO

@@ -0,0 +1,42 @@
+Metadata-Version: 2.3
+Name: pytest-adk
+Version: 0.0.1
+Summary: Helpers for testing agents with Google's adk-python
+Author: ftnext
+Author-email: ftnext <takuyafjp+develop@gmail.com>
+License: MIT License
+         
+         Copyright (c) 2026 nikkie
+         
+         Permission is hereby granted, free of charge, to any person obtaining a copy
+         of this software and associated documentation files (the "Software"), to deal
+         in the Software without restriction, including without limitation the rights
+         to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+         copies of the Software, and to permit persons to whom the Software is
+         furnished to do so, subject to the following conditions:
+         
+         The above copyright notice and this permission notice shall be included in all
+         copies or substantial portions of the Software.
+         
+         THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+         IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+         FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+         AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+         LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+         OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+         SOFTWARE.
+Classifier: Development Status :: 1 - Planning
+Classifier: Intended Audience :: Developers
+Classifier: Programming Language :: Python
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Requires-Dist: google-adk>=1.30.0
+Requires-Dist: pytest>=8 ; extra == 'dev'
+Requires-Dist: pytest-asyncio>=0.24 ; extra == 'dev'
+Requires-Python: >=3.10
+Project-URL: Repository, https://github.com/ftnext/pytest-adk
+Provides-Extra: dev

pytest_adk-0.0.1/pyproject.toml

@@ -0,0 +1,41 @@
+[project]
+name = "pytest-adk"
+version = "0.0.1"
+description = "Helpers for testing agents with Google's adk-python"
+authors = [
+    { name = "ftnext", email = "takuyafjp+develop@gmail.com" }
+]
+license = { file = "LICENSE" }
+requires-python = ">=3.10"
+classifiers = [
+    "Development Status :: 1 - Planning",
+    "Intended Audience :: Developers",
+    "Programming Language :: Python",
+    "Programming Language :: Python :: 3 :: Only",
+    "Programming Language :: Python :: 3.10",
+    "Programming Language :: Python :: 3.11",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Programming Language :: Python :: 3.14",
+]
+dependencies = [
+    "google-adk>=1.30.0",
+]
+
+[project.optional-dependencies]
+dev = [
+    "pytest>=8",
+    "pytest-asyncio>=0.24",
+]
+
+[tool.pytest]
+testpaths = ["tests"]
+asyncio_mode = "auto"
+asyncio_default_fixture_loop_scope = "function"
+
+[build-system]
+requires = ["uv_build>=0.11.2,<0.12.0"]
+build-backend = "uv_build"
+
+[project.urls]
+Repository = "https://github.com/ftnext/pytest-adk"

pytest_adk-0.0.1/src/pytest_adk/__init__.py

@@ -0,0 +1,9 @@
+# Copyright 2026 pytest-adk contributors
+
+from .resume import load_session_from_json
+from .resume import runner_from_exported_session
+
+__all__ = [
+    'load_session_from_json',
+    'runner_from_exported_session',
+]

pytest_adk-0.0.1/src/pytest_adk/resume.py

@@ -0,0 +1,126 @@
+# Copyright 2026 pytest-adk contributors
+"""Helpers to resume an exported ADK session JSON with an in-memory Runner."""
+
+from __future__ import annotations
+
+from pathlib import Path
+from typing import Optional
+
+from google.adk.agents.base_agent import BaseAgent
+from google.adk.artifacts.base_artifact_service import BaseArtifactService
+from google.adk.artifacts.in_memory_artifact_service import InMemoryArtifactService
+from google.adk.auth.credential_service.base_credential_service import BaseCredentialService
+from google.adk.auth.credential_service.in_memory_credential_service import InMemoryCredentialService
+from google.adk.memory.base_memory_service import BaseMemoryService
+from google.adk.memory.in_memory_memory_service import InMemoryMemoryService
+from google.adk.runners import Runner
+from google.adk.sessions.in_memory_session_service import InMemorySessionService
+from google.adk.sessions.session import Session
+
+
+def load_session_from_json(path_or_str: str | Path) -> Session:
+  """Load an exported session from a file path or raw JSON string.
+
+  Args:
+      path_or_str: Path to a ``.session.json``-style file, or the JSON text.
+          A ``str`` whose (stripped) first character is ``{`` or ``[`` is
+          treated as JSON content; otherwise it is treated as a file path and
+          must exist.
+
+  Returns:
+      Parsed :class:`google.adk.sessions.session.Session`.
+
+  Raises:
+      FileNotFoundError: If ``path_or_str`` refers to a path that does not
+          exist on disk.
+  """
+  if isinstance(path_or_str, Path):
+    if not path_or_str.is_file():
+      raise FileNotFoundError(path_or_str)
+    data = path_or_str.read_text(encoding='utf-8')
+  else:
+    stripped = path_or_str.lstrip()
+    # Avoid Path(...).is_file() on large JSON strings (long path errors).
+    if stripped.startswith('{') or stripped.startswith('['):
+      data = path_or_str
+    else:
+      candidate = Path(path_or_str)
+      try:
+        is_file = candidate.is_file()
+      except OSError as exc:
+        raise FileNotFoundError(path_or_str) from exc
+      if not is_file:
+        raise FileNotFoundError(path_or_str)
+      data = candidate.read_text(encoding='utf-8')
+
+  return Session.model_validate_json(data)
+
+
+async def runner_from_exported_session(
+    agent: BaseAgent,
+    exported: Session | str | Path,
+    *,
+    app_name: Optional[str] = None,
+    user_id: Optional[str] = None,
+    session_id: Optional[str] = None,
+    artifact_service: Optional[BaseArtifactService] = None,
+    memory_service: Optional[BaseMemoryService] = None,
+    credential_service: Optional[BaseCredentialService] = None,
+) -> tuple[Runner, Session]:
+  """Build an in-memory :class:`Runner` whose session service holds the export.
+
+  This mirrors the ``saved_session_file`` flow in ``google.adk.cli``: create a
+  fresh session with copied ``state``, then :meth:`append_event` for each export
+  event. Call :meth:`Runner.run_async` with the same ``user_id`` and
+  ``session.id`` to continue the conversation.
+
+  Args:
+      agent: Root agent for the runner (must match the exported app logically).
+      exported: Parsed session, or a path / JSON string for
+          :func:`load_session_from_json`.
+      app_name: Override runner app name (default: from export).
+      user_id: Override user id (default: from export).
+      session_id: Override session id (default: from export; keeps ids aligned).
+      artifact_service: Optional; defaults to :class:`InMemoryArtifactService`.
+      memory_service: Optional; defaults to :class:`InMemoryMemoryService`.
+      credential_service: Optional; defaults to
+          :class:`InMemoryCredentialService`.
+
+  Returns:
+      ``(runner, session)`` where ``session`` is the restored in-memory session
+      (also reachable via ``runner.session_service.get_session``).
+  """
+  if not isinstance(exported, Session):
+    exported = load_session_from_json(exported)
+
+  resolved_app_name = app_name or exported.app_name
+  resolved_user_id = user_id or exported.user_id
+  resolved_session_id = session_id or exported.id
+
+  session_service = InMemorySessionService()
+  session = await session_service.create_session(
+      app_name=resolved_app_name,
+      user_id=resolved_user_id,
+      state=exported.state,
+      session_id=resolved_session_id,
+  )
+
+  for event in exported.events:
+    await session_service.append_event(session, event)
+
+  runner = Runner(
+      app_name=resolved_app_name,
+      agent=agent,
+      session_service=session_service,
+      artifact_service=artifact_service
+      if artifact_service is not None
+      else InMemoryArtifactService(),
+      memory_service=memory_service
+      if memory_service is not None
+      else InMemoryMemoryService(),
+      credential_service=credential_service
+      if credential_service is not None
+      else InMemoryCredentialService(),
+  )
+
+  return runner, session