| # Copyright 2023 The Cobalt Authors. All Rights Reserved. |
| # |
| # Licensed under the Apache License, Version 2.0 (the "License"); |
| # you may not use this file except in compliance with the License. |
| # You may obtain a copy of the License at |
| # |
| # http://www.apache.org/licenses/LICENSE-2.0 |
| # |
| # Unless required by applicable law or agreed to in writing, software |
| # distributed under the License is distributed on an "AS IS" BASIS, |
| # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| # See the License for the specific language governing permissions and |
| # limitations under the License. |
| """Tests Service Worker controller activation functionality.""" |
| |
| import logging |
| import os |
| from six.moves import SimpleHTTPServer |
| |
| from cobalt.black_box_tests import black_box_tests |
| from cobalt.black_box_tests.threaded_web_server import MakeRequestHandlerClass |
| from cobalt.black_box_tests.threaded_web_server import ThreadedWebServer |
| from cobalt.tools.automated_testing import webdriver_utils |
| |
| # The base path of the requested assets is the parent directory. |
| _SERVER_ROOT_PATH = os.path.join(os.path.dirname(__file__), os.pardir) |
| |
| keys = webdriver_utils.import_selenium_module('webdriver.common.keys') |
| |
| |
| class ServiceWorkerRequestDetector(MakeRequestHandlerClass(_SERVER_ROOT_PATH)): |
| """Proxies everything to SimpleHTTPRequestHandler, except some paths.""" |
| |
| def end_headers(self): |
| self.send_my_headers() |
| SimpleHTTPServer.SimpleHTTPRequestHandler.end_headers(self) |
| |
| def send_header(self, header, value): |
| # Ensure that the Content-Type for paths ending in '.js' are always |
| # 'text/javascript'. |
| if header == 'Content-Type' and self.path.endswith('.js'): |
| SimpleHTTPServer.SimpleHTTPRequestHandler.send_header( |
| self, header, 'text/javascript') |
| else: |
| SimpleHTTPServer.SimpleHTTPRequestHandler.send_header(self, header, value) |
| |
| def send_my_headers(self): |
| # Add 'Service-Worker-Allowed' header for the main service worker scripts. |
| if self.path.endswith( |
| '/service_worker_controller_activation_test_worker.js'): |
| self.send_header('Service-Worker-Allowed', '/testdata') |
| |
| |
| class ServiceWorkerControllerActivationTest(black_box_tests.BlackBoxTestCase): |
| """Test basic Service Worker functionality.""" |
| |
| def test_service_worker_controller_activation(self): |
| |
| with ThreadedWebServer( |
| ServiceWorkerRequestDetector, |
| binding_address=self.GetBindingAddress()) as server: |
| url = server.GetURL( |
| file_name='testdata/service_worker_controller_activation_test.html') |
| |
| with self.CreateCobaltRunner(url=url) as runner: |
| runner.WaitForJSTestsSetup() |
| |
| logging.info('SendKeys NUMPAD0.') |
| runner.SendKeys(keys.Keys.NUMPAD0) |
| logging.info('Wait.') |
| runner.PollUntilFoundOrTestsFailedWithReconnects( |
| '#ResultSuccessfulRegistration') |
| |
| logging.info('SendKeys NUMPAD1.') |
| runner.SendKeys(keys.Keys.NUMPAD1) |
| runner.PollUntilFoundOrTestsFailedWithReconnects('#ResultSuccess') |
| |
| logging.info('SendKeys NUMPAD2.') |
| runner.SendKeys(keys.Keys.NUMPAD2) |
| self.assertTrue(runner.JSTestsSucceeded()) |