|
import pytest |
|
|
|
from pytube import cipher |
|
from pytube.exceptions import RegexMatchError |
|
|
|
|
|
def test_map_functions(): |
|
with pytest.raises(RegexMatchError): |
|
cipher.map_functions("asdf") |
|
|
|
|
|
def test_get_initial_function_name_with_no_match_should_error(): |
|
with pytest.raises(RegexMatchError): |
|
cipher.get_initial_function_name("asdf") |
|
|
|
|
|
def test_get_transform_object_with_no_match_should_error(): |
|
with pytest.raises(RegexMatchError): |
|
cipher.get_transform_object("asdf", var="lt") |
|
|
|
|
|
def test_reverse(): |
|
reversed_array = cipher.reverse([1, 2, 3, 4], None) |
|
assert reversed_array == [4, 3, 2, 1] |
|
|
|
|
|
def test_splice(): |
|
assert cipher.splice([1, 2, 3, 4], 2) == [3, 4] |
|
assert cipher.splice([1, 2, 3, 4], 1) == [2, 3, 4] |
|
|
|
|
|
def test_throttling_reverse(): |
|
a = [1, 2, 3, 4] |
|
cipher.throttling_reverse(a) |
|
assert a == [4, 3, 2, 1] |
|
|
|
|
|
def test_throttling_push(): |
|
a = [1, 2, 3, 4] |
|
cipher.throttling_push(a, 5) |
|
assert a == [1, 2, 3, 4, 5] |
|
|
|
|
|
def test_throttling_unshift(): |
|
a = [1, 2, 3, 4] |
|
cipher.throttling_unshift(a, 2) |
|
assert a == [3, 4, 1, 2] |
|
|
|
|
|
def test_throttling_nested_splice(): |
|
a = [1, 2, 3, 4] |
|
cipher.throttling_nested_splice(a, 2) |
|
assert a == [3, 2, 1, 4] |
|
cipher.throttling_nested_splice(a, 0) |
|
assert a == [3, 2, 1, 4] |
|
|
|
|
|
def test_throttling_prepend(): |
|
a = [1, 2, 3, 4] |
|
cipher.throttling_prepend(a, 1) |
|
assert a == [4, 1, 2, 3] |
|
a = [1, 2, 3, 4] |
|
cipher.throttling_prepend(a, 2) |
|
assert a == [3, 4, 1, 2] |
|
|
|
|
|
def test_throttling_swap(): |
|
a = [1, 2, 3, 4] |
|
cipher.throttling_swap(a, 3) |
|
assert a == [4, 2, 3, 1] |
|
|
|
|
|
def test_js_splice(): |
|
mapping = { |
|
|
|
} |
|
for args, result in mapping.items(): |
|
a = [1, 2, 3, 4] |
|
assert cipher.js_splice(a, *args) == result |
|
|
|
|
|
def test_get_throttling_function_name(base_js): |
|
base_js_code_fragments = [ |
|
|
|
{ |
|
'raw_var' : r'var Apa=[hha]', |
|
'raw_code': r'a.url="";a.C&&(b=a.get("n"))&&(b=Apa[0](b),a.set("n",b),'\ |
|
r'Apa.length||hha(""))}};', |
|
'nfunc_name': 'hha' |
|
}, |
|
|
|
{ |
|
'raw_var' : r'var $x=[uq]', |
|
'raw_code': r'a.url="";a.D&&(b=a.get("n"))&&(b=$x[0](b),a.set("n",b),'\ |
|
r'$x.length||uq(""))', |
|
'nfunc_name': 'uq' |
|
} |
|
] |
|
for code_fragment, base_js_file in zip(base_js_code_fragments, base_js): |
|
assert code_fragment['raw_var'] in base_js_file |
|
assert code_fragment['raw_code'] in base_js_file |
|
func_name = cipher.get_throttling_function_name(base_js_file) |
|
assert func_name == code_fragment['nfunc_name'] |
|
|
