mirror of
https://gitlab.gnome.org/jwestman/blueprint-compiler.git
synced 2025-07-07 01:29:26 -04:00
61acfbda98
Add a FUZZ_LEVEL environment variable that can be used to test the language server features under the fuzzer in addition to the compiler. It's not enabled by default, but can be useful to run locally.
59 lines
1.4 KiB
Python
59 lines
1.4 KiB
Python
import os
|
|
import sys
|
|
|
|
from pythonfuzz.main import PythonFuzz
|
|
|
|
from blueprintcompiler.outputs.xml import XmlOutput
|
|
|
|
sys.path.insert(0, os.path.dirname(os.path.dirname(__file__)))
|
|
|
|
from blueprintcompiler import gir, parser, tokenizer
|
|
from blueprintcompiler.completions import complete
|
|
from blueprintcompiler.errors import (
|
|
CompilerBugError,
|
|
PrintableError,
|
|
)
|
|
from blueprintcompiler.lsp import LanguageServer
|
|
|
|
fuzz_level = int(os.getenv("FUZZ_LEVEL") or "0")
|
|
|
|
|
|
@PythonFuzz
|
|
def fuzz(buf: bytes):
|
|
try:
|
|
blueprint = buf.decode("ascii")
|
|
|
|
tokens = tokenizer.tokenize(blueprint)
|
|
ast, errors, warnings = parser.parse(tokens)
|
|
|
|
if fuzz_level >= 1:
|
|
assert_ast_doesnt_crash(blueprint, tokens, ast)
|
|
|
|
xml = XmlOutput()
|
|
if errors is None and ast is not None:
|
|
xml.emit(ast)
|
|
except CompilerBugError as e:
|
|
raise e
|
|
except PrintableError:
|
|
pass
|
|
except UnicodeDecodeError:
|
|
pass
|
|
|
|
|
|
def assert_ast_doesnt_crash(text, tokens, ast):
|
|
lsp = LanguageServer()
|
|
for i in range(len(text) + 1):
|
|
ast.get_docs(i)
|
|
for i in range(len(text) + 1):
|
|
list(complete(lsp, ast, tokens, i))
|
|
for i in range(len(text) + 1):
|
|
ast.get_reference(i)
|
|
ast.get_document_symbols()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Make sure Gtk 4.0 is accessible, otherwise every test will fail on that
|
|
# and nothing interesting will be tested
|
|
gir.get_namespace("Gtk", "4.0")
|
|
|
|
fuzz()
|