object-introspection/test/test_prune.cpp

#include <gtest/gtest.h>

#include "oi/type_graph/Prune.h"
#include "test/type_graph_utils.h"

using type_graph::Prune;

TEST(PruneTest, PruneClass) {
  test(Prune::createPass(), R"(
[0] Class: MyClass (size: 8)
      Param
        Primitive: int32_t
      Param
        Value: "123"
      Parent (offset: 0)
[1]     Class: MyParent (size: 4)
          Member: a (offset: 0)
            Primitive: int32_t
      Member: a (offset: 0)
        Primitive: int32_t
      Member: b (offset: 4)
        Primitive: int32_t
      Function: foo
      Function: bar
)",
       R"(
[0] Class: MyClass (size: 8)
      Member: a (offset: 0)
        Primitive: int32_t
      Member: b (offset: 4)
        Primitive: int32_t
)");
}

TEST(PruneTest, RecurseClassMember) {
  test(Prune::createPass(), R"(
[0] Class: MyClass (size: 0)
      Member: xxx (offset: 0)
[1]     Class: ClassA (size: 12)
          Function: foo
)",
       R"(
[0] Class: MyClass (size: 0)
      Member: xxx (offset: 0)
[1]     Class: ClassA (size: 12)
)");
}

TEST(PruneTest, RecurseClassChild) {
  test(Prune::createPass(), R"(
[0] Class: MyClass (size: 0)
      Child
[1]     Class: ClassA (size: 12)
          Function: foo
)",
       R"(
[0] Class: MyClass (size: 0)
      Child
[1]     Class: ClassA (size: 12)
)");
}