WPTServe: Run .any.js and .worker.js variations

third_party/WebKit/Tools/Scripts/webkitpy/w3c/wpt_manifest.py

 # Copyright 2017 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 """WPTManifest is responsible for handling MANIFEST.json.
 
 The MANIFEST.json file contains metadata about files in web-platform-tests,
 such as what tests exist, and extra information about each test, including
 test type, options, URLs to use, and reference file paths if applicable.
 """
 
 import json
 import logging
 
+from webkitpy.common.memoized import memoized
 from webkitpy.common.webkit_finder import WebKitFinder
 
 _log = logging.getLogger(__file__)
 
 
 class WPTManifest(object):
 
     def __init__(self, json_content):
         # TODO(tkent): Create a Manifest object by Manifest.from_json().
         # See ../thirdparty/wpt/wpt/tools/manifest/manifest.py.
@@ skipping 12 matching lines @@
         """
         items = self.raw_dict['items']
         if path_in_wpt in items['manual']:
             return items['manual'][path_in_wpt]
         elif path_in_wpt in items['reftest']:
             return items['reftest'][path_in_wpt]
         elif path_in_wpt in items['testharness']:
             return items['testharness'][path_in_wpt]
         return None
 
+    @memoized
+    def _urls(self):
+        """Returns a set of the urls for all items in the manifest."""
+        urls = set()
+        if 'items' in self.raw_dict:
+            items = self.raw_dict['items']
+            for category in ('manual', 'reftest', 'testharness'):
+                if category in items:
+                    for records in items[category].values():
+                        urls.update([item[0] for item in records])
+        return urls
+
     def is_test_file(self, path_in_wpt):
         return self._items_for_path(path_in_wpt) is not None
 
+    def is_test_url(self, url):
+        """Check if the passed 'url' is associated with a test
+        in the manifest. The url must be just the pathname,
+        with a leading slash (/)."""
+        if url[0] != '/':
+            raise Exception('Test url missing leading /: %s' % url)
+        return url in self._urls()
+
     def file_path_to_url_paths(self, path_in_wpt):
         manifest_items = self._items_for_path(path_in_wpt)
         assert manifest_items is not None
-        if len(manifest_items) != 1:
-            return []
-        url = manifest_items[0][0]
-        if url[1:] != path_in_wpt:
-            # TODO(tkent): foo.any.js and bar.worker.js should be accessed
-            # as foo.any.html, foo.any.worker, and bar.worker with WPTServe.
-            return []
-        return [path_in_wpt]
+        return [item[0][1:] for item in manifest_items]
 
     @staticmethod
     def _get_extras_from_item(item):
         return item[-1]
 
     def is_slow_test(self, test_name):
         items = self._items_for_path(test_name)
         if not items:
             return False
         extras = WPTManifest._get_extras_from_item(items[0])
@@ skipping 46 matching lines @@
         proc = executive.popen(cmd, stdout=executive.PIPE, stderr=executive.PIPE, stdin=executive.PIPE, cwd=finder.webkit_base())
         out, err = proc.communicate('')
         if proc.returncode:
             _log.info('# ret> %d', proc.returncode)
             if out:
                 _log.info(out)
             if err:
                 _log.info(err)
             host.exit(proc.returncode)
         return proc.returncode, out