blob: 7dca4db2f95fc9d52767a764fa4c294ee9c6b6e7 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
|
# SPDX-License-Identifier: CC0-1.0
"""
Haketilo unit tests - exposing some special functionalities to injected scripts
"""
# This file is part of Haketilo
#
# Copyright (C) 2022 Wojtek Kosior <koszko@koszko.org>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the CC0 1.0 Universal License as published by
# the Creative Commons Corporation.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# CC0 1.0 Universal License for more details.
import pytest
import json
from selenium.webdriver.support.ui import WebDriverWait
from ..script_loader import load_script
from ..world_wide_library import some_data
def content_script():
return load_script('content/haketilo_apis.js') + ';\nstart();'
def background_script():
return load_script('background/CORS_bypass_server.js') + ';\nstart();'
@pytest.mark.ext_data({
'content_script': content_script,
'background_script': background_script
})
@pytest.mark.usefixtures('webextension')
def test_haketilo_apis_CORS_bypass(driver):
"""
Verify injected scripts will be able to bypass CORS with the help of
Haketilo API.
"""
driver.get('https://gotmyowndoma.in/')
driver.execute_script(
'''
const fetch_arg = {
url: "https://anotherdoma.in/resource/blocked/by/CORS.json",
init: {}
};
const detail = {
data: JSON.stringify(fetch_arg),
id: "abcdef"
};
window.addEventListener("haketilo_CORS_bypass-abcdef",
e => window.__response = e.detail);
window.dispatchEvent(new CustomEvent("haketilo_CORS_bypass", {detail}));
''')
get_response = lambda d: d.execute_script("return window.__response;")
response = WebDriverWait(driver, 10).until(get_response)
response = json.loads(response)
assert response['body'] == some_data.encode().hex()
assert response['status'] == 200
assert type(response['headers']) is list
|