mirror of
https://github.com/google/googletest.git
synced 2024-12-31 05:33:02 +08:00
ca4b7c9ff4
Give each of Naggy/Nice/StrictMock a base class whose constructor runs before the mocked class's constructor, and a destructor that runs after the mocked class's destructor, so that any mock methods run in either the constructor or destructor use the same strictness as other calls. PiperOrigin-RevId: 348511612
263 lines
10 KiB
C++
263 lines
10 KiB
C++
// Copyright 2008, Google Inc.
|
|
// All rights reserved.
|
|
//
|
|
// Redistribution and use in source and binary forms, with or without
|
|
// modification, are permitted provided that the following conditions are
|
|
// met:
|
|
//
|
|
// * Redistributions of source code must retain the above copyright
|
|
// notice, this list of conditions and the following disclaimer.
|
|
// * Redistributions in binary form must reproduce the above
|
|
// copyright notice, this list of conditions and the following disclaimer
|
|
// in the documentation and/or other materials provided with the
|
|
// distribution.
|
|
// * Neither the name of Google Inc. nor the names of its
|
|
// contributors may be used to endorse or promote products derived from
|
|
// this software without specific prior written permission.
|
|
//
|
|
// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
|
|
// Implements class templates NiceMock, NaggyMock, and StrictMock.
|
|
//
|
|
// Given a mock class MockFoo that is created using Google Mock,
|
|
// NiceMock<MockFoo> is a subclass of MockFoo that allows
|
|
// uninteresting calls (i.e. calls to mock methods that have no
|
|
// EXPECT_CALL specs), NaggyMock<MockFoo> is a subclass of MockFoo
|
|
// that prints a warning when an uninteresting call occurs, and
|
|
// StrictMock<MockFoo> is a subclass of MockFoo that treats all
|
|
// uninteresting calls as errors.
|
|
//
|
|
// Currently a mock is naggy by default, so MockFoo and
|
|
// NaggyMock<MockFoo> behave like the same. However, we will soon
|
|
// switch the default behavior of mocks to be nice, as that in general
|
|
// leads to more maintainable tests. When that happens, MockFoo will
|
|
// stop behaving like NaggyMock<MockFoo> and start behaving like
|
|
// NiceMock<MockFoo>.
|
|
//
|
|
// NiceMock, NaggyMock, and StrictMock "inherit" the constructors of
|
|
// their respective base class. Therefore you can write
|
|
// NiceMock<MockFoo>(5, "a") to construct a nice mock where MockFoo
|
|
// has a constructor that accepts (int, const char*), for example.
|
|
//
|
|
// A known limitation is that NiceMock<MockFoo>, NaggyMock<MockFoo>,
|
|
// and StrictMock<MockFoo> only works for mock methods defined using
|
|
// the MOCK_METHOD* family of macros DIRECTLY in the MockFoo class.
|
|
// If a mock method is defined in a base class of MockFoo, the "nice"
|
|
// or "strict" modifier may not affect it, depending on the compiler.
|
|
// In particular, nesting NiceMock, NaggyMock, and StrictMock is NOT
|
|
// supported.
|
|
|
|
// GOOGLETEST_CM0002 DO NOT DELETE
|
|
|
|
#ifndef GMOCK_INCLUDE_GMOCK_GMOCK_NICE_STRICT_H_
|
|
#define GMOCK_INCLUDE_GMOCK_GMOCK_NICE_STRICT_H_
|
|
|
|
#include <type_traits>
|
|
|
|
#include "gmock/gmock-spec-builders.h"
|
|
#include "gmock/internal/gmock-port.h"
|
|
|
|
namespace testing {
|
|
template <class MockClass>
|
|
class NiceMock;
|
|
template <class MockClass>
|
|
class NaggyMock;
|
|
template <class MockClass>
|
|
class StrictMock;
|
|
|
|
namespace internal {
|
|
template <typename T>
|
|
std::true_type StrictnessModifierProbe(const NiceMock<T>&);
|
|
template <typename T>
|
|
std::true_type StrictnessModifierProbe(const NaggyMock<T>&);
|
|
template <typename T>
|
|
std::true_type StrictnessModifierProbe(const StrictMock<T>&);
|
|
std::false_type StrictnessModifierProbe(...);
|
|
|
|
template <typename T>
|
|
constexpr bool HasStrictnessModifier() {
|
|
return decltype(StrictnessModifierProbe(std::declval<const T&>()))::value;
|
|
}
|
|
|
|
// Base classes that register and deregister with testing::Mock to alter the
|
|
// default behavior around uninteresting calls. Inheriting from one of these
|
|
// classes first and then MockClass ensures the MockClass constructor is run
|
|
// after registration, and that the MockClass destructor runs before
|
|
// deregistration. This guarantees that MockClass's constructor and destructor
|
|
// run with the same level of strictness as its instance methods.
|
|
|
|
#if GTEST_OS_WINDOWS && (defined(_MSC_VER) || defined(__clang__))
|
|
// We need to mark these classes with this declspec to ensure that
|
|
// the empty base class optimization is performed.
|
|
#define GTEST_INTERNAL_EMPTY_BASE_CLASS __declspec(empty_bases)
|
|
#else
|
|
#define GTEST_INTERNAL_EMPTY_BASE_CLASS
|
|
#endif
|
|
|
|
template <typename Base>
|
|
class NiceMockImpl {
|
|
public:
|
|
NiceMockImpl() { ::testing::Mock::AllowUninterestingCalls(this); }
|
|
|
|
~NiceMockImpl() { ::testing::Mock::UnregisterCallReaction(this); }
|
|
};
|
|
|
|
template <typename Base>
|
|
class NaggyMockImpl {
|
|
public:
|
|
NaggyMockImpl() { ::testing::Mock::WarnUninterestingCalls(this); }
|
|
|
|
~NaggyMockImpl() { ::testing::Mock::UnregisterCallReaction(this); }
|
|
};
|
|
|
|
template <typename Base>
|
|
class StrictMockImpl {
|
|
public:
|
|
StrictMockImpl() { ::testing::Mock::FailUninterestingCalls(this); }
|
|
|
|
~StrictMockImpl() { ::testing::Mock::UnregisterCallReaction(this); }
|
|
};
|
|
|
|
} // namespace internal
|
|
|
|
template <class MockClass>
|
|
class GTEST_INTERNAL_EMPTY_BASE_CLASS NiceMock
|
|
: private internal::NiceMockImpl<MockClass>,
|
|
public MockClass {
|
|
public:
|
|
static_assert(
|
|
!internal::HasStrictnessModifier<MockClass>(),
|
|
"Can't apply NiceMock to a class hierarchy that already has a "
|
|
"strictness modifier. See "
|
|
"https://github.com/google/googletest/blob/master/googlemock/docs/"
|
|
"cook_book.md#the-nice-the-strict-and-the-naggy-nicestrictnaggy");
|
|
NiceMock() : MockClass() {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
// Ideally, we would inherit base class's constructors through a using
|
|
// declaration, which would preserve their visibility. However, many existing
|
|
// tests rely on the fact that current implementation reexports protected
|
|
// constructors as public. These tests would need to be cleaned up first.
|
|
|
|
// Single argument constructor is special-cased so that it can be
|
|
// made explicit.
|
|
template <typename A>
|
|
explicit NiceMock(A&& arg) : MockClass(std::forward<A>(arg)) {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
template <typename TArg1, typename TArg2, typename... An>
|
|
NiceMock(TArg1&& arg1, TArg2&& arg2, An&&... args)
|
|
: MockClass(std::forward<TArg1>(arg1), std::forward<TArg2>(arg2),
|
|
std::forward<An>(args)...) {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
private:
|
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(NiceMock);
|
|
};
|
|
|
|
template <class MockClass>
|
|
class GTEST_INTERNAL_EMPTY_BASE_CLASS NaggyMock
|
|
: private internal::NaggyMockImpl<MockClass>,
|
|
public MockClass {
|
|
static_assert(
|
|
!internal::HasStrictnessModifier<MockClass>(),
|
|
"Can't apply NaggyMock to a class hierarchy that already has a "
|
|
"strictness modifier. See "
|
|
"https://github.com/google/googletest/blob/master/googlemock/docs/"
|
|
"cook_book.md#the-nice-the-strict-and-the-naggy-nicestrictnaggy");
|
|
|
|
public:
|
|
NaggyMock() : MockClass() {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
// Ideally, we would inherit base class's constructors through a using
|
|
// declaration, which would preserve their visibility. However, many existing
|
|
// tests rely on the fact that current implementation reexports protected
|
|
// constructors as public. These tests would need to be cleaned up first.
|
|
|
|
// Single argument constructor is special-cased so that it can be
|
|
// made explicit.
|
|
template <typename A>
|
|
explicit NaggyMock(A&& arg) : MockClass(std::forward<A>(arg)) {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
template <typename TArg1, typename TArg2, typename... An>
|
|
NaggyMock(TArg1&& arg1, TArg2&& arg2, An&&... args)
|
|
: MockClass(std::forward<TArg1>(arg1), std::forward<TArg2>(arg2),
|
|
std::forward<An>(args)...) {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
private:
|
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(NaggyMock);
|
|
};
|
|
|
|
template <class MockClass>
|
|
class GTEST_INTERNAL_EMPTY_BASE_CLASS StrictMock
|
|
: private internal::StrictMockImpl<MockClass>,
|
|
public MockClass {
|
|
public:
|
|
static_assert(
|
|
!internal::HasStrictnessModifier<MockClass>(),
|
|
"Can't apply StrictMock to a class hierarchy that already has a "
|
|
"strictness modifier. See "
|
|
"https://github.com/google/googletest/blob/master/googlemock/docs/"
|
|
"cook_book.md#the-nice-the-strict-and-the-naggy-nicestrictnaggy");
|
|
StrictMock() : MockClass() {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
// Ideally, we would inherit base class's constructors through a using
|
|
// declaration, which would preserve their visibility. However, many existing
|
|
// tests rely on the fact that current implementation reexports protected
|
|
// constructors as public. These tests would need to be cleaned up first.
|
|
|
|
// Single argument constructor is special-cased so that it can be
|
|
// made explicit.
|
|
template <typename A>
|
|
explicit StrictMock(A&& arg) : MockClass(std::forward<A>(arg)) {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
template <typename TArg1, typename TArg2, typename... An>
|
|
StrictMock(TArg1&& arg1, TArg2&& arg2, An&&... args)
|
|
: MockClass(std::forward<TArg1>(arg1), std::forward<TArg2>(arg2),
|
|
std::forward<An>(args)...) {
|
|
static_assert(sizeof(*this) == sizeof(MockClass),
|
|
"The impl subclass shouldn't introduce any padding");
|
|
}
|
|
|
|
private:
|
|
GTEST_DISALLOW_COPY_AND_ASSIGN_(StrictMock);
|
|
};
|
|
|
|
#undef GTEST_INTERNAL_EMPTY_BASE_CLASS
|
|
|
|
} // namespace testing
|
|
|
|
#endif // GMOCK_INCLUDE_GMOCK_GMOCK_NICE_STRICT_H_
|