1 /*
2  * Copyright (c) 2022-2023 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 IAM_MOCK_AUTHENTICATION_H
16 #define IAM_MOCK_AUTHENTICATION_H
17 
18 #include <gmock/gmock.h>
19 
20 #include "authentication.h"
21 
22 namespace OHOS {
23 namespace UserIam {
24 namespace UserAuth {
25 class MockAuthentication final : public Authentication {
26 public:
27     virtual ~MockAuthentication() = default;
28     MOCK_METHOD1(SetExecutor, void(uint32_t executorIndex));
29     MOCK_METHOD1(SetChallenge, void(const std::vector<uint8_t> &challenge));
30     MOCK_METHOD1(SetAccessTokenId, void(uint32_t tokenId));
31     MOCK_METHOD1(SetEndAfterFirstFail, void(bool endAfterFirstFail));
32     MOCK_METHOD1(SetCollectorUdid, void(std::string &collectorUdid));
33 
34     MOCK_METHOD2(Start,
35         bool(std::vector<std::shared_ptr<ScheduleNode>> &scheduleList, std::shared_ptr<ScheduleNodeCallback> callback));
36     MOCK_METHOD2(Update, bool(const std::vector<uint8_t> &scheduleResult, AuthResultInfo &resultInfo));
37     MOCK_CONST_METHOD0(GetAuthExecutorMsgs, std::vector<Authentication::AuthExecutorMsg>());
38     MOCK_METHOD0(Cancel, bool());
39     MOCK_CONST_METHOD0(GetAccessTokenId, uint32_t());
40     MOCK_CONST_METHOD0(GetLatestError, int32_t());
41     MOCK_CONST_METHOD0(GetUserId, int32_t());
42     MOCK_CONST_METHOD0(GetAuthType, int32_t());
43 
44 protected:
45     MOCK_METHOD1(SetLatestError, void(int32_t error));
46 };
47 } // namespace UserAuth
48 } // namespace UserIam
49 } // namespace OHOS
50 #endif // IAM_MOCK_AUTHENTICATION_H