def test_mod_call():
source_file = (
"DEF Foo()\n"
"Bar(foo, bar)\n"
"FooBar()\n"
"END"
"\n"
"DEF Bar(foo:IN, bar:IN)\n"
"END"
"\n"
"DEF FooBar()\n"
"END"
)
awaited_ast = [
SourceFile(name="Foo", statements=[
FunctionDefinition(name="Foo", body=[
FunctionCall(name="Bar", parameters=["foo", "bar"]),
FunctionCall(name="FooBar")
]),
FunctionDefinition(name="Bar", parameters=[
Parameter(name="foo", parameter_type=Parameter.TYPE.IN),
Parameter(name="bar", parameter_type=Parameter.TYPE.IN)
]),
FunctionDefinition(name="FooBar")])]
parser = Parser()
parser.add_source_file("Foo", source_file)
assert awaited_ast == parser.ast
def test_source_parsing():
source_file = (
"GLOBAL DEF Foo(bar:IN, foobar:OUT)\n"
"\n"
"END"
"\n"
"\n"
"DEF Bar()\n"
"\n"
"END"
)
awaited_ast = [
SourceFile(name="Foo", statements=[
FunctionDefinition(name="Foo", parameters=[
Parameter(name="bar", parameter_type=Parameter.TYPE.IN),
Parameter(name="foobar", parameter_type=Parameter.TYPE.OUT)
], is_global=True),
FunctionDefinition(name="Bar")
])]
parser = Parser()
parser.add_source_file("Foo", source_file)
assert awaited_ast == parser.ast
def test_module_parsing():
module_name = "Foo"
source_file = (
"GLOBAL DEF Foo(bar:IN, foobar:OUT)\n"
"\n"
"END"
"\n"
"\n"
"DEF Bar()\n"
"\n"
"END"
)
data_file = (
"DEFDAT Foo\n"
"\n"
"ENDDAT"
)
awaited_ast = [
Module(name=module_name,
source_file=SourceFile(name=module_name, statements=[
FunctionDefinition(name="Foo", parameters=[
Parameter(name="bar",
parameter_type=Parameter.TYPE.IN),
Parameter(name="foobar",
parameter_type=Parameter.TYPE.OUT)],
is_global=True),
FunctionDefinition(name="Bar")]),
data_file=DataFile(name=module_name,
statements=[DataDefinition(name="Foo")])
)]
parser = Parser()
parser.add_module(module_name, source_file, data_file)
assert awaited_ast == parser.ast
def test_file_attributes():
source_file = (
"&COMMENT Hello\n"
"DEF Foo()\n"
"END"
)
awaited_ast = [
SourceFile(name="Foo",
file_attributes=[FileAttribute(value="COMMENT Hello")],
statements=[FunctionDefinition(name="Foo")])]
parser = Parser()
parser.add_source_file("Foo", source_file)
assert awaited_ast == parser.ast
def test_simple_variable_declartion():
source_file = (
"DEF Foo()\n"
"DECL INT bar\n"
"END"
)
awaited_ast = [
SourceFile(name="Foo", statements=[
FunctionDefinition(name="Foo", body=[
VariableSymbol(name="bar", symbol_type="INT")])
])]
parser = Parser()
parser.add_source_file("Foo", source_file)
assert awaited_ast == parser.ast
def test_multi_dimensional_array_declartion():
source_file = (
"DEF Foo()\n"
"DECL INT bar[1, 2, 3]\n"
"END"
)
awaited_ast = [
SourceFile(name="Foo", statements=[
FunctionDefinition(name="Foo", body=[
VariableSymbol(name="bar", symbol_type="INT",
dimensions=[1, 2, 3])])
])]
parser = Parser()
parser.add_source_file("Foo", source_file)
assert awaited_ast == parser.ast
def test_source_file_with_single_function():
source_file = (
"DEFFCT INT Foo(bar:IN, foobar:OUT)\n"
"\n"
"ENDFCT"
)
awaited_ast = [
SourceFile(name="Foo", statements=[
FunctionDefinition(name="Foo", parameters=[
Parameter(name="bar", parameter_type=Parameter.TYPE.IN),
Parameter(name="foobar", parameter_type=Parameter.TYPE.OUT)
], returns=Type(name="INT"))])]
parser = Parser()
parser.add_source_file("Foo", source_file)
assert awaited_ast == parser.ast
def test_chained_array_declartion():
source_file = (
"DEF Foo()\n"
"DECL INT bar1[1, 1], bar2[10]\n"
"END"
)
awaited_ast = [
SourceFile(name="Foo", statements=[
FunctionDefinition(name="Foo", body=[
VariableSymbol(name="bar1", symbol_type="INT",
dimensions=[1, 1]),
VariableSymbol(name="bar2", symbol_type="INT",
dimensions=[10])])])]
parser = Parser()
parser.add_source_file("Foo", source_file)
assert awaited_ast == parser.ast
def test_declarations_and_statements():
source_file = (
"DEF Foo()\n"
"DECL INT bar\n"
"\n"
"FooBar()"
"END"
)
awaited_ast = [
SourceFile(name="Foo", statements=[
FunctionDefinition(name="Foo", body=[
VariableSymbol(name="bar", symbol_type="INT",
dimensions=[0]),
FunctionCall(name="FooBar")])])]
parser = Parser()
parser.add_source_file("Foo", source_file)
assert awaited_ast == parser.ast
def test_simple_module_definition():
module_name = "Foo"
source_file = (
"GLOBAL DEF Foo(bar:IN, foobar:OUT)\n"
"\n"
"END"
"\n"
"\n"
"DEF Bar()\n"
"\n"
"END"
)
data_file = (
"DEFDAT Foo\n"
"\n"
"ENDDAT"
)
global_symbol_table = SymbolTable(name="GLOBAL", parent_scope=None)
module_symbol_table = SymbolTable(name="Foo",
parent_scope=global_symbol_table)
function1_symbol_table = SymbolTable(name="Foo.Foo",
parent_scope=module_symbol_table)
function2_symbol_table = SymbolTable(name="Foo.Bar",
parent_scope=module_symbol_table)
function_definition1 = FunctionDefinition(
name="Foo",
parameters=[Parameter(name="bar",
parameter_type=Parameter.TYPE.IN),
Parameter(name="foobar",
parameter_type=Parameter.TYPE.OUT)],
is_global=True,
symbol_table=function1_symbol_table)
function_definition2 = FunctionDefinition(
name="Bar", symbol_table=function2_symbol_table)
awaited_ast = [
Module(name=module_name,
source_file=SourceFile(
name=module_name,
statements=[function_definition1, function_definition2],
symbol_table=module_symbol_table),
data_file=DataFile(
name=module_name,
statements=[DataDefinition(
name="Foo",
symbol_table=module_symbol_table)],
symbol_table=module_symbol_table))]
module_symbol_table.add(
FunctionSymbol.create_from_definition(function_definition1))
module_symbol_table.add(
FunctionSymbol.create_from_definition(function_definition2))
parser = Parser()
parser.add_module(module_name, source_file, data_file)
analyzer = SemanticAnalyzer()
analyzer.visit(parser.ast)
assert awaited_ast == parser.ast