testa-agent 0.1.0__tar.gz

testa_agent-0.1.0/PKG-INFO

@@ -0,0 +1,11 @@
+Metadata-Version: 2.4
+Name: testa-agent
+Version: 0.1.0
+Summary: Testa - AI-powered post-deploy testing
+Requires-Python: >=3.11
+Requires-Dist: httpx>=0.26.0
+Requires-Dist: playwright>=1.41.2
+Requires-Dist: pyyaml>=6.0.1
+Requires-Dist: click>=8.1.0
+Provides-Extra: vision
+Requires-Dist: anthropic>=0.40.0; extra == "vision"

testa_agent-0.1.0/pyproject.toml

@@ -0,0 +1,41 @@
+[build-system]
+requires = ["setuptools>=68.0", "wheel"]
+build-backend = "setuptools.build_meta"
+
+[project]
+name = "testa-agent"
+version = "0.1.0"
+description = "Testa - AI-powered post-deploy testing"
+requires-python = ">=3.11"
+dependencies = [
+    "httpx>=0.26.0",
+    "playwright>=1.41.2",
+    "pyyaml>=6.0.1",
+    "click>=8.1.0",
+]
+
+[project.optional-dependencies]
+vision = ["anthropic>=0.40.0"]
+
+[project.scripts]
+testa = "testa_agent.cli:main"
+
+[tool.setuptools.packages.find]
+include = ["testa_agent*"]
+
+[tool.mypy]
+python_version = "3.11"
+warn_return_any = false
+warn_unused_ignores = false
+disallow_untyped_defs = false
+ignore_missing_imports = true
+check_untyped_defs = true
+disable_error_code = ["import-untyped"]
+
+[[tool.mypy.overrides]]
+module = ["testa_agent.executor", "testa_agent.vision"]
+disable_error_code = ["union-attr", "return-value"]
+
+[tool.ruff]
+target-version = "py311"
+line-length = 100

testa_agent-0.1.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

testa_agent-0.1.0/testa_agent/__init__.py

@@ -0,0 +1,3 @@
+"""Testa - AI-powered post-deploy testing agent."""
+
+__version__ = "0.1.0"

testa_agent-0.1.0/testa_agent/cli.py

@@ -0,0 +1,108 @@
+"""CLI entry point for the Testa agent."""
+
+import asyncio
+import logging
+import os
+import sys
+
+import click
+
+
+@click.group()
+@click.version_option(package_name="testa")
+def main():
+    """Testa - AI-powered post-deploy testing."""
+    pass
+
+
+@main.command()
+@click.option(
+    "--config",
+    "config_path",
+    default="./testa.yaml",
+    help="Path to testa.yaml config file.",
+    type=click.Path(exists=True),
+)
+@click.option(
+    "--api-url",
+    default=None,
+    help="Testa API URL (or set TESTA_API_URL).",
+)
+@click.option(
+    "--license-key",
+    default=None,
+    help="License key (or set TESTA_LICENSE_KEY).",
+)
+@click.option(
+    "--headless/--no-headless",
+    default=True,
+    help="Run browser in headless mode.",
+)
+@click.option(
+    "--timeout",
+    default=30000,
+    help="Default step timeout in ms.",
+    type=int,
+)
+def run(config_path, api_url, license_key, headless, timeout):
+    """Run tests once and exit (CI mode).
+
+    Reads tests from a local testa.yaml, executes them, reports results
+    to the Testa API, and exits with code 0 (all pass) or 1 (any fail).
+
+    Example:
+        testa run --config ./testa.yaml
+    """
+    # Configure logging for CI output
+    logging.basicConfig(
+        level=logging.INFO,
+        format="%(asctime)s [%(levelname)s] %(message)s",
+        handlers=[logging.StreamHandler(sys.stdout)],
+    )
+
+    # Resolve config from args or env vars
+    resolved_key = license_key or os.getenv("TESTA_LICENSE_KEY") or os.getenv("LICENSE_KEY")
+    if not resolved_key:
+        click.echo(
+            "Error: License key required. Use --license-key or set TESTA_LICENSE_KEY.", err=True
+        )
+        sys.exit(1)
+
+    resolved_url: str = (
+        api_url or os.getenv("TESTA_API_URL") or os.getenv("API_URL") or "http://localhost:8000"
+    )
+
+    from testa_agent.runner import ci_run
+
+    exit_code = asyncio.run(
+        ci_run(
+            config_path=config_path,
+            api_url=resolved_url,
+            license_key=resolved_key,
+            headless=headless,
+            default_timeout_ms=timeout,
+        )
+    )
+    sys.exit(exit_code)
+
+
+@main.command()
+def serve():
+    """Start the agent in polling mode (self-hosted).
+
+    Registers with the Testa API and continuously polls for work.
+    This is the existing agent behavior for self-hosted deployments.
+
+    Environment variables:
+        LICENSE_KEY: Required license key
+        API_URL: Testa API URL (default: http://localhost:8000)
+        CONFIG_PATH: Path to testa.yaml (default: /config/testa.yaml)
+        ANTHROPIC_API_KEY: For direct vision calls
+    """
+    from testa_agent.main import run
+
+    run()
+
+
+if __name__ == "__main__":
+    main()

testa_agent-0.1.0/testa_agent/config.py

@@ -0,0 +1,64 @@
+import os
+from dataclasses import dataclass
+from typing import Optional
+
+
+@dataclass
+class SlackConfig:
+    """Slack notification configuration."""
+
+    webhook_url: Optional[str] = None
+    bot_token: Optional[str] = None  # For screenshot uploads
+    channel_id: Optional[str] = None  # Required if bot_token is set
+    channel_name: Optional[str] = None
+
+    @property
+    def is_configured(self) -> bool:
+        """Check if Slack is configured."""
+        return bool(self.webhook_url)
+
+    @property
+    def supports_screenshots(self) -> bool:
+        """Check if screenshot uploads are supported."""
+        return bool(self.bot_token and self.channel_id)
+
+
+@dataclass
+class AgentConfig:
+    """Agent configuration loaded from environment variables."""
+
+    license_key: str
+    api_url: str
+    config_path: str
+    poll_interval_seconds: int = 30
+    slack: Optional[SlackConfig] = None
+
+    @classmethod
+    def from_env(cls) -> "AgentConfig":
+        """Load configuration from environment variables."""
+        license_key = os.getenv("LICENSE_KEY")
+        if not license_key:
+            raise ValueError("LICENSE_KEY environment variable is required")
+
+        api_url = os.getenv("API_URL", "http://localhost:8000")
+        config_path = os.getenv("CONFIG_PATH", "/config/testa.yaml")
+        poll_interval = int(os.getenv("POLL_INTERVAL_SECONDS", "30"))
+
+        # Load Slack config from environment
+        slack_config = None
+        slack_webhook = os.getenv("SLACK_WEBHOOK_URL")
+        if slack_webhook:
+            slack_config = SlackConfig(
+                webhook_url=slack_webhook,
+                bot_token=os.getenv("SLACK_BOT_TOKEN"),
+                channel_id=os.getenv("SLACK_CHANNEL_ID"),
+                channel_name=os.getenv("SLACK_CHANNEL_NAME"),
+            )
+
+        return cls(
+            license_key=license_key,
+            api_url=api_url,
+            config_path=config_path,
+            poll_interval_seconds=poll_interval,
+            slack=slack_config,
+        )