aCompetentBean/gazprea-fuzzer-python
2
2
Fork 0
Code Issues 2 Pull requests Packages Projects Releases Wiki Activity

Added tests

Browse source
This commit is contained in:
Ayrton 2023-11-20 18:52:14 -07:00
parent 04675a01d6
commit 05c8f92683
3 changed files with 28 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
ast_generator/ast_generator.py
View file

@ -2,7 +2,7 @@ import random
import string import string
import xml.etree.ElementTree as ET import xml.etree.ElementTree as ET
from english_words import get_english_words_set # from english_words import get_english_words_set
from constants import * from constants import *
@ -38,7 +38,8 @@ class AstGenerator:
self.symbol_table.append(global_scope) # NOTE for debug self.symbol_table.append(global_scope) # NOTE for debug
self.current_scope = global_scope self.current_scope = global_scope
names = get_english_words_set(['web2'], lower=True) # names = get_english_words_set(['web2'], lower=True)
names = random.choices(string.ascii_letters, k=self.settings['properties']['id-length']['max'])
possible_names = filter(lambda x: self.settings['properties']['id-length']['min'] < len(x) possible_names = filter(lambda x: self.settings['properties']['id-length']['min'] < len(x)
< self.settings['properties']['id-length']['max'], names) < self.settings['properties']['id-length']['max'], names)

2
ast_generator/test/test_ast_generator.py
View file

@ -267,7 +267,7 @@ class TestGeneration(unittest.TestCase):
def test_failing_assignment(self): def test_failing_assignment(self):
self.ast_gen.ast = ET.Element("block") self.ast_gen.ast = ET.Element("block")
self.ast_gen.current_ast_element = self.ast_gen.ast self.ast_gen.current_ast_element = self.ast_gen.ast
self.ast_gen.generate_main() # self.ast_gen.generate_main()
with self.assertRaises(ValueError): with self.assertRaises(ValueError):
self.ast_gen.generate_assignment() self.ast_gen.generate_assignment()

25
test/test_correctness.py
View file

@ -13,7 +13,7 @@ from ast_generator.ast_generator import AstGenerator
from ast_parser.python_unparser import PythonUnparser from ast_parser.python_unparser import PythonUnparser
class MyTestCase(unittest.TestCase): class TestCorrectness(unittest.TestCase):
@classmethod @classmethod
def setUpClass(cls): def setUpClass(cls):
with open("config.yaml", 'r') as stream: with open("config.yaml", 'r') as stream:
@ -128,7 +128,30 @@ class MyTestCase(unittest.TestCase):
print(self.python_unparser.source) print(self.python_unparser.source)
self.fail(e) self.fail(e)
def test_routine(self):
self.ast_gen.ast = ET.Element("block")
self.ast_gen.current_ast_element = self.ast_gen.ast
self.ast_gen.generate_routine()
self.python_unparser.xml = self.ast_gen.ast
self.python_unparser.unparse()
try:
exec(self.python_unparser.source)
except Exception as e:
print(self.python_unparser.source)
self.fail(e)
def test_program(self):
self.ast_gen.generate_ast()
self.python_unparser.xml = self.ast_gen.ast
self.python_unparser.unparse()
try:
exec(self.python_unparser.source)
except Exception as e:
print(self.python_unparser.source)
self.fail(e)
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()