1 /*
2 * Copyright (c) 2024 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 <gtest/gtest.h>
17 #include <gmock/gmock.h>
18
19 #include "cloud_sync_common.h"
20 #include "dfs_error.h"
21 #include "file_transfer_manager.h"
22 #include "softbus_adapter_mock.h"
23 #include "softbus_session_mock.h"
24 #include "session_manager.h"
25 #include "socket_mock.h"
26 #include "utils_log.h"
27
28 namespace OHOS {
29 namespace FileManagement::CloudSync {
30 namespace Test {
31 using namespace testing;
32 using namespace testing::ext;
33 using namespace std;
34
35 class FileTransferManagerTest : public testing::Test {
36 public:
37 static void SetUpTestCase(void);
38 static void TearDownTestCase(void);
39 void SetUp();
40 void TearDown();
41 };
42
SetUpTestCase(void)43 void FileTransferManagerTest::SetUpTestCase(void)
44 {
45 std::cout << "SetUpTestCase" << std::endl;
46 }
47
TearDownTestCase(void)48 void FileTransferManagerTest::TearDownTestCase(void)
49 {
50 std::cout << "TearDownTestCase" << std::endl;
51 }
52
SetUp(void)53 void FileTransferManagerTest::SetUp(void)
54 {
55 std::cout << "SetUp" << std::endl;
56 }
57
TearDown(void)58 void FileTransferManagerTest::TearDown(void)
59 {
60 std::cout << "TearDown" << std::endl;
61 }
62
63 /**
64 * @tc.name: InitTest001
65 * @tc.desc: Verify the Init function
66 * @tc.type: FUNC
67 * @tc.require: IB3T5H
68 */
69 HWTEST_F(FileTransferManagerTest, InitTest001, TestSize.Level1)
70 {
71 GTEST_LOG_(INFO) << "InitTest001 start";
72 try {
73 auto sessionManager = make_shared<SessionManager>();
74 auto fileTransferManager = make_shared<FileTransferManager>(sessionManager);
75 fileTransferManager->Init();
76 fileTransferManager->OnSessionClosed();
77 EXPECT_NE(fileTransferManager->sessionManager_, nullptr);
78 } catch (...) {
79 EXPECT_TRUE(false);
80 GTEST_LOG_(INFO) << "InitTest001 failed";
81 }
82 GTEST_LOG_(INFO) << "InitTest001 end";
83 }
84
85 /**
86 * @tc.name: DownloadFileFromRemoteDeviceTest001
87 * @tc.desc: Verify the DownloadFileFromRemoteDevice function
88 * @tc.type: FUNC
89 * @tc.require: IB3T5H
90 */
91 HWTEST_F(FileTransferManagerTest, DownloadFileFromRemoteDeviceTest001, TestSize.Level1)
92 {
93 GTEST_LOG_(INFO) << "DownloadFileFromRemoteDeviceTest001 start";
94 try {
95 auto sessionManager = make_shared<SessionManager>();
96 char data[] = "test data";
97 string peerNetworkId = "test peerNetworkId";
98 auto fileTransferManager = make_shared<FileTransferManager>(sessionManager);
99 fileTransferManager->DownloadFileFromRemoteDevice(peerNetworkId, 0, 0, "data/test");
100 EXPECT_NE(fileTransferManager->sessionManager_, nullptr);
101 } catch (...) {
102 EXPECT_TRUE(false);
103 GTEST_LOG_(INFO) << "DownloadFileFromRemoteDeviceTest001 failed";
104 }
105 GTEST_LOG_(INFO) << "DownloadFileFromRemoteDeviceTest001 end";
106 }
107
108 /**
109 * @tc.name: OnMessageHandleTest001
110 * @tc.desc: Verify the OnMessageHandle function
111 * @tc.type: FUNC
112 * @tc.require: IB3T5H
113 */
114 HWTEST_F(FileTransferManagerTest, OnMessageHandleTest001, TestSize.Level1)
115 {
116 GTEST_LOG_(INFO) << "OnMessageHandleTest001 start";
117 try {
118 auto sessionManager = make_shared<SessionManager>();
119 char data[] = "test data";
120 string peerNetworkId = "test peerNetworkId";
121 auto fileTransferManager = make_shared<FileTransferManager>(sessionManager);
122 fileTransferManager->OnMessageHandle(peerNetworkId, 0, data, sizeof(data));
123 EXPECT_NE(fileTransferManager->sessionManager_, nullptr);
124 } catch (...) {
125 EXPECT_TRUE(false);
126 GTEST_LOG_(INFO) << "OnMessageHandleTest001 failed";
127 }
128 GTEST_LOG_(INFO) << "OnMessageHandleTest001 end";
129 }
130
131 /**
132 * @tc.name: HandleDownloadFileRequestTest001
133 * @tc.desc: Verify the HandleDownloadFileRequest function
134 * @tc.type: FUNC
135 * @tc.require: IB3T5H
136 */
137 HWTEST_F(FileTransferManagerTest, HandleDownloadFileRequestTest001, TestSize.Level1)
138 {
139 GTEST_LOG_(INFO) << "HandleDownloadFileRequestTest001 start";
140 try {
141 auto sessionManager = make_shared<SessionManager>();
142 char data[] = "test data";
143 string peerNetworkId = "test peerNetworkId";
144 auto fileTransferManager = make_shared<FileTransferManager>(sessionManager);
145
146 MessageInputInfo msgInputInfo = {};
147 msgInputInfo.userId = 100;
148 msgInputInfo.taskId = 100;
149 msgInputInfo.uri = "data/test";
150 MessageHandler msgHandle(msgInputInfo);
151 fileTransferManager->HandleDownloadFileRequest(msgHandle, peerNetworkId, 0);
152 fileTransferManager->HandleDownloadFileResponse(msgHandle);
153 fileTransferManager->HandleRecvFileFinished();
154 EXPECT_NE(fileTransferManager->sessionManager_, nullptr);
155 } catch (...) {
156 EXPECT_TRUE(false);
157 GTEST_LOG_(INFO) << "HandleDownloadFileRequestTest001 failed";
158 }
159 GTEST_LOG_(INFO) << "HandleDownloadFileRequestTest001 end";
160 }
161
162 /**
163 * @tc.name: IsFileExistsTest001
164 * @tc.desc: Verify the IsFileExists function
165 * @tc.type: FUNC
166 * @tc.require: IB3T5H
167 */
168 HWTEST_F(FileTransferManagerTest, IsFileExistsTest001, TestSize.Level1)
169 {
170 GTEST_LOG_(INFO) << "IsFileExistsTest001 start";
171 try {
172 auto sessionManager = make_shared<SessionManager>();
173 char data[] = "test data";
174 string peerNetworkId = "test peerNetworkId";
175 auto fileTransferManager = make_shared<FileTransferManager>(sessionManager);
176 EXPECT_TRUE(fileTransferManager->IsFileExists("data/errorPath") == false);
177 } catch (...) {
178 EXPECT_TRUE(false);
179 GTEST_LOG_(INFO) << "IsFileExistsTest001 failed";
180 }
181 GTEST_LOG_(INFO) << "IsFileExistsTest001 end";
182 }
183
184 /**
185 * @tc.name: AddTransTaskTest001
186 * @tc.desc: Verify the AddTransTask function
187 * @tc.type: FUNC
188 * @tc.require: IB3T5H
189 */
190 HWTEST_F(FileTransferManagerTest, AddTransTaskTest001, TestSize.Level1)
191 {
192 GTEST_LOG_(INFO) << "AddTransTaskTest001 start";
193 try {
194 auto sessionManager = make_shared<SessionManager>();
195 char data[] = "test data";
196 string peerNetworkId = "test peerNetworkId";
197 auto fileTransferManager = make_shared<FileTransferManager>(sessionManager);
198 fileTransferManager->AddTransTask("data/test", 0, 100);
199 fileTransferManager->FinishTransTask("data/test", 0);
200 fileTransferManager->RemoveTransTask(100);
201 EXPECT_NE(fileTransferManager->sessionManager_, nullptr);
202 } catch (...) {
203 EXPECT_TRUE(false);
204 GTEST_LOG_(INFO) << "AddTransTaskTest001 failed";
205 }
206 GTEST_LOG_(INFO) << "AddTransTaskTest001 end";
207 }
208 } // namespace Test
209 } // namespace CloudSync
210 } // namespace OHOS