8f2828c4b4
ccdbb5941 Export of internal Abseil changes 01f5f81f9 Export of internal Abseil changes 2c92bdc7c Export of internal Abseil changes e7ebf9803 Export of internal Abseil changes 2eba343b5 Export of internal Abseil changes a8b03d90e Export of internal Abseil changes 1d31b5c36 Export of internal Abseil changes da3a87690 Export of internal Abseil changes 8faf20461 Exclude empty directories (#697) 2069dc796 Export of internal Abseil changes 4832bf6bf Added a BUILD file in root to expose license. (#695) af8f994af Export of internal Abseil changes 33caf1097 Export of internal Abseil changes cf1a02e2d Export of internal Abseil changes git-subtree-dir: third_party/abseil_cpp git-subtree-split: ccdbb5941f992fabda7eae3ce72f55efc17c826a
97 lines
3.8 KiB
C++
97 lines
3.8 KiB
C++
//
|
|
// Copyright 2019 The Abseil Authors.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// https://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
#ifndef ABSL_RANDOM_INTERNAL_MOCK_OVERLOAD_SET_H_
|
|
#define ABSL_RANDOM_INTERNAL_MOCK_OVERLOAD_SET_H_
|
|
|
|
#include <type_traits>
|
|
|
|
#include "gmock/gmock.h"
|
|
#include "gtest/gtest.h"
|
|
#include "absl/random/internal/mock_helpers.h"
|
|
#include "absl/random/mocking_bit_gen.h"
|
|
|
|
namespace absl {
|
|
ABSL_NAMESPACE_BEGIN
|
|
namespace random_internal {
|
|
|
|
template <typename DistrT, typename Fn>
|
|
struct MockSingleOverload;
|
|
|
|
// MockSingleOverload
|
|
//
|
|
// MockSingleOverload hooks in to gMock's `ON_CALL` and `EXPECT_CALL` macros.
|
|
// EXPECT_CALL(mock_single_overload, Call(...))` will expand to a call to
|
|
// `mock_single_overload.gmock_Call(...)`. Because expectations are stored on
|
|
// the MockingBitGen (an argument passed inside `Call(...)`), this forwards to
|
|
// arguments to MockingBitGen::Register.
|
|
//
|
|
// The underlying KeyT must match the KeyT constructed by DistributionCaller.
|
|
template <typename DistrT, typename Ret, typename... Args>
|
|
struct MockSingleOverload<DistrT, Ret(MockingBitGen&, Args...)> {
|
|
static_assert(std::is_same<typename DistrT::result_type, Ret>::value,
|
|
"Overload signature must have return type matching the "
|
|
"distribution result_type.");
|
|
using KeyT = Ret(DistrT, std::tuple<Args...>);
|
|
auto gmock_Call(
|
|
absl::MockingBitGen& gen, // NOLINT(google-runtime-references)
|
|
const ::testing::Matcher<Args>&... matchers)
|
|
-> decltype(MockHelpers::MockFor<KeyT>(gen).gmock_Call(matchers...)) {
|
|
return MockHelpers::MockFor<KeyT>(gen).gmock_Call(matchers...);
|
|
}
|
|
};
|
|
|
|
template <typename DistrT, typename Ret, typename Arg, typename... Args>
|
|
struct MockSingleOverload<DistrT, Ret(Arg, MockingBitGen&, Args...)> {
|
|
static_assert(std::is_same<typename DistrT::result_type, Ret>::value,
|
|
"Overload signature must have return type matching the "
|
|
"distribution result_type.");
|
|
using KeyT = Ret(DistrT, std::tuple<Arg, Args...>);
|
|
auto gmock_Call(
|
|
const ::testing::Matcher<Arg>& matcher,
|
|
absl::MockingBitGen& gen, // NOLINT(google-runtime-references)
|
|
const ::testing::Matcher<Args>&... matchers)
|
|
-> decltype(MockHelpers::MockFor<KeyT>(gen).gmock_Call(matcher,
|
|
matchers...)) {
|
|
return MockHelpers::MockFor<KeyT>(gen).gmock_Call(matcher, matchers...);
|
|
}
|
|
};
|
|
|
|
// MockOverloadSet
|
|
//
|
|
// MockOverloadSet takes a distribution and a collection of signatures and
|
|
// performs overload resolution amongst all the overloads. This makes
|
|
// `EXPECT_CALL(mock_overload_set, Call(...))` expand and do overload resolution
|
|
// correctly.
|
|
template <typename DistrT, typename... Signatures>
|
|
struct MockOverloadSet;
|
|
|
|
template <typename DistrT, typename Sig>
|
|
struct MockOverloadSet<DistrT, Sig> : public MockSingleOverload<DistrT, Sig> {
|
|
using MockSingleOverload<DistrT, Sig>::gmock_Call;
|
|
};
|
|
|
|
template <typename DistrT, typename FirstSig, typename... Rest>
|
|
struct MockOverloadSet<DistrT, FirstSig, Rest...>
|
|
: public MockSingleOverload<DistrT, FirstSig>,
|
|
public MockOverloadSet<DistrT, Rest...> {
|
|
using MockSingleOverload<DistrT, FirstSig>::gmock_Call;
|
|
using MockOverloadSet<DistrT, Rest...>::gmock_Call;
|
|
};
|
|
|
|
} // namespace random_internal
|
|
ABSL_NAMESPACE_END
|
|
} // namespace absl
|
|
#endif // ABSL_RANDOM_INTERNAL_MOCK_OVERLOAD_SET_H_
|