use_mocker_test.cc

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

#include "autowired/autowired.h"

namespace auto_wired::v1::test::use_mocker {

static int test_flag = 0;

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

class B {
public:
    AUTO_WIRED_DECLARE_BEGIN(B)
    AUTO_WIRED_DECLARE_CLASS(a)
    AUTO_WIRED_DECLARE_END()

    virtual ~B() = default;

public:
    virtual int F() {
        return 2;
    }

private:
    A& a;
};

class BMocker : public B {
public:
    BMocker() : B(auto_wired::AutoWired::MockInstance()) {}

    virtual int F() override {
        return 3;
    }
};

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

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

    int c{3};

private:
    A& a;
    B& b;
};

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

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

    w.InitAll();

    EXPECT_EQ(test_flag, 3);

    w.DeInitAll();
}

TEST_F(UseMockerTest, has_mocker) {
    test_flag = 0;
    auto w = auto_wired::AutoWired();

    auto* b_mocker = new BMocker();
    w.UseMocker<B>(b_mocker);

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

    w.InitAll();

    EXPECT_EQ(test_flag, 4);

    w.DeInitAll();
}

}  // namespace auto_wired::v1::test::use_mocker