80 lines
2.1 KiB
Python
80 lines
2.1 KiB
Python
import re
|
|
import unittest
|
|
|
|
from python_aternos import atjsparse
|
|
|
|
class TestJs2Py(unittest.TestCase):
|
|
|
|
def setUp(self) -> None:
|
|
|
|
self.tests = []
|
|
with open('token.txt', 'rt') as f:
|
|
lines = re.split(r'[\r\n]', f.read())
|
|
del lines[len(lines)-1] # Remove empty string
|
|
self.tests = lines
|
|
|
|
self.results = [
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2rKOA1IFdBcHhEM616cb',
|
|
'2iXh5W5uEYq5fWJIazQ6',
|
|
'CuUcmZ27Fb8bVBNw12Vj',
|
|
'YPPe8Ph7vzYaZ9PF9oQP',
|
|
'UfLlemvKEE16ltk0hZNM',
|
|
'S1Oban9UGRXVIepREw9q',
|
|
'S1Oban9UGRXVIepREw9q',
|
|
'KYDDyT1DWOJTZpNtJWhM',
|
|
'lZPFwRqIGIf8JKk1LG02',
|
|
'KbxzYCJUrFjWzbeZcAmE',
|
|
'KbxzYCJUrFjWzbeZcAmE'
|
|
]
|
|
|
|
def test_base64(self) -> None:
|
|
|
|
encoded = 'QEhlbGxvIFdvcmxkIQ=='
|
|
decoded = atjsparse.atob(encoded)
|
|
self.assertEqual(decoded, '@Hello World!')
|
|
|
|
def test_conv(self) -> None:
|
|
|
|
token = '(() => {window["AJAX_TOKEN"]=("2r" + "KO" + "A1" + "IFdBcHhEM" + "61" + "6cb");})();'
|
|
f = atjsparse.to_ecma5_function(token)
|
|
self.assertEqual(f, '(function(){window["AJAX_TOKEN"]=("2r" + "KO" + "A1" + "IFdBcHhEM" + "61" + "6cb");})()')
|
|
|
|
def test_ecma6parse(self) -> None:
|
|
|
|
code = '''
|
|
window.t0 =
|
|
window['document']&&
|
|
!window[["p","Ma"].reverse().join('')]||
|
|
!window[["ut","meo","i","etT","s"].reverse().join('')];'''
|
|
|
|
part1 = '''window.t1 = Boolean(window['document']);'''
|
|
part2 = '''window.t2 = Boolean(!window[["p","Ma"].reverse().join('')]);'''
|
|
part3 = '''window.t3 = Boolean(!window[["ut","meo","i","etT","s"].reverse().join('')]);'''
|
|
|
|
ctx0 = atjsparse.exec(code)
|
|
ctx1 = atjsparse.exec(part1)
|
|
ctx2 = atjsparse.exec(part2)
|
|
ctx3 = atjsparse.exec(part3)
|
|
|
|
self.assertEqual(ctx1.window['t1'], True)
|
|
self.assertEqual(ctx2.window['t2'], False)
|
|
self.assertEqual(ctx3.window['t3'], False)
|
|
|
|
def test_exec(self) -> None:
|
|
|
|
for i, f in enumerate(self.tests):
|
|
ctx = atjsparse.exec(f)
|
|
res = ctx.window['AJAX_TOKEN']
|
|
self.assertEqual(res, self.results[i])
|
|
|
|
def tearDown(self) -> None:
|
|
del self.tests
|
|
del self.results
|