jac-loadtest 0.2.2__tar.gz → 0.3.0__tar.gz

{jac_loadtest-0.2.2 → jac_loadtest-0.3.0}/PKG-INFO

@@ -1,17 +1,29 @@
 Metadata-Version: 2.4
 Name: jac-loadtest
-Version: 0.2.2
-Summary: HAR-based load testing CLI for jac-scale applications
+Version: 0.3.0
+Summary: HAR-based load testing library for jac-scale applications
+Author-email: Sahan Udayanga <sahanudayangaof@gmail.com>, Ravimal Ranathunga <ravimalranathunga@gmail.com>
+Maintainer-email: Sahan Udayanga <sahanudayangaof@gmail.com>, Ravimal Ranathunga <ravimalranathunga@gmail.com>
+License-Expression: MIT
+Keywords: load-testing,har,jac,jaclang,jaseci,jac-scale,performance
 Requires-Python: >=3.12
-Description-Content-Type: text/markdown
-Requires-Dist: jaclang==0.15.2
-Requires-Dist: jac-scale==0.2.16
+Project-URL: Repository, https://github.com/SahanUday/jac-loadtest
+Project-URL: Source, https://github.com/SahanUday/jac-loadtest/main/jac-loadtest-cli
+Project-URL: Issues, https://github.com/SahanUday/jac-loadtest/issues
+Requires-Dist: jac-scale>=0.2.16
+Requires-Dist: aiohttp>=3.9.0,<4.0.0
 Requires-Dist: rich>=13.0.0
+Requires-Dist: requests>=2.28.0
 Provides-Extra: test
 Requires-Dist: pytest>=8.0; extra == "test"
 Requires-Dist: pytest-asyncio>=0.23; extra == "test"
 Requires-Dist: pytest-mock>=3.12; extra == "test"
-Requires-Dist: aiohttp[speedups]>=3.9; extra == "test"
+Requires-Dist: aiohttp[speedups]>=3.9.0,<4.0.0; extra == "test"
+Provides-Extra: dev
+Requires-Dist: pytest>=8.0; extra == "dev"
+Requires-Dist: pytest-asyncio>=0.23; extra == "dev"
+Requires-Dist: pytest-mock>=3.12; extra == "dev"
+Description-Content-Type: text/markdown
 
 # jac-loadtest
 
@@ -63,10 +75,10 @@ conda create -n load python=3.12
 conda activate load
 
 # 2. Install the package in editable mode (runtime deps only)
-pip install -e .
+jac install -e .
 
 # 3. Also install test dependencies (when running tests)
-pip install -e ".[test]"
+jac install -x test
 
 # 4. Verify the command is registered
 jac loadtest --help
@@ -75,17 +87,15 @@ jac loadtest --help
 ## Project Layout
 
 ```
-jac_loadtest/
-├── plugin.py      — registers `jac loadtest` via jaclang entry-points
-├── cli.py         — argument wiring; JacMetaImporter bootstrap
-├── config.py      — LoadTestConfig dataclass (three-layer resolution)
+jac_loadtest_cli/
+├── plugin.jac     — registers `jac loadtest` via jaclang entry-points
+├── cli.jac        — argument wiring and run orchestration
+├── config.jac     — LoadTestConfig dataclass (three-layer resolution)
 ├── core/          — HAR parser, load engine, metrics (no jac-scale knowledge)
 ├── bridge/        — jac-scale-aware adapters (auth, topology)
 └── output/        — console, JSON, HTML reporters
 ```
 
-The hard boundary between `core/` and `bridge/` is what keeps the eventual migration to `jac-scale[loadtest]` a file move rather than a rewrite.
-
 ## HAR Compatibility
 
 Tested with HAR **1.1** and **1.2** (the format exported by Chrome DevTools, Firefox, Postman, and Insomnia). Files from other versions are parsed with a warning — open an issue if something breaks.

{jac_loadtest-0.2.2 → jac_loadtest-0.3.0}/README.md

@@ -48,10 +48,10 @@ conda create -n load python=3.12
 conda activate load
 
 # 2. Install the package in editable mode (runtime deps only)
-pip install -e .
+jac install -e .
 
 # 3. Also install test dependencies (when running tests)
-pip install -e ".[test]"
+jac install -x test
 
 # 4. Verify the command is registered
 jac loadtest --help
@@ -60,17 +60,15 @@ jac loadtest --help
 ## Project Layout
 
 ```
-jac_loadtest/
-├── plugin.py      — registers `jac loadtest` via jaclang entry-points
-├── cli.py         — argument wiring; JacMetaImporter bootstrap
-├── config.py      — LoadTestConfig dataclass (three-layer resolution)
+jac_loadtest_cli/
+├── plugin.jac     — registers `jac loadtest` via jaclang entry-points
+├── cli.jac        — argument wiring and run orchestration
+├── config.jac     — LoadTestConfig dataclass (three-layer resolution)
 ├── core/          — HAR parser, load engine, metrics (no jac-scale knowledge)
 ├── bridge/        — jac-scale-aware adapters (auth, topology)
 └── output/        — console, JSON, HTML reporters
 ```
 
-The hard boundary between `core/` and `bridge/` is what keeps the eventual migration to `jac-scale[loadtest]` a file move rather than a rewrite.
-
 ## HAR Compatibility
 
 Tested with HAR **1.1** and **1.2** (the format exported by Chrome DevTools, Firefox, Postman, and Insomnia). Files from other versions are parsed with a warning — open an issue if something breaks.

jac_loadtest-0.3.0/_jac_build.py

@@ -0,0 +1,19 @@
+"""Bundled PEP 517 build backend — generated by jac bundle, do not edit."""
+
+
+def get_requires_for_build_wheel(config_settings=None):
+    return []
+
+
+def build_wheel(wheel_directory, config_settings=None, metadata_directory=None):
+    from pathlib import Path
+    from jaclang.project.config import get_config
+    from jaclang.publish.builder import WheelBuilder
+
+    proj_root = Path.cwd()
+    config = get_config(start_path=proj_root, force_discover=True)
+    if config is None:
+        raise RuntimeError(f"No jac.toml found in {proj_root}.")
+    return WheelBuilder(config=config, project_root=proj_root).build_to(
+        Path(wheel_directory)
+    ).name

jac_loadtest-0.3.0/jac.toml

@@ -0,0 +1,41 @@
+[project]
+name = "jac-loadtest"
+version = "0.3.0"
+description = "HAR-based load testing library for jac-scale applications"
+authors = [{ name = "Sahan Udayanga", email = "sahanudayangaof@gmail.com" }, { name = "Ravimal Ranathunga", email = "ravimalranathunga@gmail.com" }]
+maintainers = [{ name = "Sahan Udayanga", email = "sahanudayangaof@gmail.com" }, { name = "Ravimal Ranathunga", email = "ravimalranathunga@gmail.com" }]
+license = "MIT"
+readme = "README.md"
+keywords = ["load-testing", "har", "jac", "jaclang", "jaseci", "jac-scale", "performance"]
+requires-python = ">=3.12"
+
+[project.urls]
+repository = "https://github.com/SahanUday/jac-loadtest"
+source = "https://github.com/SahanUday/jac-loadtest/main/jac-loadtest-cli"
+issues = "https://github.com/SahanUday/jac-loadtest/issues"
+
+[project.include]
+packages = ["jac_loadtest_cli"]
+
+[project.include.data]
+jac_loadtest_cli = ["templates/*.html"]
+
+[dependencies]
+jac-scale = ">=0.2.16"
+aiohttp = ">=3.9.0,<4.0.0"
+rich = ">=13.0.0"
+requests = ">=2.28.0"
+
+[dev-dependencies]
+pytest = ">=8.0"
+pytest-asyncio = ">=0.23"
+pytest-mock = ">=3.12"
+
+[optional-dependencies.test]
+pytest = ">=8.0"
+pytest-asyncio = ">=0.23"
+pytest-mock = ">=3.12"
+"aiohttp[speedups]" = ">=3.9.0,<4.0.0"
+
+[entrypoints.jac]
+loadtest = "jac_loadtest_cli.plugin:loadtest"

jac_loadtest-0.3.0/jac_loadtest_cli/bridge/auth.jac

@@ -0,0 +1,94 @@
+import csv;
+import aiohttp;
+import from ..config { LoadTestConfig }
+
+class AuthenticationError(Exception) { }
+
+obj Credential {
+    has username: str;
+    has password: str;
+}
+
+obj AuthProvider {
+    has _credentials: list[Credential];
+    has _login_path: str = "/user/login";
+
+    class def from_config(config: LoadTestConfig) -> AuthProvider | None {
+        if config.credentials_file {
+            creds = _load_csv(config.credentials_file);
+        } elif config.username and config.password {
+            creds = [Credential(username=config.username, password=config.password)];
+        } else {
+            return None;
+        }
+        return AuthProvider(_credentials=creds, _login_path=config.login_path);
+    }
+
+    def get_credential(vu_id: int) -> Credential {
+        return self._credentials[vu_id % len(self._credentials)];
+    }
+
+    async def authenticate(vu_id: int, session: aiohttp.ClientSession, base_url: str) -> str {
+        cred = self.get_credential(vu_id);
+        identity_type = "email" if "@" in cred.username else "username";
+        payload = {
+            "identity": {"type": identity_type, "value": cred.username},
+            "credential": {"type": "password", "password": cred.password},
+        };
+        url = base_url.rstrip("/") + self._login_path;
+        try {
+            async with session.post(url, json=payload) as resp {
+                if resp.status == 401 {
+                    raise AuthenticationError(
+                        f"Login failed for VU {vu_id} ({identity_type}={cred.username!r}): "
+                        "401 Unauthorized — check credentials."
+                    );
+                }
+                if not resp.ok {
+                    raise AuthenticationError(
+                        f"Login failed for VU {vu_id} ({identity_type}={cred.username!r}): "
+                        f"server returned {resp.status}."
+                    );
+                }
+                body = await resp.json();
+            }
+        } except aiohttp.ClientConnectorError as exc {
+            raise AuthenticationError(
+                f"Cannot reach login endpoint {url!r}: {exc}"
+            ) from exc;
+        }
+        try {
+            return body["data"]["token"];
+        } except (KeyError, TypeError) as exc {
+            raise AuthenticationError(
+                f"Unexpected login response shape from {url!r}: {body!r}"
+            ) from exc;
+        }
+    }
+}
+
+def _load_csv(path: str) -> list[Credential] {
+    credentials = [];
+    with open(path, newline="") as f {
+        reader = csv.reader(f);
+        i = 0;
+        for row in reader {
+            if len(row) < 2 {
+                i += 1;
+                continue;
+            }
+            username = row[0].strip();
+            password = row[1].strip();
+            if i == 0 and username.lower() == "username" {
+                i += 1;
+                continue;
+            }
+            credentials.append(Credential(username=username, password=password));
+            i += 1;
+        }
+    }
+    if not credentials {
+        raise ValueError(f"No credentials found in {path!r}");
+    }
+    return credentials;
+}

jac_loadtest-0.3.0/jac_loadtest_cli/bridge/topology.jac

@@ -0,0 +1,165 @@
+import json;
+import os;
+import from urllib.parse { urlparse, urlunparse }
+import from ..config { LoadTestConfig }
+
+obj ServiceRoute {
+    has name: str;
+    has prefix: str;
+    has url: str;
+}
+
+obj TopologyRouter {
+    has _routes: list;
+    has _fallback_url: str | None = None;
+
+    def __post_init__() {
+        self._routes = sorted(self._routes, key=lambda r: len(r.prefix), reverse=True);
+    }
+
+    def resolve(entry_url: str) -> tuple {
+        parsed = urlparse(entry_url);
+        path = parsed.path;
+
+        for route in self._routes {
+            if _prefix_matches(path, route.prefix) {
+                t = urlparse(route.url);
+                if route.prefix {
+                    service_path = path[len(route.prefix):] or "/";
+                } else {
+                    service_path = path;
+                }
+                base_path = t.path.rstrip("/") if t.path else "";
+                routed_path = f"{base_path}{service_path}" if base_path else service_path;
+                routed = urlunparse((
+                    t.scheme, t.netloc,
+                    routed_path, parsed.params, parsed.query, "",
+                ));
+                return (routed, route.name);
+            }
+        }
+
+        if self._fallback_url {
+            t = urlparse(self._fallback_url);
+            base_path = t.path.rstrip("/") if t.path else "";
+            routed_path = f"{base_path}{path}" if base_path else path;
+            routed = urlunparse((
+                t.scheme, t.netloc,
+                routed_path, parsed.params, parsed.query, "",
+            ));
+            return (routed, "gateway");
+        }
+
+        raise ValueError(
+            f"No route for path '{path}' and no --url fallback provided. "
+            "Use --services-map or add [plugins.scale.microservices.routes] to jac.toml."
+        );
+    }
+
+    class def from_config(config: LoadTestConfig) -> TopologyRouter {
+        if config.mode != "microservice" {
+            return TopologyRouter(
+                _routes=[ServiceRoute(name="monolith", prefix="", url=config.url or "")],
+                _fallback_url=None,
+            );
+        }
+        return _build_microservice_topology(config);
+    }
+}
+
+def _build_microservice_topology(config: LoadTestConfig) -> TopologyRouter {
+    toml_routes = _load_toml_routes();
+
+    if config.services_map {
+        try {
+            services_json = json.loads(config.services_map);
+        } except json.JSONDecodeError as exc {
+            raise ValueError(f"--services-map is not valid JSON: {exc}") from exc;
+        }
+
+        routes = [];
+        for item in services_json.items() {
+            name = item[0];
+            url = item[1];
+            prefix = toml_routes.get(name);
+            if prefix is None {
+                if name.startswith("/") {
+                    prefix = name;
+                } else {
+                    raise ValueError(
+                        f"--services-map key '{name}' has no matching route in jac.toml "
+                        "and does not start with '/'. "
+                        "Use a path prefix as the key or add '[plugins.scale.microservices.routes]' to jac.toml."
+                    );
+                }
+            }
+            routes.append(ServiceRoute(
+                name=name.lstrip("/") if name.startswith("/") else name,
+                prefix=prefix,
+                url=url.rstrip("/"),
+            ));
+        }
+        return TopologyRouter(_routes=routes, _fallback_url=config.url);
+    }
+
+    if not toml_routes {
+        raise ValueError(
+            "Microservice mode requires either --services-map or "
+            "[plugins.scale.microservices.routes] in jac.toml. Neither was found."
+        );
+    }
+
+    routes = [];
+    missing = [];
+    for item in toml_routes.items() {
+        name = item[0];
+        prefix = item[1];
+        env_var = f"JAC_SV_{name.upper()}_URL";
+        url = os.environ.get(env_var);
+        if not url {
+            missing.append(env_var);
+        } else {
+            routes.append(ServiceRoute(name=name, prefix=prefix, url=url.rstrip("/")));
+        }
+    }
+
+    if missing {
+        raise ValueError(
+            f"Microservice mode: missing environment variable(s): {', '.join(missing)}. "
+            "Expected format: JAC_SV_<SERVICE_NAME>_URL=http://host:port"
+        );
+    }
+
+    return TopologyRouter(_routes=routes, _fallback_url=config.url);
+}
+
+def _prefix_matches(path: str, prefix: str) -> bool {
+    if not prefix {
+        return True;
+    }
+    return path == prefix or path.startswith(prefix + "/");
+}
+
+def _load_toml_routes -> dict {
+    try {
+        import from pathlib { Path }
+        import from jac_scale.config_loader { get_scale_config, reset_scale_config }
+        reset_scale_config();
+        ms_config = get_scale_config(project_dir=Path.cwd()).get_microservices_config();
+        return dict(ms_config.get("routes", {}));
+    } except Exception {
+        return {};
+    }
+}
+
+def resolve_url(path: str, routing_table: dict, fallback_url: str | None) -> str {
+    for prefix in sorted(routing_table, key=len, reverse=True) {
+        if path.startswith(prefix) {
+            return routing_table[prefix];
+        }
+    }
+    if fallback_url {
+        return fallback_url;
+    }
+    raise ValueError(f"No route for path '{path}' and no --url fallback provided.");
+}

jac_loadtest-0.3.0/jac_loadtest_cli/cli.jac

@@ -0,0 +1,152 @@
+import sys;
+import asyncio;
+import time;
+import from .config { from_args, LoadTestConfig }
+import from .core.har_parser { parse_har, HarEntry }
+import from .core.engine { run_all_vus }
+import from .core.metrics { MetricsCollector }
+import from .core.process_runner { run_multiprocess }
+import from .output.reporter { render_console, render_json, render_html }
+import from .bridge.auth { AuthProvider, AuthenticationError }
+import from .bridge.topology { TopologyRouter }
+import from rich.console { Console as _Console }
+
+def run(args: object) {
+    config: LoadTestConfig = from_args(args);
+
+    if config.mode == "monolith" and not config.url {
+        print("Error: --url is required for monolith mode", file=sys.stderr);
+        sys.exit(2);
+    }
+
+    if not config.har_file {
+        print("Error: har_file positional argument is required", file=sys.stderr);
+        sys.exit(2);
+    }
+
+    entries: list[HarEntry] = [];
+    try {
+        entries = parse_har(
+            config.har_file,
+            target_url=config.url if config.mode == "monolith" else None,
+            include_static=config.include_static,
+            login_path=config.login_path,
+        );
+    } except (FileNotFoundError, ValueError) as exc {
+        print(f"Error: {exc}", file=sys.stderr);
+        sys.exit(2);
+    }
+
+    if not entries {
+        print(
+            "Error: no API entries found in HAR file after filtering. "
+            "Use --include-static to include static assets.",
+            file=sys.stderr,
+        );
+        sys.exit(2);
+    }
+
+    auth_provider: AuthProvider | None = AuthProvider.from_config(config);
+
+    if config.mode == "microservice" and auth_provider is not None and not config.url {
+        print(
+            "Error: --url (gateway URL) is required when using authentication "
+            "in microservice mode",
+            file=sys.stderr,
+        );
+        sys.exit(2);
+    }
+
+    try {
+        topology = TopologyRouter.from_config(config);
+    } except ValueError as exc {
+        print(f"Error: {exc}", file=sys.stderr);
+        sys.exit(2);
+    }
+
+    _console = _Console(stderr=True);
+
+    t_start = time.time();
+
+    try {
+        with _console.status(f"Running load test — {config.vus} VUs...") {
+            if config.workers > 1 {
+                metrics = run_multiprocess(entries, config, topology, auth_provider);
+            } else {
+                metrics = MetricsCollector(max_samples=config.max_samples);
+                asyncio.run(
+                    run_all_vus(
+                        entries, config, metrics,
+                        topology=topology,
+                        auth_provider=auth_provider,
+                    )
+                );
+            }
+        }
+    } except AuthenticationError as exc {
+        print(f"Error: {exc}", file=sys.stderr);
+        sys.exit(2);
+    }
+
+    duration_s = time.time() - t_start;
+    stats = metrics.compute_endpoint_stats(t_start=t_start);
+    snapshots = metrics.generate_timeseries(t_start);
+    _comp = metrics.completion_percentiles(t_start);
+    completion_p50 = _comp[0];
+    completion_p95 = _comp[1];
+    completion_p99 = _comp[2];
+
+    fmt = config.report_format;
+
+    if fmt == "json" {
+        output = render_json(
+            stats, config,
+            actual_duration_s=duration_s,
+            total_rps=metrics.global_rps(duration_s),
+            snapshots=snapshots,
+            completion_p50_s=completion_p50,
+            completion_p95_s=completion_p95,
+            completion_p99_s=completion_p99,
+        );
+        if config.report_out {
+            with open(config.report_out, "w", encoding="utf-8") as fh {
+                fh.write(output);
+            }
+            print(f"JSON report written to {config.report_out}");
+        } else {
+            print(output);
+        }
+    } elif fmt == "html" {
+        if not config.report_out {
+            print(
+                "Error: --report-out <path> is required for --report-format html",
+                file=sys.stderr,
+            );
+            sys.exit(2);
+        }
+        output = render_html(
+            stats, config,
+            actual_duration_s=duration_s,
+            total_rps=metrics.global_rps(duration_s),
+            snapshots=snapshots,
+            completion_p50_s=completion_p50,
+            completion_p95_s=completion_p95,
+            completion_p99_s=completion_p99,
+        );
+        with open(config.report_out, "w", encoding="utf-8") as fh {
+            fh.write(output);
+        }
+        print(f"HTML report written to {config.report_out}", file=sys.stderr);
+    } else {
+        render_console(
+            stats, config,
+            actual_duration_s=duration_s,
+            total_rps=metrics.global_rps(duration_s),
+            completion_p50_s=completion_p50,
+            completion_p95_s=completion_p95,
+            completion_p99_s=completion_p99,
+        );
+    }
+}
+
+