1 /*
2  * Copyright (c) 2024 Huawei Device Co., Ltd.
3  * Licensed under the Apache License, Version 2.0 (the "License") = 0;
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_ACCESSTOKEN_KIT_MOCK_H
17 #define OHOS_FILEMGMT_BACKUP_ACCESSTOKEN_KIT_MOCK_H
18 
19 #include <gmock/gmock.h>
20 
21 #include "accesstoken_kit.h"
22 #include "tokenid_kit.h"
23 
24 namespace OHOS::FileManagement::Backup {
25 class BAccessTokenKit {
26 public:
27     virtual Security::AccessToken::ATokenTypeEnum GetTokenType(Security::AccessToken::AccessTokenID) = 0;
28     virtual int GetHapTokenInfo(Security::AccessToken::AccessTokenID, Security::AccessToken::HapTokenInfo&) = 0;
29     virtual int VerifyAccessToken(Security::AccessToken::AccessTokenID, const std::string&) = 0;
30     virtual bool IsSystemAppByFullTokenID(uint64_t) = 0;
31 public:
32     BAccessTokenKit() = default;
33     virtual ~BAccessTokenKit() = default;
34 public:
35     static inline std::shared_ptr<BAccessTokenKit> token = nullptr;
36 };
37 
38 class AccessTokenKitMock : public BAccessTokenKit {
39 public:
40     MOCK_METHOD(Security::AccessToken::ATokenTypeEnum, GetTokenType, (Security::AccessToken::AccessTokenID));
41     MOCK_METHOD(int, GetHapTokenInfo, (Security::AccessToken::AccessTokenID, Security::AccessToken::HapTokenInfo&));
42     MOCK_METHOD(int, VerifyAccessToken, (Security::AccessToken::AccessTokenID, const std::string&));
43     MOCK_METHOD(bool, IsSystemAppByFullTokenID, (uint64_t));
44 };
45 } // namespace OHOS::FileManagement::Backup
46 #endif // OHOS_FILEMGMT_BACKUP_ACCESSTOKEN_KIT_MOCK_H