Will Hawkins
2f7d05a3fd
Compilation interface(s) now contain a compilation context. The change will make it easier to expand on what each step of the compilation process needs to know to complete its task without having to make major changes to the interface. Signed-off-by: Will Hawkins <hawkinsw@obs.cr>
119 lines
3.4 KiB
Swift
119 lines
3.4 KiB
Swift
// p4rse, Copyright 2026, Will Hawkins
|
|
//
|
|
// This file is part of p4rse.
|
|
//
|
|
// This file is free software: you can redistribute it and/or modify
|
|
// it under the terms of the GNU General Public License as published by
|
|
// the Free Software Foundation, either version 3 of the License, or
|
|
// (at your option) any later version.
|
|
//
|
|
// This program is distributed in the hope that it will be useful,
|
|
// but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License
|
|
// along with this program. If not, see <https://www.gnu.org/licenses/>.
|
|
|
|
import Common
|
|
import Foundation
|
|
import Macros
|
|
import Runtime
|
|
import SwiftTreeSitter
|
|
import Testing
|
|
import TreeSitter
|
|
import TreeSitterP4
|
|
import P4Lang
|
|
|
|
@testable import P4Compiler
|
|
|
|
@Test func test_simple_parser_syntax_error() async throws {
|
|
let simple_parser_declaration = """
|
|
parser main_parser() {
|
|
state
|
|
transition start;
|
|
}
|
|
};
|
|
"""
|
|
#expect(
|
|
#RequireErrorResult(
|
|
Error(withMessage: "Could not compile the P4 program"),
|
|
Program.Compile(simple_parser_declaration)))
|
|
}
|
|
|
|
@Test func test_simple_parser_with_statement() async throws {
|
|
let simple_parser_declaration = """
|
|
parser main_parser() {
|
|
state start {
|
|
true;
|
|
transition start;
|
|
}
|
|
};
|
|
"""
|
|
|
|
let program = try #UseOkResult(Program.Compile(simple_parser_declaration))
|
|
let parser = try #UseOkResult(program.find_parser(withName: Identifier(name: "main_parser")))
|
|
|
|
#expect(parser.states.count() == 1)
|
|
|
|
let state = try! #require(parser.states.find(withIdentifier: Identifier(name: "start")))
|
|
#expect(state.state == Identifier(name: "start"))
|
|
#expect(state.statements.count == 1)
|
|
}
|
|
|
|
@Test func test_simple_parser_with_instantiation() async throws {
|
|
let simple_parser_declaration = """
|
|
parser main_parser() {
|
|
state start {
|
|
true;
|
|
false;
|
|
transition start;
|
|
}
|
|
};
|
|
bool() main;
|
|
"""
|
|
|
|
let program = try #UseOkResult(Program.Compile(simple_parser_declaration))
|
|
#expect(#RequireOkResult(program.find_parser(withName: Identifier(name: "main_parser"))))
|
|
}
|
|
|
|
@Test func test_invalid_transition_expression_keyset_expressions() async throws {
|
|
let simple_parser_declaration = """
|
|
parser main_parser() {
|
|
state start {
|
|
transition select (false) {
|
|
asdf: reject;
|
|
asde: reject;
|
|
};
|
|
}
|
|
};
|
|
"""
|
|
|
|
let compilation_error = try #UseErrorResult(Program.Compile(simple_parser_declaration))
|
|
|
|
#expect(compilation_error.msg.contains("asde"))
|
|
#expect(compilation_error.msg.contains("asdf"))
|
|
}
|
|
|
|
@Test func test_simple_parser_macro_nodetype_test() async throws {
|
|
let simple = """
|
|
parser main_parser() {
|
|
state start {
|
|
transition select (false) {
|
|
asdf: reject;
|
|
asde: reject;
|
|
};
|
|
}
|
|
};
|
|
"""
|
|
|
|
let p = try! #UseOkResult(ConfigureP4Parser())
|
|
let result = try! #require(p.parse(simple))
|
|
|
|
#expect(
|
|
#RequireErrorResult<(EvaluatableStatement, CompilerContext)>(
|
|
Error(withMessage: "{2, 154}: Did not find assignment statement"),
|
|
ParserAssignmentStatement.Compile(
|
|
node: result.rootNode!, withContext: CompilerContext(withNames: LexicalScopes()))))
|
|
}
|