libnixf/test/Parse: add test for parsing op

nix-community · Feb 9, 2024 · 38ce0de · 38ce0de
1 parent 8655b97
commit 38ce0de
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/libnixf/test/Parse/ParseExpr.cpp b/libnixf/test/Parse/ParseExpr.cpp
@@ -1,6 +1,7 @@
 #include <gtest/gtest.h>
 
 #include "Parser.h"
+#include "nixf/Basic/Nodes/Op.h"
 
 namespace {
 
@@ -234,4 +235,22 @@ TEST(Parser, ExprDispatch_r_curly_id_ellipsis) {
   ASSERT_EQ(AST->kind(), Node::NK_ExprLambda);
 }
 
+TEST(Parser, ExprOp) {
+  auto Src = R"(1 + 2 * 3)"sv;
+
+  std::vector<Diagnostic> Diags;
+  Parser P(Src, Diags);
+  auto AST = P.parseExpr();
+
+  ASSERT_TRUE(AST);
+  ASSERT_EQ(AST->kind(), Node::NK_ExprBinOp);
+
+  auto &BinOp = *static_cast<ExprBinOp *>(AST.get());
+
+  ASSERT_TRUE(BinOp.lhs());
+  ASSERT_TRUE(BinOp.rhs());
+  ASSERT_EQ(BinOp.rhs()->kind(), Node::NK_ExprBinOp);
+  ASSERT_EQ(BinOp.lhs()->kind(), Node::NK_ExprInt);
+}
+
 } // namespace