Switch SemanticsIR dumps to produce YAML (#2517)

The parser and lexer already produce YAML, so this is fundamentally a consistency issue. I've been thinking about this, and was looking again because I'm working on adding callables, and figured I'd just fix it now.
2026-10-06 07:14:42 +01:00 · 2023-01-18 17:43:17 -08:00
parent 0d279b388a
commit a1f2d6341f
26 changed files with 755 additions and 665 deletions
@@ -0,0 +1,76 @@
+// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
+// Exceptions. See /LICENSE for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
+#include "toolchain/semantics/semantics_ir.h"
+
+#include <gmock/gmock.h>
+#include <gtest/gtest.h>
+
+#include <forward_list>
+
+#include "gmock/gmock.h"
+#include "gtest/gtest.h"
+#include "toolchain/common/yaml_test_helpers.h"
+#include "toolchain/diagnostics/diagnostic_emitter.h"
+#include "toolchain/lexer/tokenized_buffer.h"
+
+namespace Carbon::Testing {
+namespace {
+
+using ::testing::_;
+using ::testing::AllOf;
+using ::testing::Contains;
+using ::testing::Each;
+using ::testing::ElementsAre;
+using ::testing::MatchesRegex;
+using ::testing::Pair;
+
+TEST(SemanticsIRTest, YAML) {
+  DiagnosticConsumer& consumer = ConsoleDiagnosticConsumer();
+  llvm::Expected<SourceBuffer> source =
+      SourceBuffer::CreateFromText("var x: i32 = 0;");
+  TokenizedBuffer tokens = TokenizedBuffer::Lex(*source, consumer);
+  ParseTree parse_tree =
+      ParseTree::Parse(tokens, consumer, /*vlog_stream=*/nullptr);
+  SemanticsIR builtin_ir = SemanticsIR::MakeBuiltinIR();
+  SemanticsIR semantics_ir = SemanticsIR::MakeFromParseTree(
+      builtin_ir, tokens, parse_tree, consumer, /*vlog_stream=*/nullptr);
+
+  std::string print_output;
+  llvm::raw_string_ostream print_stream(print_output);
+  semantics_ir.Print(print_stream);
+  print_stream.flush();
+
+  // Matches the ID of a node. The numbers may change because of builtin
+  // cross-references, so this code is only doing loose structural checks.
+  auto node_id = Yaml::Scalar(MatchesRegex("node\\d+"));
+
+  EXPECT_THAT(
+      Yaml::Value::FromText(print_output),
+      ElementsAre(Yaml::Mapping(ElementsAre(
+          Pair("cross_reference_irs_size", "1"),
+          Pair("integer_literals", Yaml::Sequence(ElementsAre("0"))),
+          Pair("strings", Yaml::Sequence(ElementsAre("x"))),
+          Pair("nodes",
+               Yaml::Sequence(AllOf(
+                   // kind is required, other parts are optional.
+                   Each(Yaml::Mapping(Contains(Pair("kind", _)))),
+                   // A 0-arg node.
+                   Contains(Yaml::Mapping(ElementsAre(
+                       Pair("kind", "VarStorage"), Pair("type", node_id)))),
+                   // A 1-arg node.
+                   Contains(Yaml::Mapping(ElementsAre(
+                       Pair("kind", "IntegerLiteral"), Pair("arg0", "int0"),
+                       Pair("type", node_id)))),
+                   // A 2-arg node.
+                   Contains(Yaml::Mapping(ElementsAre(
+                       Pair("kind", "BindName"), Pair("arg0", "str0"),
+                       Pair("arg1", node_id), Pair("type", node_id))))))),
+          // This production has only one node block.
+          Pair("node_blocks",
+               Yaml::Sequence(ElementsAre(Yaml::Sequence(Each(node_id)))))))));
+}
+
+}  // namespace
+}  // namespace Carbon::Testing