autowired_v1_test.cc

#include "gtest/gtest.h"
#include "snapshot/snapshot.h"

#include "autowired/autowired.h"

namespace auto_wired::v1::test {

static int test_flag = 0;

class A {
public:
    int a{1};
};

class B {
public:
    AUTO_WIRED_DECLARE_BEGIN(B,
                             auto_wired::AutoWired::RegisterOptions::WithNeedInit(),
                             auto_wired::AutoWired::RegisterOptions::WithNeedDeInit())
    AUTO_WIRED_DECLARE_CLASS(a)
    AUTO_WIRED_DECLARE_END()

    void Init() {
        if (test_flag == 0) {
            test_flag += a.a;
        }
    }

    void DeInit() {
        if (test_flag == 5) {
            test_flag = 55;
        }
    }

    int b{2};

private:
    A& a;
};

class BB {
public:
    AUTO_WIRED_DECLARE_BEGIN(BB, auto_wired::AutoWired::RegisterOptions::WithOrder(-1))
    AUTO_WIRED_DECLARE_CLASS(a)
    AUTO_WIRED_DECLARE_END()

private:
    A& a;
};

class BBB {
public:
    AUTO_WIRED_DECLARE_BEGIN(BBB, auto_wired::AutoWired::RegisterOptions::WithOrder(999))
    AUTO_WIRED_DECLARE_CLASS(a)
    AUTO_WIRED_DECLARE_END()

private:
    A& a;
};

class C {
public:
    AUTO_WIRED_DECLARE_BEGIN(C,
                             auto_wired::AutoWired::RegisterOptions::WithNeedInit(),
                             auto_wired::AutoWired::RegisterOptions::WithNeedDeInit())
    AUTO_WIRED_DECLARE_CLASS(a)
    AUTO_WIRED_DECLARE_CLASS(b)
    AUTO_WIRED_DECLARE_CLASS(bb)
    AUTO_WIRED_DECLARE_CLASS(bbb)
    AUTO_WIRED_DECLARE_END()

    void Init() {
        if (test_flag == 1) {
            test_flag += a.a;
            test_flag += b.b;
        }
    }

    void DeInit() {
        test_flag = 5;
    }

    int c{3};

private:
    A& a;
    B& b;
    BB& bb;
    BBB& bbb;
};

class D {
public:
    AUTO_WIRED_DECLARE_BEGIN(D)
    AUTO_WIRED_DECLARE_CLASS(c)
    AUTO_WIRED_DECLARE_END()

private:
    C& c;
};

class AutoWiredV1Test : public testing::Test {
protected:
    virtual void SetUp() override {}
};

TEST_F(AutoWiredV1Test, auto_wired_v1_test) {
    test_flag = 0;
    auto w = auto_wired::AutoWired();
    auto& c = w.GetClassRef<C>();
    EXPECT_EQ(c.c, 3);

    w.InitAll();
    EXPECT_EQ(test_flag, 4);

    w.DeInitAll();
    EXPECT_EQ(test_flag, 55);
}

TEST_F(AutoWiredV1Test, auto_wired_all) {
    test_flag = 0;
    auto w = auto_wired::AutoWired();
    w.Register<D>();
    w.AutoWiredAll();

    auto& c = w.GetClassRef<C>();
    EXPECT_EQ(c.c, 3);

    w.InitAll();

    EXPECT_EQ(test_flag, 4);

    w.DeInitAll();
    EXPECT_EQ(test_flag, 55);

    {
        auto [has_loop, topological_order] = w.GetTopologicalOrder();
        EXPECT_FALSE(has_loop);

        EXPECT_EQ(topological_order,
                  std::vector<std::string>({
                          std::string("auto_wired::v1::test::A"),
                          std::string("auto_wired::v1::test::BB"),
                          std::string("auto_wired::v1::test::B"),
                          std::string("auto_wired::v1::test::BBB"),
                          std::string("auto_wired::v1::test::C"),
                          std::string("auto_wired::v1::test::D"),
                  }));
    }
}

}  // namespace auto_wired::v1::test