telemetry/third_party/webpagereplay/script_injector_test.py - Issue 2210063003: Rename third_party/webpagereplay to third_party/web-page-replay

Unified Diff: telemetry/third_party/webpagereplay/script_injector_test.py

Issue 2210063003: Rename third_party/webpagereplay to third_party/web-page-replay (Closed) Base URL: https://github.com/catapult-project/catapult@master

Patch Set: Created 4 years, 4 months ago

Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.

Jump to:

View side-by-side diff with in-line comments

Download patch

« no previous file with comments | « telemetry/third_party/webpagereplay/script_injector.py ('k') | telemetry/third_party/webpagereplay/servermanager.py » ('j') | no next file with comments »
Expand Comments ('e') | Collapse Comments ('c') | Hide Comments ('s')

Index: telemetry/third_party/webpagereplay/script_injector_test.py

diff --git a/telemetry/third_party/webpagereplay/script_injector_test.py b/telemetry/third_party/webpagereplay/script_injector_test.py

deleted file mode 100755

index 5eed0986d5ddb240365073fe222f696cd589b63a..0000000000000000000000000000000000000000

--- a/telemetry/third_party/webpagereplay/script_injector_test.py

+++ /dev/null

@@ -1,129 +0,0 @@

-#!/usr/bin/env python

-# 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.

-import httparchive

-import script_injector

-import unittest

-LONG_COMMENT = ''

-COMMENT_OR_NOT = ('', LONG_COMMENT)

-SCRIPT_TO_INJECT = 'var flag = 0;'

-EXPECTED_SCRIPT = '<script>' + SCRIPT_TO_INJECT + '</script>'

-TEXT_HTML = 'text/html'

-TEXT_CSS = 'text/css'

-APPLICATION = 'application/javascript'

-SEPARATOR = httparchive.ArchivedHttpResponse.CHUNK_EDIT_SEPARATOR

-SEPARATORS_OR_NOT = ('', SEPARATOR, SEPARATOR*3)

-TEMPLATE_HEAD = """\

-{boundary_at_start}\

-<!doc{boundary_in_doctype}type html>{boundary_after_doctype}\

-<ht{boundary_in_html}ml>{boundary_after_html}\

-<he{boundary_in_head}ad>{injection}{boundary_after_head}\

-</head></html>\

-"""

-TEMPLATE_HTML = """\

-{boundary_at_start}\

-<!doc{boundary_in_doctype}type html>{boundary_after_doctype}\

-<ht{boundary_in_html}ml>{injection}{boundary_after_html}\

-</html>\

-"""

-TEMPLATE_DOCTYPE = """\

-{boundary_at_start}\

-<!doc{boundary_in_doctype}type html>{injection}{boundary_after_doctype}\

-<body></body>\

-"""

-TEMPLATE_RAW = """\

-{boundary_at_start}\

-{injection}<body></body>\

-"""

-NORMAL_TEMPLATES = (TEMPLATE_HEAD, TEMPLATE_HTML,

- TEMPLATE_DOCTYPE, TEMPLATE_RAW)

-TEMPLATE_COMMENT = """\

-{comment_before_doctype}<!doctype html>{comment_after_doctype}\

-<html>{comment_after_html}<head>{injection}</head></html>\

-"""

-def _wrap_inject_script(source, application, script_to_inject):

- text_chunks = source.split(SEPARATOR)

- text_chunks, just_injected = script_injector.InjectScript(

- text_chunks, application, script_to_inject)

- result = SEPARATOR.join(text_chunks)

- return result, just_injected

-class ScriptInjectorTest(unittest.TestCase):

- def _assert_no_injection(self, source, application):

- new_source, just_injected = _wrap_inject_script(

- source, application, SCRIPT_TO_INJECT)

- self.assertEqual(new_source, source)

- self.assertFalse(just_injected)

- def _assert_successful_injection(self, template):

- source, just_injected = _wrap_inject_script(

- template.format(injection=''), TEXT_HTML, SCRIPT_TO_INJECT)

- self.assertEqual(source, template.format(injection=EXPECTED_SCRIPT))

- self.assertTrue(just_injected)

- def test_unsupported_content_type(self):

- self._assert_no_injection('abc', TEXT_CSS)

- self._assert_no_injection('abc', APPLICATION)

- def test_empty_content_as_already_injected(self):

- self._assert_no_injection('', TEXT_HTML)

- def test_non_html_content_with_html_content_type(self):

- self._assert_no_injection('{"test": 1"}', TEXT_HTML)

- def test_already_injected(self):

- parameters = {'injection': SCRIPT_TO_INJECT}

- for template in NORMAL_TEMPLATES:

- for parameters['boundary_at_start'] in SEPARATORS_OR_NOT:

- for parameters['boundary_in_doctype'] in SEPARATORS_OR_NOT:

- for parameters['boundary_after_doctype'] in SEPARATORS_OR_NOT:

- for parameters['boundary_in_html'] in SEPARATORS_OR_NOT:

- for parameters['boundary_after_html'] in SEPARATORS_OR_NOT:

- for parameters['boundary_in_head'] in SEPARATORS_OR_NOT:

- for parameters['boundary_after_head'] in SEPARATORS_OR_NOT:

- source = template.format(**parameters)

- self._assert_no_injection(source, TEXT_HTML)

- def test_normal(self):

- parameters = {'injection': '{injection}'}

- for template in NORMAL_TEMPLATES:

- for parameters['boundary_at_start'] in SEPARATORS_OR_NOT:

- for parameters['boundary_in_doctype'] in SEPARATORS_OR_NOT:

- for parameters['boundary_after_doctype'] in SEPARATORS_OR_NOT:

- for parameters['boundary_in_html'] in SEPARATORS_OR_NOT:

- for parameters['boundary_after_html'] in SEPARATORS_OR_NOT:

- for parameters['boundary_in_head'] in SEPARATORS_OR_NOT:

- for parameters['boundary_after_head'] in SEPARATORS_OR_NOT:

- template = template.format(**parameters)

- self._assert_successful_injection(template)

- def test_comments(self):

- parameters = {'injection': '{injection}'}

- for parameters['comment_before_doctype'] in COMMENT_OR_NOT:

- for parameters['comment_after_doctype'] in COMMENT_OR_NOT:

- for parameters['comment_after_html'] in COMMENT_OR_NOT:

- template = TEMPLATE_COMMENT.format(**parameters)

- self._assert_successful_injection(template)

-if __name__ == '__main__':

- unittest.main()

« no previous file with comments | « telemetry/third_party/webpagereplay/script_injector.py ('k') | telemetry/third_party/webpagereplay/servermanager.py » ('j') | no next file with comments »