Source code
Revision control
Copy as Markdown
Other Tools
import argparse
import json
import os
import sys
from typing import TypedDict
import jsone
import yaml
here = os.path.dirname(__file__)
config_path = os.path.join(here, "..", "chunks.yml")
# Functional form required: keys contain hyphens, which are invalid Python identifiers
MatrixEntry = TypedDict(
"MatrixEntry",
{
"test-type": str,
"current-chunk": int,
"total-chunks": int,
"timeout-minutes": int,
},
)
def get_parser() -> argparse.ArgumentParser:
parser = argparse.ArgumentParser(
description="Generate a GitHub Actions test matrix from tools/ci/chunks.yml"
)
parser.add_argument("--browser", required=True, help="Browser name")
return parser
def get_matrix(browser: str, template: object = None) -> list[MatrixEntry]:
if template is None:
with open(config_path) as f:
template = yaml.safe_load(f)
config = jsone.render(template, {"browser": browser, "ci": "github-actions"})
default_timeout = config["defaults"]["timeout"]
includes: list[MatrixEntry] = []
for test_type, settings in config["test_types"].items():
total_chunks = settings["chunks"]
if total_chunks == 0:
continue
timeout: int = settings.get("timeout") or default_timeout
for chunk in range(1, total_chunks + 1):
includes.append(
{
"test-type": test_type,
"current-chunk": chunk,
"total-chunks": total_chunks,
"timeout-minutes": timeout,
}
)
return includes
def run(venv: object, browser: str, **kwargs: object) -> None:
json.dump({"include": get_matrix(browser)}, sys.stdout)
print()