import os.path
from abc import ABCMeta, abstractproperty
from inspect import isabstract
from typing import (Any, Dict, Hashable, List, Optional, Sequence, Text, Tuple, Type,
TYPE_CHECKING, Union, cast)
from urllib.parse import urljoin, urlparse, parse_qs
from .utils import to_os_path
if TYPE_CHECKING:
from .manifest import Manifest
Fuzzy = Dict[Optional[Tuple[str, str, str]], List[int]]
PageRanges = Dict[str, List[int]]
item_types: Dict[str, Type["ManifestItem"]] = {}
class ManifestItemMeta(ABCMeta):
"""Custom metaclass that registers all the subclasses in the
item_types dictionary according to the value of their item_type
attribute, and otherwise behaves like an ABCMeta."""
def __new__(cls: Type["ManifestItemMeta"], name: str, bases: Tuple[type], attrs: Dict[str, Any]) -> "ManifestItemMeta":
inst = super().__new__(cls, name, bases, attrs)
if isabstract(inst):
return inst
assert issubclass(inst, ManifestItem)
item_type = cast(str, inst.item_type)
item_types[item_type] = inst
return inst
class ManifestItem(metaclass=ManifestItemMeta):
__slots__ = ("_tests_root", "path")
def __init__(self, tests_root: Text, path: Text) -> None:
self._tests_root = tests_root
self.path = path
@abstractproperty
def id(self) -> Text:
"""The test's id (usually its url)"""
pass
@abstractproperty
def item_type(self) -> str:
"""The item's type"""
pass
@property
def path_parts(self) -> Tuple[Text, ...]:
return tuple(self.path.split(os.path.sep))
def key(self) -> Hashable:
"""A unique identifier for the test"""
return (self.item_type, self.id)
def __eq__(self, other: Any) -> bool:
if not hasattr(other, "key"):
return False
return bool(self.key() == other.key())
def __hash__(self) -> int:
return hash(self.key())
def __repr__(self) -> str:
return f"<{self.__module__}.{self.__class__.__name__} id={self.id!r}, path={self.path!r}>"
def to_json(self) -> Tuple[Any, ...]:
return ()
@classmethod
def from_json(cls,
manifest: "Manifest",
path: Text,
obj: Any
) -> "ManifestItem":
path = to_os_path(path)
tests_root = manifest.tests_root
assert tests_root is not None
return cls(tests_root, path)
class URLManifestItem(ManifestItem):
__slots__ = ("url_base", "_url", "_extras", "_flags")
def __init__(self,
tests_root: Text,
path: Text,
url_base: Text,
url: Optional[Text],
**extras: Any
) -> None:
super().__init__(tests_root, path)
assert url_base[0] == "/"
self.url_base = url_base
assert url is None or url[0] != "/"
self._url = url
self._extras = extras
parsed_url = urlparse(self.url)
self._flags = (set(parsed_url.path.rsplit("/", 1)[1].split(".")[1:-1]) |
set(parse_qs(parsed_url.query).get("wpt_flags", [])))
@property
def id(self) -> Text:
return self.url
@property
def url(self) -> Text:
rel_url = self._url or self.path.replace(os.path.sep, "/")
# we can outperform urljoin, because we know we just have path relative URLs
if self.url_base == "/":
return "/" + rel_url
return urljoin(self.url_base, rel_url)
@property
def https(self) -> bool:
return "https" in self._flags or "serviceworker" in self._flags or "serviceworker-module" in self._flags
@property
def h2(self) -> bool:
return "h2" in self._flags
@property
def subdomain(self) -> bool:
# Note: this is currently hard-coded to check for `www`, rather than
# all possible valid subdomains. It can be extended if needed.
return "www" in self._flags
def to_json(self) -> Tuple[Optional[Text], Dict[Any, Any]]:
rel_url = None if self._url == self.path.replace(os.path.sep, "/") else self._url
rv: Tuple[Optional[Text], Dict[Any, Any]] = (rel_url, {})
return rv
@classmethod
def from_json(cls,
manifest: "Manifest",
path: Text,
obj: Tuple[Text, Dict[Any, Any]]
) -> "URLManifestItem":
path = to_os_path(path)
url, extras = obj
tests_root = manifest.tests_root
assert tests_root is not None
return cls(tests_root,
path,
manifest.url_base,
url,
**extras)
class TestharnessTest(URLManifestItem):
__slots__ = ()
item_type = "testharness"
@property
def timeout(self) -> Optional[Text]:
return self._extras.get("timeout")
@property
def pac(self) -> Optional[Text]:
return self._extras.get("pac")
@property
def testdriver(self) -> Optional[Text]:
return self._extras.get("testdriver")
@property
def jsshell(self) -> Optional[Text]:
return self._extras.get("jsshell")
@property
def script_metadata(self) -> Optional[List[Tuple[Text, Text]]]:
return self._extras.get("script_metadata")
def to_json(self) -> Tuple[Optional[Text], Dict[Text, Any]]:
rv = super().to_json()
if self.timeout is not None:
rv[-1]["timeout"] = self.timeout
if self.pac is not None:
rv[-1]["pac"] = self.pac
if self.testdriver:
rv[-1]["testdriver"] = self.testdriver
if self.jsshell:
rv[-1]["jsshell"] = True
if self.script_metadata:
rv[-1]["script_metadata"] = [(k, v) for (k,v) in self.script_metadata]
return rv
class RefTest(URLManifestItem):
__slots__ = ("references",)
item_type = "reftest"
def __init__(self,
tests_root: Text,
path: Text,
url_base: Text,
url: Optional[Text],
references: Optional[List[Tuple[Text, Text]]] = None,
**extras: Any
):
super().__init__(tests_root, path, url_base, url, **extras)
if references is None:
self.references: List[Tuple[Text, Text]] = []
else:
self.references = references
@property
def timeout(self) -> Optional[Text]:
return self._extras.get("timeout")
@property
def viewport_size(self) -> Optional[Text]:
return self._extras.get("viewport_size")
@property
def dpi(self) -> Optional[Text]:
return self._extras.get("dpi")
@property
def fuzzy(self) -> Fuzzy:
fuzzy: Union[Fuzzy, List[Tuple[Optional[Sequence[Text]], List[int]]]] = self._extras.get("fuzzy", {})
if not isinstance(fuzzy, list):
return fuzzy
rv: Fuzzy = {}
for k, v in fuzzy: # type: Tuple[Optional[Sequence[Text]], List[int]]
if k is None:
key: Optional[Tuple[Text, Text, Text]] = None
else:
# mypy types this as Tuple[Text, ...]
assert len(k) == 3
key = tuple(k) # type: ignore
rv[key] = v
return rv
def to_json(self) -> Tuple[Optional[Text], List[Tuple[Text, Text]], Dict[Text, Any]]: # type: ignore
rel_url = None if self._url == self.path else self._url
rv: Tuple[Optional[Text], List[Tuple[Text, Text]], Dict[Text, Any]] = (rel_url, self.references, {})
extras = rv[-1]
if self.timeout is not None:
extras["timeout"] = self.timeout
if self.viewport_size is not None:
extras["viewport_size"] = self.viewport_size
if self.dpi is not None:
extras["dpi"] = self.dpi
if self.fuzzy:
extras["fuzzy"] = list(self.fuzzy.items())
return rv
@classmethod
def from_json(cls, # type: ignore
manifest: "Manifest",
path: Text,
obj: Tuple[Text, List[Tuple[Text, Text]], Dict[Any, Any]]
) -> "RefTest":
tests_root = manifest.tests_root
assert tests_root is not None
path = to_os_path(path)
url, references, extras = obj
return cls(tests_root,
path,
manifest.url_base,
url,
references,
**extras)
class PrintRefTest(RefTest):
__slots__ = ("references",)
item_type = "print-reftest"
@property
def page_ranges(self) -> PageRanges:
return cast(PageRanges, self._extras.get("page_ranges", {}))
def to_json(self): # type: ignore
rv = super().to_json()
if self.page_ranges:
rv[-1]["page_ranges"] = self.page_ranges
return rv
class ManualTest(URLManifestItem):
__slots__ = ()
item_type = "manual"
class ConformanceCheckerTest(URLManifestItem):
__slots__ = ()
item_type = "conformancechecker"
class VisualTest(URLManifestItem):
__slots__ = ()
item_type = "visual"
class CrashTest(URLManifestItem):
__slots__ = ()
item_type = "crashtest"
@property
def timeout(self) -> Optional[Text]:
return None
class WebDriverSpecTest(URLManifestItem):
__slots__ = ()
item_type = "wdspec"
@property
def timeout(self) -> Optional[Text]:
return self._extras.get("timeout")
def to_json(self) -> Tuple[Optional[Text], Dict[Text, Any]]:
rv = super().to_json()
if self.timeout is not None:
rv[-1]["timeout"] = self.timeout
return rv
class SupportFile(ManifestItem):
__slots__ = ()
item_type = "support"
@property
def id(self) -> Text:
return self.path
class SpecItem(ManifestItem):
__slots__ = ("specs")
item_type = "spec"
def __init__(self,
tests_root: Text,
path: Text,
specs: List[Text]
) -> None:
super().__init__(tests_root, path)
self.specs = specs
@property
def id(self) -> Text:
return self.path
def to_json(self) -> Tuple[Optional[Text], Dict[Text, Any]]:
rv: Tuple[Optional[Text], Dict[Any, Any]] = (None, {})
for i in range(len(self.specs)):
spec_key = f"spec_link{i+1}"
rv[-1][spec_key] = self.specs[i]
return rv
@classmethod
def from_json(cls,
manifest: "Manifest",
path: Text,
obj: Any
) -> "ManifestItem":
"""Not properly implemented and is not used."""
return cls("/", "", [])