1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
// Copyright 2020 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "base/test/gmock_move_support.h"
#include <memory>
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace {
using ::testing::DoAll;
using ::testing::Pointee;
using MoveOnly = std::unique_ptr<int>;
struct MockFoo {
MOCK_METHOD(void, ByRef, (MoveOnly&), ());
MOCK_METHOD(void, ByVal, (MoveOnly), ());
MOCK_METHOD(void, TwiceByRef, (MoveOnly&, MoveOnly&), ());
MOCK_METHOD(bool, ByValWithReturnValue, (MoveOnly), ());
};
} // namespace
TEST(GmockMoveSupportTest, MoveArgByRef) {
MoveOnly result;
MockFoo foo;
EXPECT_CALL(foo, ByRef).WillOnce(MoveArg(&result));
MoveOnly arg = std::make_unique<int>(123);
foo.ByRef(arg);
EXPECT_THAT(result, Pointee(123));
}
TEST(GmockMoveSupportTest, MoveArgByVal) {
MoveOnly result;
MockFoo foo;
EXPECT_CALL(foo, ByVal).WillOnce(MoveArg(&result));
foo.ByVal(std::make_unique<int>(456));
EXPECT_THAT(result, Pointee(456));
}
TEST(GmockMoveSupportTest, MoveArgsTwiceByRef) {
MoveOnly result1;
MoveOnly result2;
MockFoo foo;
EXPECT_CALL(foo, TwiceByRef)
.WillOnce(DoAll(MoveArg<0>(&result1), MoveArg<1>(&result2)));
MoveOnly arg1 = std::make_unique<int>(123);
MoveOnly arg2 = std::make_unique<int>(456);
foo.TwiceByRef(arg1, arg2);
EXPECT_THAT(result1, Pointee(123));
EXPECT_THAT(result2, Pointee(456));
}
TEST(GmockMoveSupportTest, MoveArgAndReturn) {
MoveOnly result;
MockFoo foo;
EXPECT_CALL(foo, ByValWithReturnValue)
.WillOnce(MoveArgAndReturn(&result, true));
EXPECT_TRUE(foo.ByValWithReturnValue(std::make_unique<int>(123)));
EXPECT_THAT(result, Pointee(123));
}
|
