1 // Copyright 2017 The Chromium Authors
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "base/test/mock_callback.h"
6
7 #include "base/functional/callback.h"
8 #include "testing/gmock/include/gmock/gmock.h"
9
10 using testing::InSequence;
11 using testing::Return;
12
13 namespace base {
14 namespace {
15
TEST(MockCallbackTest,ZeroArgs)16 TEST(MockCallbackTest, ZeroArgs) {
17 MockCallback<RepeatingClosure> mock_closure;
18 EXPECT_CALL(mock_closure, Run());
19 mock_closure.Get().Run();
20
21 MockCallback<RepeatingCallback<int()>> mock_int_callback;
22 {
23 InSequence sequence;
24 EXPECT_CALL(mock_int_callback, Run()).WillOnce(Return(42));
25 EXPECT_CALL(mock_int_callback, Run()).WillOnce(Return(88));
26 }
27 EXPECT_EQ(42, mock_int_callback.Get().Run());
28 EXPECT_EQ(88, mock_int_callback.Get().Run());
29 }
30
TEST(MockCallbackTest,WithArgs)31 TEST(MockCallbackTest, WithArgs) {
32 MockCallback<RepeatingCallback<int(int, int)>> mock_two_int_callback;
33 EXPECT_CALL(mock_two_int_callback, Run(1, 2)).WillOnce(Return(42));
34 EXPECT_CALL(mock_two_int_callback, Run(0, 0)).WillRepeatedly(Return(-1));
35 RepeatingCallback<int(int, int)> two_int_callback =
36 mock_two_int_callback.Get();
37 EXPECT_EQ(-1, two_int_callback.Run(0, 0));
38 EXPECT_EQ(42, two_int_callback.Run(1, 2));
39 EXPECT_EQ(-1, two_int_callback.Run(0, 0));
40 }
41
TEST(MockCallbackTest,ZeroArgsOnce)42 TEST(MockCallbackTest, ZeroArgsOnce) {
43 MockCallback<OnceClosure> mock_closure;
44 EXPECT_CALL(mock_closure, Run());
45 mock_closure.Get().Run();
46
47 MockCallback<OnceCallback<int()>> mock_int_callback;
48 EXPECT_CALL(mock_int_callback, Run()).WillOnce(Return(88));
49 EXPECT_EQ(88, mock_int_callback.Get().Run());
50 }
51
TEST(MockCallbackTest,WithArgsOnce)52 TEST(MockCallbackTest, WithArgsOnce) {
53 MockCallback<OnceCallback<int(int, int)>> mock_two_int_callback;
54 EXPECT_CALL(mock_two_int_callback, Run(1, 2)).WillOnce(Return(42));
55 OnceCallback<int(int, int)> two_int_callback = mock_two_int_callback.Get();
56 EXPECT_EQ(42, std::move(two_int_callback).Run(1, 2));
57 }
58
TEST(MockCallbackTest,Typedefs)59 TEST(MockCallbackTest, Typedefs) {
60 static_assert(std::is_same_v<MockCallback<RepeatingCallback<int()>>,
61 MockRepeatingCallback<int()>>,
62 "Repeating typedef differs for zero args");
63 static_assert(std::is_same_v<MockCallback<RepeatingCallback<int(int, int)>>,
64 MockRepeatingCallback<int(int, int)>>,
65 "Repeating typedef differs for multiple args");
66 static_assert(std::is_same_v<MockCallback<RepeatingCallback<void()>>,
67 MockRepeatingClosure>,
68 "Repeating typedef differs for closure");
69 static_assert(std::is_same_v<MockCallback<OnceCallback<int()>>,
70 MockOnceCallback<int()>>,
71 "Once typedef differs for zero args");
72 static_assert(std::is_same_v<MockCallback<OnceCallback<int(int, int)>>,
73 MockOnceCallback<int(int, int)>>,
74 "Once typedef differs for multiple args");
75 static_assert(std::is_same_v<MockCallback<RepeatingCallback<void()>>,
76 MockRepeatingClosure>,
77 "Once typedef differs for closure");
78 }
79
80 } // namespace
81 } // namespace base
82