98 lines
3.4 KiB
Python
98 lines
3.4 KiB
Python
import argparse
|
|
import json
|
|
import os
|
|
from collections import defaultdict
|
|
from typing import Any, Dict, Iterable, List, Text
|
|
|
|
from .manifest import load_and_update, Manifest
|
|
from .log import get_logger
|
|
|
|
wpt_root = os.path.abspath(os.path.join(os.path.dirname(__file__), os.pardir, os.pardir))
|
|
|
|
logger = get_logger()
|
|
|
|
|
|
def abs_path(path: str) -> str:
|
|
return os.path.abspath(os.path.expanduser(path))
|
|
|
|
|
|
def create_parser() -> argparse.ArgumentParser:
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument(
|
|
"-p", "--path", type=abs_path, help="Path to manifest file.")
|
|
parser.add_argument(
|
|
"--src-root", type=abs_path, help="Path to root of sourcetree.")
|
|
parser.add_argument(
|
|
"--tests-root", type=abs_path, default=wpt_root, help="Path to root of tests.")
|
|
parser.add_argument(
|
|
"--no-update", dest="update", action="store_false",
|
|
help="Don't update manifest before continuing")
|
|
parser.add_argument(
|
|
"-r", "--rebuild", action="store_true",
|
|
help="Force a full rebuild of the manifest.")
|
|
parser.add_argument(
|
|
"--url-base", default="/",
|
|
help="Base url to use as the mount point for tests in this manifest.")
|
|
parser.add_argument(
|
|
"--cache-root", default=os.path.join(wpt_root, ".wptcache"),
|
|
help="Path in which to store any caches (default <tests_root>/.wptcache/)")
|
|
parser.add_argument(
|
|
"--json", action="store_true",
|
|
help="Output as JSON")
|
|
parser.add_argument(
|
|
"test_ids", nargs="+",
|
|
help="Test ids for which to get paths")
|
|
return parser
|
|
|
|
|
|
def get_path_id_map(src_root: Text, tests_root: Text, manifest_file: Manifest, test_ids: Iterable[Text]) -> Dict[Text, List[Text]]:
|
|
test_ids = set(test_ids)
|
|
path_id_map: Dict[Text, List[Text]] = defaultdict(list)
|
|
|
|
compute_rel_path = src_root != tests_root
|
|
|
|
for item_type, path, tests in manifest_file:
|
|
for test in tests:
|
|
if test.id in test_ids:
|
|
if compute_rel_path:
|
|
rel_path = os.path.relpath(os.path.join(tests_root, path),
|
|
src_root)
|
|
else:
|
|
rel_path = path
|
|
path_id_map[rel_path].append(test.id)
|
|
return path_id_map
|
|
|
|
|
|
def get_paths(**kwargs: Any) -> Dict[Text, List[Text]]:
|
|
tests_root = kwargs["tests_root"]
|
|
assert tests_root is not None
|
|
path = kwargs["path"]
|
|
if path is None:
|
|
path = os.path.join(kwargs["tests_root"], "MANIFEST.json")
|
|
src_root = kwargs["src_root"]
|
|
if src_root is None:
|
|
src_root = tests_root
|
|
|
|
manifest_file = load_and_update(tests_root,
|
|
path,
|
|
kwargs["url_base"],
|
|
update=kwargs["update"],
|
|
rebuild=kwargs["rebuild"],
|
|
cache_root=kwargs["cache_root"])
|
|
|
|
return get_path_id_map(src_root, tests_root, manifest_file, kwargs["test_ids"])
|
|
|
|
|
|
def write_output(path_id_map: Dict[Text, List[Text]], as_json: bool) -> None:
|
|
if as_json:
|
|
print(json.dumps(path_id_map))
|
|
else:
|
|
for path, test_ids in sorted(path_id_map.items()):
|
|
print(path)
|
|
for test_id in sorted(test_ids):
|
|
print(" " + test_id)
|
|
|
|
|
|
def run(**kwargs: Any) -> None:
|
|
path_id_map = get_paths(**kwargs)
|
|
write_output(path_id_map, as_json=kwargs["json"])
|