• Home
  • Line#
  • Scopes#
  • Navigate#
  • Raw
  • Download
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 #ifndef OHOS_FILEMGMT_BACKUP_SERVICE_STUB_MOCK_H
17 #define OHOS_FILEMGMT_BACKUP_SERVICE_STUB_MOCK_H
18 
19 #include <gmock/gmock.h>
20 
21 namespace OHOS::FileManagement::Backup {
22 class BServiceStub {
23 public:
24     virtual int32_t OnRemoteRequest(uint32_t, MessageParcel&, MessageParcel&, MessageOption&) = 0;
25     virtual int32_t CmdInitRestoreSession(MessageParcel&, MessageParcel&) = 0;
26     virtual int32_t CmdInitBackupSession(MessageParcel&, MessageParcel&) = 0;
27     virtual int32_t CmdStart(MessageParcel&, MessageParcel&) = 0;
28     virtual int32_t CmdGetLocalCapabilities(MessageParcel&, MessageParcel&) = 0;
29     virtual int32_t CmdGetLocalCapabilitiesForBdInfos(MessageParcel&, MessageParcel&) = 0;
30     virtual int32_t CmdPublishFile(MessageParcel&, MessageParcel&) = 0;
31     virtual int32_t CmdAppFileReady(MessageParcel&, MessageParcel&) = 0;
32     virtual int32_t CmdAppDone(MessageParcel&, MessageParcel&) = 0;
33     virtual int32_t CmdResultReport(MessageParcel&, MessageParcel&) = 0;
34     virtual int32_t CmdGetFileHandle(MessageParcel&, MessageParcel&) = 0;
35     virtual int32_t CmdAppendBundlesRestoreSession(MessageParcel&, MessageParcel&) = 0;
36     virtual int32_t CmdAppendBundlesDetailsRestoreSession(MessageParcel&, MessageParcel&) = 0;
37     virtual int32_t CmdAppendBundlesBackupSession(MessageParcel&, MessageParcel&) = 0;
38     virtual int32_t CmdAppendBundlesDetailsBackupSession(MessageParcel&, MessageParcel&) = 0;
39     virtual int32_t CmdFinish(MessageParcel&, MessageParcel&) = 0;
40     virtual int32_t CmdRelease(MessageParcel&, MessageParcel&) = 0;
41     virtual int32_t CmdCancel(MessageParcel&, MessageParcel&) = 0;
42     virtual int32_t CmdGetLocalCapabilitiesIncremental(MessageParcel&, MessageParcel&) = 0;
43     virtual int32_t CmdGetAppLocalListAndDoIncrementalBackup(MessageParcel&, MessageParcel&) = 0;
44     virtual int32_t CmdInitIncrementalBackupSession(MessageParcel&, MessageParcel&) = 0;
45     virtual int32_t CmdAppendBundlesIncrementalBackupSession(MessageParcel&, MessageParcel&) = 0;
46     virtual int32_t CmdAppendBundlesDetailsIncrementalBackupSession(MessageParcel&, MessageParcel&) = 0;
47     virtual int32_t CmdPublishIncrementalFile(MessageParcel&, MessageParcel&) = 0;
48     virtual int32_t CmdPublishSAIncrementalFile(MessageParcel&, MessageParcel&) = 0;
49     virtual int32_t CmdAppIncrementalFileReady(MessageParcel&, MessageParcel&) = 0;
50     virtual int32_t CmdAppIncrementalDone(MessageParcel&, MessageParcel&) = 0;
51     virtual int32_t CmdGetIncrementalFileHandle(MessageParcel&, MessageParcel&) = 0;
52     virtual int32_t CmdGetBackupInfo(MessageParcel&, MessageParcel&) = 0;
53     virtual int32_t CmdUpdateTimer(MessageParcel&, MessageParcel&) = 0;
54     virtual int32_t CmdUpdateSendRate(MessageParcel&, MessageParcel&) = 0;
55     virtual int32_t CmdStopExtTimer(MessageParcel&, MessageParcel&) = 0;
56     virtual int32_t CmdRefreshDataSize(MessageParcel&, MessageParcel&) = 0;
57     virtual int32_t CmdGetBackupDataSize(MessageParcel&, MessageParcel&) = 0;
58     virtual void ServiceStubSupplement() = 0;
59     virtual void ServiceStubSuppAppendBundles() = 0;
60 
61 public:
62     BServiceStub() = default;
63     virtual ~BServiceStub() = default;
64 public:
65     static inline std::shared_ptr<BServiceStub> stub = nullptr;
66 };
67 
68 class ServiceStubMock : public BServiceStub {
69 public:
70     MOCK_METHOD(int32_t, OnRemoteRequest, (uint32_t, MessageParcel&, MessageParcel&, MessageOption&));
71     MOCK_METHOD(int32_t, CmdInitRestoreSession, (MessageParcel&, MessageParcel&));
72     MOCK_METHOD(int32_t, CmdInitBackupSession, (MessageParcel&, MessageParcel&));
73     MOCK_METHOD(int32_t, CmdStart, (MessageParcel&, MessageParcel&));
74     MOCK_METHOD(int32_t, CmdGetLocalCapabilities, (MessageParcel&, MessageParcel&));
75     MOCK_METHOD(int32_t, CmdGetLocalCapabilitiesForBdInfos, (MessageParcel&, MessageParcel&));
76     MOCK_METHOD(int32_t, CmdPublishFile, (MessageParcel&, MessageParcel&));
77     MOCK_METHOD(int32_t, CmdAppFileReady, (MessageParcel&, MessageParcel&));
78     MOCK_METHOD(int32_t, CmdAppDone, (MessageParcel&, MessageParcel&));
79     MOCK_METHOD(int32_t, CmdResultReport, (MessageParcel&, MessageParcel&));
80     MOCK_METHOD(int32_t, CmdGetFileHandle, (MessageParcel&, MessageParcel&));
81     MOCK_METHOD(int32_t, CmdAppendBundlesRestoreSession, (MessageParcel&, MessageParcel&));
82     MOCK_METHOD(int32_t, CmdAppendBundlesDetailsRestoreSession, (MessageParcel&, MessageParcel&));
83     MOCK_METHOD(int32_t, CmdAppendBundlesBackupSession, (MessageParcel&, MessageParcel&));
84     MOCK_METHOD(int32_t, CmdAppendBundlesDetailsBackupSession, (MessageParcel&, MessageParcel&));
85     MOCK_METHOD(int32_t, CmdFinish, (MessageParcel&, MessageParcel&));
86     MOCK_METHOD(int32_t, CmdRelease, (MessageParcel&, MessageParcel&));
87     MOCK_METHOD(int32_t, CmdGetLocalCapabilitiesIncremental, (MessageParcel&, MessageParcel&));
88     MOCK_METHOD(int32_t, CmdGetAppLocalListAndDoIncrementalBackup, (MessageParcel&, MessageParcel&));
89     MOCK_METHOD(int32_t, CmdInitIncrementalBackupSession, (MessageParcel&, MessageParcel&));
90     MOCK_METHOD(int32_t, CmdAppendBundlesIncrementalBackupSession, (MessageParcel&, MessageParcel&));
91     MOCK_METHOD(int32_t, CmdAppendBundlesDetailsIncrementalBackupSession, (MessageParcel&, MessageParcel&));
92     MOCK_METHOD(int32_t, CmdPublishIncrementalFile, (MessageParcel&, MessageParcel&));
93     MOCK_METHOD(int32_t, CmdPublishSAIncrementalFile, (MessageParcel&, MessageParcel&));
94     MOCK_METHOD(int32_t, CmdAppIncrementalFileReady, (MessageParcel&, MessageParcel&));
95     MOCK_METHOD(int32_t, CmdAppIncrementalDone, (MessageParcel&, MessageParcel&));
96     MOCK_METHOD(int32_t, CmdGetIncrementalFileHandle, (MessageParcel&, MessageParcel&));
97     MOCK_METHOD(int32_t, CmdGetBackupInfo, (MessageParcel&, MessageParcel&));
98     MOCK_METHOD(int32_t, CmdUpdateTimer, (MessageParcel&, MessageParcel&));
99     MOCK_METHOD(int32_t, CmdUpdateSendRate, (MessageParcel&, MessageParcel&));
100     MOCK_METHOD(int32_t, CmdStopExtTimer, (MessageParcel&, MessageParcel&));
101     MOCK_METHOD(int32_t, CmdRefreshDataSize, (MessageParcel&, MessageParcel&));
102     MOCK_METHOD(int32_t, CmdGetBackupDataSize, (MessageParcel&, MessageParcel&));
103     MOCK_METHOD(void, ServiceStubSupplement, ());
104     MOCK_METHOD(void, ServiceStubSuppAppendBundles, ());
105 };
106 } // namespace OHOS::FileManagement::Backup
107 #endif // OHOS_FILEMGMT_BACKUP_SERVICE_STUB_MOCK_H
108