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 <gtest/gtest.h>
17 #include <gmock/gmock.h>
18 
19 #include "download_asset_callback_stub.h"
20 #include "cloud_download_uri_manager.h"
21 #include "asset_callback_mock.h"
22 #include "dfs_error.h"
23 
24 namespace OHOS {
25 namespace FileManagement::CloudSync {
26 namespace Test {
27 using namespace testing;
28 using namespace testing::ext;
29 using namespace std;
30 
31 class DownloadAssetCallbackStubMock : public DownloadAssetCallbackStub {
32 public:
DownloadAssetCallbackStubMock()33     DownloadAssetCallbackStubMock() {}
OnFinished(const TaskId taskId,const std::string & uri,const int32_t result)34     void OnFinished(const TaskId taskId, const std::string &uri, const int32_t result) override
35     {}
36 };
37 
38 class DownloadAssetCallbackStubTest : public testing::Test {
39 public:
40     static void SetUpTestCase(void);
41     static void TearDownTestCase(void);
42     void SetUp();
43     void TearDown();
44     std::shared_ptr<DownloadAssetCallbackStub> downloadAssetCallbackStub_;
45 };
46 
SetUpTestCase(void)47 void DownloadAssetCallbackStubTest::SetUpTestCase(void)
48 {
49     std::cout << "SetUpTestCase" << std::endl;
50 }
51 
TearDownTestCase(void)52 void DownloadAssetCallbackStubTest::TearDownTestCase(void)
53 {
54     std::cout << "TearDownTestCase" << std::endl;
55 }
56 
SetUp(void)57 void DownloadAssetCallbackStubTest::SetUp(void)
58 {
59     downloadAssetCallbackStub_ = std::make_shared<DownloadAssetCallbackStubMock>();
60     std::cout << "SetUp" << std::endl;
61 }
62 
TearDown(void)63 void DownloadAssetCallbackStubTest::TearDown(void)
64 {
65     std::cout << "TearDown" << std::endl;
66 }
67 
68 /**
69  * @tc.name: OnRemoteRequestTest001
70  * @tc.desc: Verify the OnRemoteRequest function.
71  * @tc.type: FUNC
72  * @tc.require: I6H5MH
73  */
74 HWTEST_F(DownloadAssetCallbackStubTest, OnRemoteRequestTest001, TestSize.Level1)
75 {
76     GTEST_LOG_(INFO) << "OnRemoteRequest Start";
77     try {
78         MessageParcel data;
79         MessageParcel reply;
80         MessageOption option;
81         EXPECT_EQ(E_SERVICE_DESCRIPTOR_IS_EMPTY, downloadAssetCallbackStub_->OnRemoteRequest(
82             IDownloadAssetCallback::SERVICE_CMD_ON_DOWNLOAD_FINSHED, data, reply, option));
83     } catch(...) {
84         EXPECT_TRUE(false);
85         GTEST_LOG_(INFO) << " OnRemoteRequest ERROR";
86     }
87     GTEST_LOG_(INFO) << "OnRemoteRequest End";
88 }
89 
90 /**
91  * @tc.name: OnRemoteRequestTest002
92  * @tc.desc: Verify the OnRemoteRequest function.
93  * @tc.type: FUNC
94  * @tc.require: I6H5MH
95  */
96 HWTEST_F(DownloadAssetCallbackStubTest, OnRemoteRequestTest002, TestSize.Level1)
97 {
98     GTEST_LOG_(INFO) << "OnRemoteRequest Start";
99     try {
100         uint32_t code = 100;
101         MessageParcel data;
102         MessageParcel reply;
103         MessageOption option;
104         EXPECT_TRUE(data.WriteInterfaceToken(IDownloadAssetCallback::GetDescriptor()));
105         int32_t ret = downloadAssetCallbackStub_->OnRemoteRequest(code, data, reply, option);
106         EXPECT_NE(E_OK, ret);
107     } catch(...) {
108         EXPECT_TRUE(false);
109         GTEST_LOG_(INFO) << " OnRemoteRequest ERROR";
110     }
111     GTEST_LOG_(INFO) << "OnRemoteRequest End";
112 }
113 
114 /**
115  * @tc.name: OnRemoteRequestTest003
116  * @tc.desc: Verify the OnRemoteRequest function.
117  * @tc.type: FUNC
118  * @tc.require: issueI7UYAL
119  */
120 HWTEST_F(DownloadAssetCallbackStubTest, OnRemoteRequestTest003, TestSize.Level1)
121 {
122     GTEST_LOG_(INFO) << "OnRemoteRequest Start";
123     try {
124         uint32_t code = IDownloadAssetCallback::SERVICE_CMD_ON_DOWNLOAD_FINSHED;
125         MessageParcel data;
126         MessageParcel reply;
127         MessageOption option;
128         EXPECT_TRUE(data.WriteInterfaceToken(IDownloadAssetCallback::GetDescriptor()));
129         int32_t ret = downloadAssetCallbackStub_->OnRemoteRequest(code, data, reply, option);
130         EXPECT_EQ(E_OK, ret);
131     } catch(...) {
132         EXPECT_TRUE(false);
133         GTEST_LOG_(INFO) << " OnRemoteRequest ERROR";
134     }
135     GTEST_LOG_(INFO) << "OnRemoteRequest End";
136 }
137 
138 /**
139  * @tc.name: HandleOnFinishedTest001
140  * @tc.desc: Verify the HandleOnFinished function.
141  * @tc.type: FUNC
142  * @tc.require: I6H5MH
143  */
144 HWTEST_F(DownloadAssetCallbackStubTest, HandleOnFinishedTest001, TestSize.Level1)
145 {
146     GTEST_LOG_(INFO) << "HandleOnFinished Start";
147     try {
148         MessageParcel data;
149         MessageParcel reply;
150         int ret = downloadAssetCallbackStub_->HandleOnFinished(data, reply);
151         EXPECT_EQ(E_OK, ret);
152     } catch(...) {
153         EXPECT_TRUE(false);
154         GTEST_LOG_(INFO) << " HandleOnFinished ERROR";
155     }
156     GTEST_LOG_(INFO) << "HandleOnFinished End";
157 }
158 
159 } // namespace Test
160 } // namespace FileManagement::CloudSync {
161 } // namespace OHOS
162