blob: 811502d0ce03c3edd959f50bd529cd9ec06808ba [file] [log] [blame]
shiqiane35fdd92008-12-10 05:08:54 +00001$$ -*- mode: c++; -*-
2$$ This is a Pump source file. Please use Pump to convert it to
3$$ gmock-generated-function-mockers.h.
4$$
5$var n = 10 $$ The maximum arity we support.
6// Copyright 2007, Google Inc.
7// All rights reserved.
8//
9// Redistribution and use in source and binary forms, with or without
10// modification, are permitted provided that the following conditions are
11// met:
12//
13// * Redistributions of source code must retain the above copyright
14// notice, this list of conditions and the following disclaimer.
15// * Redistributions in binary form must reproduce the above
16// copyright notice, this list of conditions and the following disclaimer
17// in the documentation and/or other materials provided with the
18// distribution.
19// * Neither the name of Google Inc. nor the names of its
20// contributors may be used to endorse or promote products derived from
21// this software without specific prior written permission.
22//
23// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
24// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
25// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
26// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
27// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
28// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
29// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
30// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
31// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
32// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
33// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
34//
35// Author: wan@google.com (Zhanyong Wan)
36
37// Google Mock - a framework for writing C++ mock classes.
38//
39// This file implements function mockers of various arities.
40
41#ifndef GMOCK_INCLUDE_GMOCK_GMOCK_GENERATED_FUNCTION_MOCKERS_H_
42#define GMOCK_INCLUDE_GMOCK_GMOCK_GENERATED_FUNCTION_MOCKERS_H_
43
zhanyong.wan53e08c42010-09-14 05:38:21 +000044#include "gmock/gmock-spec-builders.h"
45#include "gmock/internal/gmock-internal-utils.h"
shiqiane35fdd92008-12-10 05:08:54 +000046
kosak389bad62014-11-17 01:08:51 +000047#if GTEST_HAS_STD_FUNCTION_
48# include <functional>
49#endif
50
shiqiane35fdd92008-12-10 05:08:54 +000051namespace testing {
shiqiane35fdd92008-12-10 05:08:54 +000052namespace internal {
53
54template <typename F>
55class FunctionMockerBase;
56
57// Note: class FunctionMocker really belongs to the ::testing
58// namespace. However if we define it in ::testing, MSVC will
59// complain when classes in ::testing::internal declare it as a
60// friend class template. To workaround this compiler bug, we define
61// FunctionMocker in ::testing::internal and import it into ::testing.
62template <typename F>
63class FunctionMocker;
64
65
66$range i 0..n
67$for i [[
68$range j 1..i
69$var typename_As = [[$for j [[, typename A$j]]]]
70$var As = [[$for j, [[A$j]]]]
71$var as = [[$for j, [[a$j]]]]
72$var Aas = [[$for j, [[A$j a$j]]]]
73$var ms = [[$for j, [[m$j]]]]
74$var matchers = [[$for j, [[const Matcher<A$j>& m$j]]]]
75template <typename R$typename_As>
76class FunctionMocker<R($As)> : public
77 internal::FunctionMockerBase<R($As)> {
78 public:
79 typedef R F($As);
80 typedef typename internal::Function<F>::ArgumentTuple ArgumentTuple;
81
82 MockSpec<F>& With($matchers) {
83
84$if i >= 1 [[
kosakbd018832014-04-02 20:30:00 +000085 this->current_spec().SetMatchers(::testing::make_tuple($ms));
shiqiane35fdd92008-12-10 05:08:54 +000086
87]]
88 return this->current_spec();
89 }
90
91 R Invoke($Aas) {
zhanyong.wan21a58462009-11-12 19:18:08 +000092 // Even though gcc and MSVC don't enforce it, 'this->' is required
93 // by the C++ standard [14.6.4] here, as the base class type is
94 // dependent on the template argument (and thus shouldn't be
95 // looked into when resolving InvokeWith).
96 return this->InvokeWith(ArgumentTuple($as));
shiqiane35fdd92008-12-10 05:08:54 +000097 }
98};
99
100
101]]
102} // namespace internal
103
104// The style guide prohibits "using" statements in a namespace scope
105// inside a header file. However, the FunctionMocker class template
106// is meant to be defined in the ::testing namespace. The following
107// line is just a trick for working around a bug in MSVC 8.0, which
108// cannot handle it if we define FunctionMocker in ::testing.
109using internal::FunctionMocker;
110
zhanyong.wan20d1a232013-03-01 06:58:38 +0000111// GMOCK_RESULT_(tn, F) expands to the result type of function type F.
112// We define this as a variadic macro in case F contains unprotected
113// commas (the same reason that we use variadic macros in other places
114// in this file).
shiqiane35fdd92008-12-10 05:08:54 +0000115// INTERNAL IMPLEMENTATION - DON'T USE IN USER CODE!!!
zhanyong.wan20d1a232013-03-01 06:58:38 +0000116#define GMOCK_RESULT_(tn, ...) \
117 tn ::testing::internal::Function<__VA_ARGS__>::Result
shiqiane35fdd92008-12-10 05:08:54 +0000118
zhanyong.wan20d1a232013-03-01 06:58:38 +0000119// The type of argument N of the given function type.
shiqiane35fdd92008-12-10 05:08:54 +0000120// INTERNAL IMPLEMENTATION - DON'T USE IN USER CODE!!!
zhanyong.wan20d1a232013-03-01 06:58:38 +0000121#define GMOCK_ARG_(tn, N, ...) \
122 tn ::testing::internal::Function<__VA_ARGS__>::Argument##N
shiqiane35fdd92008-12-10 05:08:54 +0000123
zhanyong.wan20d1a232013-03-01 06:58:38 +0000124// The matcher type for argument N of the given function type.
shiqiane35fdd92008-12-10 05:08:54 +0000125// INTERNAL IMPLEMENTATION - DON'T USE IN USER CODE!!!
zhanyong.wan20d1a232013-03-01 06:58:38 +0000126#define GMOCK_MATCHER_(tn, N, ...) \
127 const ::testing::Matcher<GMOCK_ARG_(tn, N, __VA_ARGS__)>&
shiqiane35fdd92008-12-10 05:08:54 +0000128
129// The variable for mocking the given method.
130// INTERNAL IMPLEMENTATION - DON'T USE IN USER CODE!!!
zhanyong.wan68be1112009-03-25 03:56:48 +0000131#define GMOCK_MOCKER_(arity, constness, Method) \
zhanyong.wanccedc1c2010-08-09 22:46:12 +0000132 GTEST_CONCAT_TOKEN_(gmock##constness##arity##_##Method##_, __LINE__)
shiqiane35fdd92008-12-10 05:08:54 +0000133
134
135$for i [[
136$range j 1..i
137$var arg_as = [[$for j, \
zhanyong.wan20d1a232013-03-01 06:58:38 +0000138 [[GMOCK_ARG_(tn, $j, __VA_ARGS__) gmock_a$j]]]]
shiqiane35fdd92008-12-10 05:08:54 +0000139$var as = [[$for j, [[gmock_a$j]]]]
140$var matcher_as = [[$for j, \
zhanyong.wan20d1a232013-03-01 06:58:38 +0000141 [[GMOCK_MATCHER_(tn, $j, __VA_ARGS__) gmock_a$j]]]]
shiqiane35fdd92008-12-10 05:08:54 +0000142// INTERNAL IMPLEMENTATION - DON'T USE IN USER CODE!!!
zhanyong.wan20d1a232013-03-01 06:58:38 +0000143#define GMOCK_METHOD$i[[]]_(tn, constness, ct, Method, ...) \
144 GMOCK_RESULT_(tn, __VA_ARGS__) ct Method( \
145 $arg_as) constness { \
kosakbd018832014-04-02 20:30:00 +0000146 GTEST_COMPILE_ASSERT_((::testing::tuple_size< \
zhanyong.wan20d1a232013-03-01 06:58:38 +0000147 tn ::testing::internal::Function<__VA_ARGS__>::ArgumentTuple>::value == $i), \
shiqiane35fdd92008-12-10 05:08:54 +0000148 this_method_does_not_take_$i[[]]_argument[[$if i != 1 [[s]]]]); \
zhanyong.wan68be1112009-03-25 03:56:48 +0000149 GMOCK_MOCKER_($i, constness, Method).SetOwnerAndName(this, #Method); \
150 return GMOCK_MOCKER_($i, constness, Method).Invoke($as); \
shiqiane35fdd92008-12-10 05:08:54 +0000151 } \
zhanyong.wan20d1a232013-03-01 06:58:38 +0000152 ::testing::MockSpec<__VA_ARGS__>& \
shiqiane35fdd92008-12-10 05:08:54 +0000153 gmock_##Method($matcher_as) constness { \
zhanyong.waned6c9272011-02-23 19:39:27 +0000154 GMOCK_MOCKER_($i, constness, Method).RegisterOwner(this); \
155 return GMOCK_MOCKER_($i, constness, Method).With($as); \
shiqiane35fdd92008-12-10 05:08:54 +0000156 } \
zhanyong.wan20d1a232013-03-01 06:58:38 +0000157 mutable ::testing::FunctionMocker<__VA_ARGS__> GMOCK_MOCKER_($i, constness, Method)
shiqiane35fdd92008-12-10 05:08:54 +0000158
159
160]]
161$for i [[
zhanyong.wan20d1a232013-03-01 06:58:38 +0000162#define MOCK_METHOD$i(m, ...) GMOCK_METHOD$i[[]]_(, , , m, __VA_ARGS__)
shiqiane35fdd92008-12-10 05:08:54 +0000163
164]]
165
166
167$for i [[
zhanyong.wan20d1a232013-03-01 06:58:38 +0000168#define MOCK_CONST_METHOD$i(m, ...) GMOCK_METHOD$i[[]]_(, const, , m, __VA_ARGS__)
shiqiane35fdd92008-12-10 05:08:54 +0000169
170]]
171
172
173$for i [[
zhanyong.wan20d1a232013-03-01 06:58:38 +0000174#define MOCK_METHOD$i[[]]_T(m, ...) GMOCK_METHOD$i[[]]_(typename, , , m, __VA_ARGS__)
shiqiane35fdd92008-12-10 05:08:54 +0000175
176]]
177
178
179$for i [[
zhanyong.wan20d1a232013-03-01 06:58:38 +0000180#define MOCK_CONST_METHOD$i[[]]_T(m, ...) \
181 GMOCK_METHOD$i[[]]_(typename, const, , m, __VA_ARGS__)
shiqiane35fdd92008-12-10 05:08:54 +0000182
183]]
184
185
186$for i [[
zhanyong.wan20d1a232013-03-01 06:58:38 +0000187#define MOCK_METHOD$i[[]]_WITH_CALLTYPE(ct, m, ...) \
188 GMOCK_METHOD$i[[]]_(, , ct, m, __VA_ARGS__)
shiqiane35fdd92008-12-10 05:08:54 +0000189
190]]
191
192
193$for i [[
zhanyong.wan20d1a232013-03-01 06:58:38 +0000194#define MOCK_CONST_METHOD$i[[]]_WITH_CALLTYPE(ct, m, ...) \
195 GMOCK_METHOD$i[[]]_(, const, ct, m, __VA_ARGS__)
shiqiane35fdd92008-12-10 05:08:54 +0000196
197]]
198
199
200$for i [[
zhanyong.wan20d1a232013-03-01 06:58:38 +0000201#define MOCK_METHOD$i[[]]_T_WITH_CALLTYPE(ct, m, ...) \
202 GMOCK_METHOD$i[[]]_(typename, , ct, m, __VA_ARGS__)
shiqiane35fdd92008-12-10 05:08:54 +0000203
204]]
205
206
207$for i [[
zhanyong.wan20d1a232013-03-01 06:58:38 +0000208#define MOCK_CONST_METHOD$i[[]]_T_WITH_CALLTYPE(ct, m, ...) \
209 GMOCK_METHOD$i[[]]_(typename, const, ct, m, __VA_ARGS__)
shiqiane35fdd92008-12-10 05:08:54 +0000210
211]]
212
zhanyong.wanf3aa4d22009-09-25 22:34:47 +0000213// A MockFunction<F> class has one mock method whose type is F. It is
214// useful when you just want your test code to emit some messages and
215// have Google Mock verify the right messages are sent (and perhaps at
216// the right times). For example, if you are exercising code:
217//
218// Foo(1);
219// Foo(2);
220// Foo(3);
221//
222// and want to verify that Foo(1) and Foo(3) both invoke
223// mock.Bar("a"), but Foo(2) doesn't invoke anything, you can write:
224//
225// TEST(FooTest, InvokesBarCorrectly) {
226// MyMock mock;
227// MockFunction<void(string check_point_name)> check;
228// {
229// InSequence s;
230//
231// EXPECT_CALL(mock, Bar("a"));
232// EXPECT_CALL(check, Call("1"));
233// EXPECT_CALL(check, Call("2"));
234// EXPECT_CALL(mock, Bar("a"));
235// }
236// Foo(1);
237// check.Call("1");
238// Foo(2);
239// check.Call("2");
240// Foo(3);
241// }
242//
243// The expectation spec says that the first Bar("a") must happen
244// before check point "1", the second Bar("a") must happen after check
245// point "2", and nothing should happen between the two check
246// points. The explicit check points make it easy to tell which
247// Bar("a") is called by which call to Foo().
kosaka9e02a92014-06-17 23:19:54 +0000248//
249// MockFunction<F> can also be used to exercise code that accepts
250// std::function<F> callbacks. To do so, use AsStdFunction() method
251// to create std::function proxy forwarding to original object's Call.
252// Example:
253//
254// TEST(FooTest, RunsCallbackWithBarArgument) {
255// MockFunction<int(string)> callback;
256// EXPECT_CALL(callback, Call("bar")).WillOnce(Return(1));
257// Foo(callback.AsStdFunction());
258// }
zhanyong.wanf3aa4d22009-09-25 22:34:47 +0000259template <typename F>
260class MockFunction;
261
262
263$for i [[
264$range j 0..i-1
kosaka9e02a92014-06-17 23:19:54 +0000265$var ArgTypes = [[$for j, [[A$j]]]]
266$var ArgNames = [[$for j, [[a$j]]]]
267$var ArgDecls = [[$for j, [[A$j a$j]]]]
zhanyong.wanf3aa4d22009-09-25 22:34:47 +0000268template <typename R$for j [[, typename A$j]]>
kosaka9e02a92014-06-17 23:19:54 +0000269class MockFunction<R($ArgTypes)> {
zhanyong.wanf3aa4d22009-09-25 22:34:47 +0000270 public:
zhanyong.wan32de5f52009-12-23 00:13:23 +0000271 MockFunction() {}
272
kosaka9e02a92014-06-17 23:19:54 +0000273 MOCK_METHOD$i[[]]_T(Call, R($ArgTypes));
274
kosak5b9cbbb2014-11-17 00:28:55 +0000275#if GTEST_HAS_STD_FUNCTION_
kosaka9e02a92014-06-17 23:19:54 +0000276 std::function<R($ArgTypes)> AsStdFunction() {
kosaka6e32f02015-07-24 20:05:26 +0000277 return [this]($ArgDecls) -> R {
kosaka9e02a92014-06-17 23:19:54 +0000278 return this->Call($ArgNames);
279 };
280 }
kosak5b9cbbb2014-11-17 00:28:55 +0000281#endif // GTEST_HAS_STD_FUNCTION_
zhanyong.wan32de5f52009-12-23 00:13:23 +0000282
283 private:
284 GTEST_DISALLOW_COPY_AND_ASSIGN_(MockFunction);
zhanyong.wanf3aa4d22009-09-25 22:34:47 +0000285};
286
287
288]]
shiqiane35fdd92008-12-10 05:08:54 +0000289} // namespace testing
290
291#endif // GMOCK_INCLUDE_GMOCK_GMOCK_GENERATED_FUNCTION_MOCKERS_H_