1 /* 2 * Copyright (c) 2025 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_ACCESS_TOKEN_KIT_MOCK_H 16 #define OHOS_ACCESS_TOKEN_KIT_MOCK_H 17 18 #include <gmock/gmock.h> 19 #include "accesstoken_kit.h" 20 #include "access_token.h" 21 22 namespace OHOS { 23 namespace Security { 24 namespace AccessToken { 25 class BAccessTokenKit { 26 public: 27 virtual ATokenTypeEnum GetTokenTypeFlag(AccessTokenID) = 0; 28 virtual ATokenTypeEnum GetTokenType(AccessTokenID) = 0; 29 virtual int GetHapTokenInfo(AccessTokenID, HapTokenInfo&) = 0; 30 virtual int GetNativeTokenInfo(AccessTokenID, NativeTokenInfo&) = 0; 31 virtual int VerifyAccessToken(AccessTokenID, const std::string&) = 0; 32 BAccessTokenKit() = default; 33 virtual ~BAccessTokenKit() = default; 34 private: 35 static inline std::shared_ptr<BAccessTokenKit> accessTokenkit = nullptr; 36 }; 37 38 class AccessTokenKitMock : public BAccessTokenKit { 39 public: 40 MOCK_METHOD(ATokenTypeEnum, GetTokenTypeFlag, (AccessTokenID)); 41 MOCK_METHOD(ATokenTypeEnum, GetTokenType, (AccessTokenID)); 42 MOCK_METHOD(int, GetHapTokenInfo, (AccessTokenID, HapTokenInfo&)); 43 MOCK_METHOD(int, GetNativeTokenInfo, (AccessTokenID, NativeTokenInfo&)); 44 MOCK_METHOD(int, VerifyAccessToken, (AccessTokenID, const std::string&)); 45 }; 46 } 47 } 48 } 49 #endif //OHOS_ACCESS_TOKEN_KIT_MOCK_H 50