Begin Implementation of Binary Operator Support

Signed-off-by: Will Hawkins <hawkinsw@obs.cr>

Tests/p4rseTests/RuntimeTests.swift

@@ -197,3 +197,101 @@ import TreeSitterP4
       Error(withMessage: "No key matched the selector"),
       runtime.run()))
 }
+
+@Test func test_simple_parser_binary_operator_equal() async throws {
+  let simple = """
+    parser main_parser() {
+       state start {
+           transition select (true == true) {
+              true: accept;
+              false: reject;
+           };
+       }
+    };
+  """
+
+  let program = try #UseOkResult(Program.Compile(simple))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  #expect(state_result == P4Lang.accept)
+}
+
+@Test func test_simple_parser_binary_operator_equal_not_equal() async throws {
+  let simple = """
+    parser main_parser() {
+       state start {
+           transition select (true == false) {
+              true: accept;
+              false: reject;
+           };
+       }
+    };
+  """
+
+  let program = try #UseOkResult(Program.Compile(simple))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  #expect(state_result == P4Lang.reject)
+}
+
+@Test func test_simple_parser_binary_operator_equal_integer() async throws {
+  let simple = """
+    parser main_parser() {
+       state start {
+           transition select (5 == 5) {
+              true: accept;
+              false: reject;
+           };
+       }
+    };
+  """
+
+  let program = try #UseOkResult(Program.Compile(simple))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  #expect(state_result == P4Lang.accept)
+}
+
+@Test func test_simple_parser_binary_operator_equal_not_equal_integer() async throws {
+  let simple = """
+    parser main_parser() {
+       state start {
+           transition select (5 == 6) {
+              true: accept;
+              false: reject;
+           };
+       }
+    };
+  """
+
+  let program = try #UseOkResult(Program.Compile(simple))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  #expect(state_result == P4Lang.reject)
+}
+
+@Test func test_simple_parser_binary_operator_equal_invalid_types() async throws {
+  let simple = """
+    parser main_parser() {
+       state start {
+           transition select (5 == true) {
+              true: accept;
+              false: reject;
+           };
+       }
+    };
+  """
+
+  let program = try #UseOkResult(Program.Compile(simple))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+
+  // TODO: This test should throw an error.
+
+  #expect(state_result == P4Lang.reject)
+}

Tests/p4rseTests/ValueTypeParserTests.swift

@@ -19,6 +19,7 @@ import Common
 import Foundation
 import Macros
 import P4Runtime
+import P4Lang
 import SwiftTreeSitter
 import Testing
 import TreeSitter
@@ -116,3 +117,115 @@ import TreeSitterP4
       ),
       Program.Compile(simple_parser_declaration)))
 }
+
+@Test func test_expression_in_declaration_initializer() async throws {
+  let simple_parser_declaration = """
+      parser main_parser() {
+        state start {
+          bool where_to = 5 == 5 == true;
+          transition select (where_to) {
+            true: accept;
+            false: reject;
+          };
+        }
+      };
+    """
+  let program = try #UseOkResult(Program.Compile(simple_parser_declaration))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  // 5 == 5 == true
+  // true == true
+  // true
+  #expect(state_result == P4Lang.accept)
+}
+
+@Test func test_expression_in_declaration_initializer2() async throws {
+  let simple_parser_declaration = """
+      parser main_parser() {
+        state start {
+          bool where_to = 5 == 5 == false;
+          transition select (where_to) {
+            true: accept;
+            false: reject;
+          };
+        }
+      };
+    """
+  let program = try #UseOkResult(Program.Compile(simple_parser_declaration))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  // 5 == 5 == true
+  // true == false
+  // false
+  #expect(state_result == P4Lang.reject)
+}
+
+@Test func test_expression_in_declaration_initializer_false() async throws {
+  let simple_parser_declaration = """
+      parser main_parser() {
+        state start {
+          bool where_to = 6 == 5 == true;
+          transition select (where_to) {
+            true: accept;
+            false: reject;
+          };
+        }
+      };
+    """
+  let program = try #UseOkResult(Program.Compile(simple_parser_declaration))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  // 6 == 5 == true
+  // false == true
+  // false
+  #expect(state_result == P4Lang.reject)
+}
+
+@Test func test_expression_in_declaration_initializer_false2() async throws {
+  let simple_parser_declaration = """
+      parser main_parser() {
+        state start {
+          bool where_to = 6 == 5 == false;
+          transition select (where_to) {
+            true: accept;
+            false: reject;
+          };
+        }
+      };
+    """
+  let program = try #UseOkResult(Program.Compile(simple_parser_declaration))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  // 6 == 5 == false
+  // false == false
+  // true
+  #expect(state_result == P4Lang.accept)
+}
+
+@Test func test_expression_in_declaration_initializer_invalid_types() async throws {
+  let simple_parser_declaration = """
+      parser main_parser() {
+        state start {
+          bool where_to = false == 5 == true;
+          transition select (where_to) {
+            true: accept;
+            false: reject;
+          };
+        }
+      };
+    """
+  let program = try #UseOkResult(Program.Compile(simple_parser_declaration))
+  let runtime = try #UseOkResult(P4Runtime.ParserRuntime.create(program: program))
+  let (state_result, _) = try! #UseOkResult(runtime.run())
+
+  // TODO: This test should throw an error.
+
+  // false == 5 == true
+  // false == true
+  // false
+  #expect(state_result == P4Lang.reject)
+}