1 /*
2 * Copyright (c) 2022 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "mock_idevice.h"
17
18 namespace OHOS {
19 namespace HDI {
20 namespace Nnrt {
21 namespace V1_0 {
Get(bool isStub)22 sptr<INnrtDevice> INnrtDevice::Get(bool isStub)
23 {
24 return INnrtDevice::Get("device_service", isStub);
25 }
26
Get(const std::string & serviceName,bool isStub)27 sptr<INnrtDevice> INnrtDevice::Get(const std::string& serviceName, bool isStub)
28 {
29 if (isStub) {
30 return nullptr;
31 }
32
33 sptr<INnrtDevice> mockIDevice = sptr<MockIDevice>(new (std::nothrow) MockIDevice());
34 if (mockIDevice == nullptr) {
35 return nullptr;
36 }
37 std::string deviceName = "MockDevice";
38 EXPECT_CALL(*((V1_0::MockIDevice*)mockIDevice.GetRefPtr()), GetDeviceName(::testing::_))
39 .WillRepeatedly(::testing::DoAll(::testing::SetArgReferee<0>(deviceName), ::testing::Return(HDF_SUCCESS)));
40
41 std::string vendorName = "MockVendor";
42 EXPECT_CALL(*((V1_0::MockIDevice*)mockIDevice.GetRefPtr()), GetVendorName(::testing::_))
43 .WillRepeatedly(::testing::DoAll(::testing::SetArgReferee<0>(vendorName), ::testing::Return(HDF_SUCCESS)));
44
45 V1_0::DeviceStatus deviceStatus = V1_0::DeviceStatus::AVAILABLE;
46 EXPECT_CALL(*((V1_0::MockIDevice*)mockIDevice.GetRefPtr()), GetDeviceStatus(::testing::_))
47 .WillRepeatedly(::testing::DoAll(::testing::SetArgReferee<0>(deviceStatus), ::testing::Return(HDF_SUCCESS)));
48
49 return mockIDevice;
50 }
51 } // V1_0
52 } // Nnrt
53 } // HDI
54 } // OHOS