1 /*
2 * Copyright (c) 2023 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 <gmock/gmock.h>
17 #include <gtest/gtest.h>
18
19 #include "ipc/distributed_file_daemon_ipc_interface_code.h"
20 #include "ipc/daemon_stub.h"
21 #include "utils_log.h"
22
23 namespace OHOS::Storage::DistributedFile::Test {
24 using namespace testing;
25 using namespace testing::ext;
26 using namespace std;
27
28 namespace {
29 DistributedHardware::DmDeviceInfo deviceInfo = {
30 .deviceId = "testdevid",
31 .deviceName = "testdevname",
32 .networkId = "testnetworkid",
33 };
34 }
35
36 class MockDaemonStub : public DaemonStub {
37 public:
38 MOCK_METHOD1(OpenP2PConnection, int32_t(const DistributedHardware::DmDeviceInfo &deviceInfo));
39 MOCK_METHOD1(CloseP2PConnection, int32_t(const DistributedHardware::DmDeviceInfo &deviceInfo));
40 };
41
42 class DaemonStubTest : public testing::Test {
43 public:
44 static void SetUpTestCase(void);
45 static void TearDownTestCase(void);
46 void SetUp();
47 void TearDown();
48 std::shared_ptr<DaemonStub> daemonStub_;
49 };
50
SetUpTestCase(void)51 void DaemonStubTest::SetUpTestCase(void)
52 {
53 GTEST_LOG_(INFO) << "SetUpTestCase";
54 }
55
TearDownTestCase(void)56 void DaemonStubTest::TearDownTestCase(void)
57 {
58 GTEST_LOG_(INFO) << "TearDownTestCase";
59 }
60
SetUp(void)61 void DaemonStubTest::SetUp(void)
62 {
63 daemonStub_ = std::make_shared<MockDaemonStub>();
64 GTEST_LOG_(INFO) << "SetUp";
65 }
66
TearDown(void)67 void DaemonStubTest::TearDown(void)
68 {
69 GTEST_LOG_(INFO) << "TearDown";
70 }
71
72 /**
73 * @tc.name: DaemonStubOnRemoteRequestTest001
74 * @tc.desc: Verify the OnRemoteRequest function
75 * @tc.type: FUNC
76 * @tc.require: I7M6L1
77 */
78 HWTEST_F(DaemonStubTest, DaemonStubOnRemoteRequestTest001, TestSize.Level1)
79 {
80 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest001 Start";
81 try {
82 MessageParcel data;
83 MessageParcel reply;
84 MessageOption option;
85 int ret = daemonStub_->OnRemoteRequest(
86 static_cast<uint32_t>(DistributedFileDaemonInterfaceCode::DISTRIBUTED_FILE_OPEN_P2P_CONNECTION),
87 data, reply, option);
88 EXPECT_EQ(ret, IDaemon::DFS_DAEMON_DESCRIPTOR_IS_EMPTY);
89 } catch (...) {
90 EXPECT_TRUE(false);
91 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest001 ERROR";
92 }
93 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest001 End";
94 }
95
96 /**
97 * @tc.name: DaemonStubOnRemoteRequestTest002
98 * @tc.desc: Verify the OnRemoteRequest function
99 * @tc.type: FUNC
100 * @tc.require: I7M6L1
101 */
102 HWTEST_F(DaemonStubTest, DaemonStubOnRemoteRequestTest002, TestSize.Level1)
103 {
104 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest002 Start";
105 try {
106 MessageParcel data;
107 MessageParcel reply;
108 MessageOption option;
109 int ret = daemonStub_->OnRemoteRequest(
110 static_cast<uint32_t>(DistributedFileDaemonInterfaceCode::DISTRIBUTED_FILE_CLOSE_P2P_CONNECTION),
111 data, reply, option);
112 EXPECT_EQ(ret, IDaemon::DFS_DAEMON_DESCRIPTOR_IS_EMPTY);
113 } catch (...) {
114 EXPECT_TRUE(false);
115 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest002 ERROR";
116 }
117 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest002 End";
118 }
119
120 /**
121 * @tc.name: DaemonStubOnRemoteRequestTest003
122 * @tc.desc: Verify the OnRemoteRequest function
123 * @tc.type: FUNC
124 * @tc.require: I7M6L1
125 */
126 HWTEST_F(DaemonStubTest, DaemonStubOnRemoteRequestTest003, TestSize.Level1)
127 {
128 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest003 Start";
129 try {
130 int32_t code = 1000;
131 MessageParcel data;
132 EXPECT_TRUE(data.WriteInterfaceToken(IDaemon::GetDescriptor()));
133 MessageParcel reply;
134 MessageOption option;
135 int ret = daemonStub_->OnRemoteRequest(code, data, reply, option);
136 EXPECT_NE(ret, 0);
137 } catch (...) {
138 EXPECT_TRUE(false);
139 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest003 ERROR";
140 }
141 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest003 End";
142 }
143
144 /**
145 * @tc.name: DaemonStubOnRemoteRequestTest004
146 * @tc.desc: Verify the DaemonStubOnRemoteRequest function
147 * @tc.type: FUNC
148 * @tc.require: I7M6L1
149 */
150 HWTEST_F(DaemonStubTest, DaemonStubOnRemoteRequestTest004, TestSize.Level1)
151 {
152 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest004 Start";
153 try {
154 MessageParcel data;
155 EXPECT_TRUE(data.WriteInterfaceToken(IDaemon::GetDescriptor()));
156 MessageParcel reply;
157 MessageOption option;
158
159 EXPECT_TRUE(data.WriteCString(deviceInfo.deviceId));
160 EXPECT_TRUE(data.WriteCString(deviceInfo.deviceName));
161 EXPECT_TRUE(data.WriteCString(deviceInfo.networkId));
162
163 int ret = daemonStub_->OnRemoteRequest(
164 static_cast<uint32_t>(DistributedFileDaemonInterfaceCode::DISTRIBUTED_FILE_OPEN_P2P_CONNECTION),
165 data, reply, option);
166 EXPECT_EQ(ret, 0);
167 } catch (...) {
168 EXPECT_TRUE(false);
169 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest004 ERROR";
170 }
171 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest004 End";
172 }
173
174 /**
175 * @tc.name: DaemonStubOnRemoteRequestTest005
176 * @tc.desc: Verify the DaemonStubOnRemoteRequest function
177 * @tc.type: FUNC
178 * @tc.require: I7M6L1
179 */
180 HWTEST_F(DaemonStubTest, DaemonStubOnRemoteRequestTest005, TestSize.Level1)
181 {
182 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest005 Start";
183 try {
184 MessageParcel data;
185 EXPECT_TRUE(data.WriteInterfaceToken(IDaemon::GetDescriptor()));
186 MessageParcel reply;
187 MessageOption option;
188
189 EXPECT_TRUE(data.WriteCString(deviceInfo.deviceId));
190 EXPECT_TRUE(data.WriteCString(deviceInfo.deviceName));
191 EXPECT_TRUE(data.WriteCString(deviceInfo.networkId));
192
193 int ret = daemonStub_->OnRemoteRequest(
194 static_cast<uint32_t>(DistributedFileDaemonInterfaceCode::DISTRIBUTED_FILE_CLOSE_P2P_CONNECTION),
195 data, reply, option);
196 EXPECT_EQ(ret, 0);
197 } catch (...) {
198 EXPECT_TRUE(false);
199 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest005 ERROR";
200 }
201 GTEST_LOG_(INFO) << "DaemonStubOnRemoteRequestTest005 End";
202 }
203 } // namespace OHOS::Storage::DistributedFile::Test