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 #ifndef OHOS_STORAGE_FILE_DFS_IASSET_RECV_CALLBACK_MOCK_H
16 #define OHOS_STORAGE_FILE_DFS_IASSET_RECV_CALLBACK_MOCK_H
17 #include "i_asset_recv_callback.h"
18 
19 #include <gmock/gmock.h>
20 
21 #include "iremote_stub.h"
22 
23 #include "asset_obj.h"
24 
25 namespace OHOS::Storage::DistributedFile {
26 class IAssetRecvCallbackMock : public IRemoteStub<IAssetRecvCallback> {
27 public:
28     int code_ = 0;
IAssetRecvCallbackMock()29     IAssetRecvCallbackMock() : code_(0) {}
~IAssetRecvCallbackMock()30     virtual ~IAssetRecvCallbackMock() {}
31 
32     MOCK_METHOD4(SendRequest, int(uint32_t, MessageParcel &, MessageParcel &, MessageOption &));
33     MOCK_METHOD4(OnStart, int32_t(const std::string &srcNetworkId,
34                                   const std::string &dstNetworkId,
35                                   const std::string &sessionId,
36                                   const std::string &dstBundleName));
37     MOCK_METHOD3(OnFinished, int32_t(const std::string &srcNetworkId,
38                                      const sptr<AssetObj> &assetObj,
39                                      int32_t result));
40 };
41 }
42 #endif