#include <gtest/gtest.h>

#include "oi/SymbolService.h"
#include "oi/type_graph/AddChildren.h"
#include "oi/type_graph/NodeTracker.h"
#include "oi/type_graph/Printer.h"
#include "oi/type_graph/TypeGraph.h"
#include "test_drgn_parser.h"

using namespace type_graph;

class AddChildrenTest : public DrgnParserTest {
 protected:
  std::string run(std::string_view function,
                  DrgnParserOptions options) override;
};

std::string AddChildrenTest::run(std::string_view function,
                                 DrgnParserOptions options) {
  TypeGraph typeGraph;
  auto drgnParser = getDrgnParser(typeGraph, options);
  auto* drgnRoot = getDrgnRoot(function);

  Type& type = drgnParser.parse(drgnRoot);
  typeGraph.addRoot(type);
  NodeTracker tracker;

  auto pass = AddChildren::createPass(drgnParser, *symbols_);
  pass.run(typeGraph, tracker);

  std::stringstream out;
  Printer printer{out, tracker, typeGraph.size()};
  printer.print(type);

  return out.str();
}

TEST_F(AddChildrenTest, SimpleStruct) {
  // Should do nothing
  test("oid_test_case_simple_struct", R"(
[1] Pointer
[0]   Struct: SimpleStruct (size: 16)
        Member: a (offset: 0)
          Primitive: int32_t
        Member: b (offset: 4)
          Primitive: int8_t
        Member: c (offset: 8)
          Primitive: int64_t
)");
}

TEST_F(AddChildrenTest, InheritanceStatic) {
  // Should do nothing
  test("oid_test_case_inheritance_access_public", R"(
[4] Pointer
[0]   Class: Public (size: 8)
        Parent (offset: 0)
[1]       Class: Base (size: 4)
            Member: base_int (offset: 0)
[3]           Typedef: int32_t
[2]             Typedef: __int32_t
                  Primitive: int32_t
        Member: public_int (offset: 4)
          [3]
)");
}

TEST_F(AddChildrenTest, InheritancePolymorphic) {
  testMultiCompiler("oid_test_case_inheritance_polymorphic_a_as_a", R"(
[1]  Pointer
[0]    Class: A (size: 16)
         Member: _vptr$A (offset: 0)
           Primitive: StubbedPointer
         Member: int_a (offset: 8)
           Primitive: int32_t
         Function: ~A (virtual)
         Function: myfunc (virtual)
         Function: A
         Function: A
         Child
[8]        Class: B (size: 40)
             Parent (offset: 0)
               [0]
             Member: vec_b (offset: 16)
[4]            Container: std::vector (size: 24)
                 Param
                   Primitive: int32_t
                 Param
[5]                Class: allocator<int> (size: 1)
                     Param
                       Primitive: int32_t
                     Parent (offset: 0)
[7]                    Typedef: __allocator_base<int>
[6]                      Class: new_allocator<int> (size: 1)
                           Param
                             Primitive: int32_t
                           Function: new_allocator
                           Function: new_allocator
                           Function: allocate
                           Function: deallocate
                           Function: _M_max_size
                     Function: allocator
                     Function: allocator
                     Function: operator=
                     Function: ~allocator
                     Function: allocate
                     Function: deallocate
             Function: ~B (virtual)
             Function: myfunc (virtual)
             Function: B
             Function: B
             Child
[10]           Class: C (size: 48)
                 Parent (offset: 0)
                   [8]
                 Member: int_c (offset: 40)
                   Primitive: int32_t
                 Function: ~C (virtual)
                 Function: myfunc (virtual)
                 Function: C
                 Function: C
)",
                    R"(
[1]  Pointer
[0]    Class: A (size: 16)
         Member: _vptr.A (offset: 0)
           Primitive: StubbedPointer
         Member: int_a (offset: 8)
           Primitive: int32_t
         Function: operator=
         Function: A
         Function: A
         Function: ~A (virtual)
         Function: myfunc (virtual)
         Child
[7]        Class: B (size: 40)
             Parent (offset: 0)
               [0]
             Member: vec_b (offset: 16)
[4]            Container: std::vector (size: 24)
                 Param
                   Primitive: int32_t
                 Param
[5]                Class: allocator<int> (size: 1)
                     Parent (offset: 0)
[6]                    Class: new_allocator<int> (size: 1)
                         Param
                           Primitive: int32_t
                         Function: new_allocator
                         Function: new_allocator
                         Function: allocate
                         Function: deallocate
                         Function: _M_max_size
                     Function: allocator
                     Function: allocator
                     Function: operator=
                     Function: ~allocator
                     Function: allocate
                     Function: deallocate
             Function: operator=
             Function: B
             Function: B
             Function: ~B (virtual)
             Function: myfunc (virtual)
             Child
[9]            Class: C (size: 48)
                 Parent (offset: 0)
                   [7]
                 Member: int_c (offset: 40)
                   Primitive: int32_t
                 Function: operator=
                 Function: C
                 Function: C
                 Function: ~C (virtual)
                 Function: myfunc (virtual)
)");
}