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
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
/*
* Copyright (C) 2020 The Android Open Source Project
*
* 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
*
* http://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 ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_1_3_UTILS_TEST_MOCK_DEVICE_H
#define ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_1_3_UTILS_TEST_MOCK_DEVICE_H
#include <android/hardware/neuralnetworks/1.3/IDevice.h>
#include <gmock/gmock.h>
#include <gtest/gtest.h>
#include <hidl/Status.h>
namespace android::hardware::neuralnetworks::V1_3::utils {
using CacheToken =
hidl_array<uint8_t, static_cast<uint32_t>(V1_2::Constant::BYTE_SIZE_OF_CACHE_TOKEN)>;
class MockDevice final : public IDevice {
public:
static sp<MockDevice> create();
// IBase methods below.
MOCK_METHOD(Return<void>, ping, (), (override));
MOCK_METHOD(Return<bool>, linkToDeathRet, ());
Return<bool> linkToDeath(const sp<hidl_death_recipient>& recipient, uint64_t /*cookie*/);
// V1_0 methods below.
MOCK_METHOD(Return<void>, getCapabilities, (getCapabilities_cb cb), (override));
MOCK_METHOD(Return<void>, getSupportedOperations,
(const V1_0::Model& model, getSupportedOperations_cb cb), (override));
MOCK_METHOD(Return<V1_0::ErrorStatus>, prepareModel,
(const V1_0::Model& model, const sp<V1_0::IPreparedModelCallback>& callback),
(override));
MOCK_METHOD(Return<V1_0::DeviceStatus>, getStatus, (), (override));
// V1_1 methods below.
MOCK_METHOD(Return<void>, getCapabilities_1_1, (getCapabilities_1_1_cb cb), (override));
MOCK_METHOD(Return<void>, getSupportedOperations_1_1,
(const V1_1::Model& model, getSupportedOperations_1_1_cb cb), (override));
MOCK_METHOD(Return<V1_0::ErrorStatus>, prepareModel_1_1,
(const V1_1::Model& model, V1_1::ExecutionPreference preference,
const sp<V1_0::IPreparedModelCallback>& callback),
(override));
// V1_2 methods below.
MOCK_METHOD(Return<void>, getVersionString, (getVersionString_cb cb), (override));
MOCK_METHOD(Return<void>, getType, (getType_cb cb), (override));
MOCK_METHOD(Return<void>, getCapabilities_1_2, (getCapabilities_1_2_cb cb), (override));
MOCK_METHOD(Return<void>, getSupportedExtensions, (getSupportedExtensions_cb cb), (override));
MOCK_METHOD(Return<void>, getSupportedOperations_1_2,
(const V1_2::Model& model, getSupportedOperations_1_2_cb cb), (override));
MOCK_METHOD(Return<void>, getNumberOfCacheFilesNeeded, (getNumberOfCacheFilesNeeded_cb cb),
(override));
MOCK_METHOD(Return<V1_0::ErrorStatus>, prepareModel_1_2,
(const V1_2::Model& model, V1_1::ExecutionPreference preference,
const hidl_vec<hidl_handle>& modelCache, const hidl_vec<hidl_handle>& dataCache,
const CacheToken& token, const sp<V1_2::IPreparedModelCallback>& callback),
(override));
MOCK_METHOD(Return<V1_0::ErrorStatus>, prepareModelFromCache,
(const hidl_vec<hidl_handle>& modelCache, const hidl_vec<hidl_handle>& dataCache,
const CacheToken& token, const sp<V1_2::IPreparedModelCallback>& callback),
(override));
// V1_3 methods below.
MOCK_METHOD(Return<void>, getCapabilities_1_3, (getCapabilities_1_3_cb cb), (override));
MOCK_METHOD(Return<void>, getSupportedOperations_1_3,
(const V1_3::Model& model, getSupportedOperations_1_3_cb cb), (override));
MOCK_METHOD(Return<V1_3::ErrorStatus>, prepareModel_1_3,
(const V1_3::Model& model, V1_1::ExecutionPreference preference,
V1_3::Priority priority, const V1_3::OptionalTimePoint& deadline,
const hidl_vec<hidl_handle>& modelCache, const hidl_vec<hidl_handle>& dataCache,
const CacheToken& token, const sp<V1_3::IPreparedModelCallback>& callback),
(override));
MOCK_METHOD(Return<V1_3::ErrorStatus>, prepareModelFromCache_1_3,
(const V1_3::OptionalTimePoint& deadline, const hidl_vec<hidl_handle>& modelCache,
const hidl_vec<hidl_handle>& dataCache, const CacheToken& token,
const sp<V1_3::IPreparedModelCallback>& callback),
(override));
MOCK_METHOD(Return<void>, allocate,
(const V1_3::BufferDesc& desc,
const hidl_vec<sp<V1_3::IPreparedModel>>& preparedModels,
const hidl_vec<V1_3::BufferRole>& inputRoles,
const hidl_vec<V1_3::BufferRole>& outputRoles, allocate_cb cb),
(override));
// Helper methods.
void simulateCrash();
private:
sp<hidl_death_recipient> mDeathRecipient;
};
inline sp<MockDevice> MockDevice::create() {
auto mockDevice = sp<MockDevice>::make();
// Setup default actions for each relevant call.
const auto ret = []() -> Return<bool> { return true; };
// Setup default actions for each relevant call.
ON_CALL(*mockDevice, linkToDeathRet()).WillByDefault(testing::Invoke(ret));
// These EXPECT_CALL(...).Times(testing::AnyNumber()) calls are to suppress warnings on the
// uninteresting methods calls.
EXPECT_CALL(*mockDevice, linkToDeathRet()).Times(testing::AnyNumber());
return mockDevice;
}
inline Return<bool> MockDevice::linkToDeath(const sp<hidl_death_recipient>& recipient,
uint64_t /*cookie*/) {
mDeathRecipient = recipient;
return linkToDeathRet();
}
inline void MockDevice::simulateCrash() {
ASSERT_NE(nullptr, mDeathRecipient.get());
// Currently, the utils::Device will not use the `cookie` or `who` arguments, so we pass in 0
// and nullptr for these arguments instead. Normally, they are used by the hidl_death_recipient
// to determine which object is dead. However, the utils::Device code only pairs a single death
// recipient with a single HIDL interface object, so these arguments are redundant.
mDeathRecipient->serviceDied(0, nullptr);
}
} // namespace android::hardware::neuralnetworks::V1_3::utils
#endif // ANDROID_HARDWARE_INTERFACES_NEURALNETWORKS_1_3_UTILS_TEST_MOCK_DEVICE_H
|
