1 /* 2 * Copyright (c) 2022 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 ACCOUNT_IAM_CLIENT_TEST_CALLBACK_H 17 #define ACCOUNT_IAM_CLIENT_TEST_CALLBACK_H 18 19 #include <vector> 20 #include <gmock/gmock.h> 21 #include "account_iam_client_callback.h" 22 #include "account_iam_info.h" 23 24 namespace OHOS { 25 namespace AccountTest { 26 class MockIDMCallback final : public AccountSA::IDMCallback { 27 public: 28 MOCK_METHOD2(OnResult, void(int32_t result, const AccountSA::Attributes &extraInfo)); 29 MOCK_METHOD3(OnAcquireInfo, void(int32_t module, uint32_t acquireInfo, const AccountSA::Attributes &extraInfo)); 30 }; 31 32 class MockGetCredInfoCallback final : public AccountSA::GetCredInfoCallback { 33 public: 34 MOCK_METHOD2(OnCredentialInfo, void(int32_t result, const std::vector<AccountSA::CredentialInfo> &infoList)); 35 }; 36 37 class MockGetSetPropCallback final : public AccountSA::GetSetPropCallback { 38 public: 39 MOCK_METHOD2(OnResult, void(int32_t result, const AccountSA::Attributes &extraInfo)); 40 }; 41 } // namespace AccountTest 42 } // namespace OHOS 43 44 #endif // ACCOUNT_IAM_CLIENT_TEST_CALLBACK_H