python-aternos/tests/js2py_test.py

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