rs_bindings_from_cc/ir_test.cc

// Part of the Crubit 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 "rs_bindings_from_cc/ir.h"

#include <string>

#include "testing/base/public/gunit.h"
#include "third_party/json/src/json.hpp"

namespace rs_bindings_from_cc {

namespace {

TEST(IrTest, TypeToJson) {
  nlohmann::json expected = nlohmann::json::parse(R"j({
      "rs_name": "CompoundRs",
      "cc_name": "CompoundCc",
      "cc_const": false,
      "type_params": [
          { "rs_name": "i32", "cc_name": "int", "cc_const": false, "type_params": []}
      ]
  })j");
  auto type = Type{.rs_name = "CompoundRs",
                   .cc_name = "CompoundCc",
                   .type_params = {Type{"i32", "int"}}};
  EXPECT_EQ(type.ToJson(), expected);
}

TEST(IrTest, IR) {
  nlohmann::json expected = nlohmann::json::parse(
      R"j({
            "used_headers": [{ "name": "foo/bar.h" }],
            "functions": [{
              "identifier": { "identifier": "hello_world" },
              "mangled_name": "#$mangled_name$#",
              "return_type": { "rs_name": "i32", "cc_name": "int", "cc_const": false, "type_params": [] },
              "params": [
                {
                  "identifier": {"identifier": "arg" },
                  "type": { "rs_name": "i32", "cc_name": "int", "cc_const": false, "type_params": [] }
                }
              ],
              "is_inline": false
            }],
            "records": [
              {
                "identifier": {"identifier": "SomeStruct" },
                "fields": [
                  {
                    "identifier": {"identifier": "public_int" },
                    "type": {"rs_name": "i32", "cc_name": "int", "cc_const": false, "type_params": [] },
                    "access": "Public",
                    "offset": 0
                  },
                  {
                    "identifier": {"identifier": "protected_int" },
                    "type": {"rs_name": "i32", "cc_name": "int", "cc_const": false, "type_params": [] },
                    "access": "Protected",
                    "offset": 32
                  },
                  {
                    "identifier": {"identifier": "private_int" },
                    "type": {"rs_name": "i32", "cc_name": "int", "cc_const": false, "type_params": [] },
                    "access": "Private",
                    "offset": 64
                  }
                ],
                "size": 12,
                "alignment": 4
              }
            ]
      })j");
  IR
      ir =
          {
              .used_headers = {HeaderName("foo/bar.h")},
              .functions = {Func{
                  .identifier = Identifier("hello_world"),
                  .mangled_name = "#$mangled_name$#",
                  .return_type = Type{"i32", "int"},
                  .params = {FuncParam{Type{"i32", "int"}, Identifier("arg")}},
                  .is_inline = false}},
              .records = {Record{.identifier = Identifier("SomeStruct"),
                                 .fields =
                                     {
                                         Field{.identifier =
                                                   Identifier("public_int"),
                                               .type = Type{"i32", "int"},
                                               .access = kPublic,
                                               .offset = 0},
                                         Field{.identifier = Identifier(
                                                   "protected_int"),
                                               .type = Type{"i32", "int"},
                                               .access = kProtected,
                                               .offset = 32},
                                         Field{.identifier = Identifier(
                                                   "private_int"),
                                               .type = Type{"i32", "int"},
                                               .access = kPrivate,
                                               .offset = 64},
                                     },
                                 .size = 12,
                                 .alignment = 4}}};
  EXPECT_EQ(ir.ToJson(), expected);
}

}  // namespace
}  // namespace rs_bindings_from_cc