#include <library/cpp/testing/gmock_in_unittest/gmock.h>

#include <library/cpp/testing/unittest/registar.h>

#include <util/generic/string.h>

// Set this variable to true if you want to see failures
/////////////////////////////////////////////////////////
static const bool fail = false;
/////////////////////////////////////////////////////////

class ITestIface {
public:
    virtual ~ITestIface() {
    }

    virtual void Func1() = 0;

    virtual int Func2(const TString&) const = 0;
};

class TTestMock: public ITestIface {
public:
    MOCK_METHOD(void, Func1, (), (override));
    MOCK_METHOD(int, Func2, (const TString&), (const, override));
};

using namespace testing;

Y_UNIT_TEST_SUITE(TExampleGMockTest) {
    Y_UNIT_TEST(TSimpleTest) {
        TTestMock mock;
        EXPECT_CALL(mock, Func1())
            .Times(AtLeast(1));

        if (!fail) {
            mock.Func1();
        }
    }

    Y_UNIT_TEST(TNonExpectedCallTest) {
        TTestMock mock;
        EXPECT_CALL(mock, Func1())
            .Times(AtMost(1));
        mock.Func1();
        if (fail) {
            mock.Func1();
        }
    }

    Y_UNIT_TEST(TReturnValuesTest) {
        TTestMock mock;
        EXPECT_CALL(mock, Func2(TString("1")))
            .WillOnce(Return(1))
            .WillRepeatedly(Return(42));

        EXPECT_CALL(mock, Func2(TString("hello")))
            .WillOnce(Return(-1));

        UNIT_ASSERT_VALUES_EQUAL(mock.Func2("hello"), -1);

        UNIT_ASSERT_VALUES_EQUAL(mock.Func2("1"), 1);
        UNIT_ASSERT_VALUES_EQUAL(mock.Func2("1"), 42);
        UNIT_ASSERT_VALUES_EQUAL(mock.Func2("1"), 42);
        UNIT_ASSERT_VALUES_EQUAL(mock.Func2("1"), 42);
        UNIT_ASSERT_VALUES_EQUAL(mock.Func2("1"), 42);

        if (fail) {
            UNIT_ASSERT_VALUES_EQUAL(mock.Func2("hello"), -1); // expected to return -1 only once
        }
    }

    Y_UNIT_TEST(TStrictCallSequenceTest) {
        TTestMock mock;
        {
            InSequence seq;
            EXPECT_CALL(mock, Func1())
                .Times(1);
            EXPECT_CALL(mock, Func2(_))
                .Times(2)
                .WillOnce(Return(1))
                .WillOnce(Return(2));
            EXPECT_CALL(mock, Func1());
        }
        mock.Func1();
        UNIT_ASSERT_VALUES_EQUAL(mock.Func2("sample"), 1);
        if (fail) {
            mock.Func1();
        }
        UNIT_ASSERT_VALUES_EQUAL(mock.Func2(""), 2);
        if (!fail) {
            mock.Func1();
        }
    }

    Y_UNIT_TEST(TUninterestingMethodIsFailureTest) {
        StrictMock<TTestMock> mock;
        EXPECT_CALL(mock, Func1())
            .Times(1);
        mock.Func1();
        if (fail) {
            mock.Func1();
        }
    }
}