1 /*
2  * Copyright (C) 2021 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 MOCK_ACCESSIBLE_ABILITY_CONNECTION_H
17 #define MOCK_ACCESSIBLE_ABILITY_CONNECTION_H
18 
19 #include <gmock/gmock.h>
20 #include "accessible_ability_connection.h"
21 
22 namespace OHOS {
23 namespace Accessibility {
24 class MockAccessibleAbilityChannel : public AccessibleAbilityChannel {
25 public:
26     MockAccessibleAbilityChannel(const int32_t accountId, const std::string &clientName);
27     ~MockAccessibleAbilityChannel();
28 
29     MOCK_METHOD5(SearchElementInfoByAccessibilityId,
30         RetError(const ElementBasicInfo elementBasicInfo, const int32_t requestId,
31             const sptr<IAccessibilityElementOperatorCallback>& callback, const int32_t mode, bool isFilter));
32 
33     MOCK_METHOD5(SearchElementInfosByText,
34         RetError(const int32_t accessibilityWindowId, const int64_t elementId, const std::string& text,
35             const int32_t requestId, const sptr<IAccessibilityElementOperatorCallback>& callback));
36     MOCK_METHOD5(FindFocusedElementInfo,
37         RetError(const int32_t accessibilityWindowId, const int64_t elementId, const int32_t focusType,
38             const int32_t requestId, const sptr<IAccessibilityElementOperatorCallback>& callback));
39 
40     MOCK_METHOD5(
41         FocusMoveSearch, RetError(const int32_t accessibilityWindowId, const int64_t elementId, const int32_t direction,
42                              const int32_t requestId, const sptr<IAccessibilityElementOperatorCallback>& callback));
43     MOCK_METHOD6(ExecuteAction, RetError(const int32_t accessibilityWindowId, const int64_t elementId,
44         const int32_t action, const std::map<std::string, std::string> &actionArguments, const int32_t requestId,
45         const sptr<IAccessibilityElementOperatorCallback>& callback));
46     MOCK_METHOD1(GetWindows, RetError(std::vector<AccessibilityWindowInfo> &windows));
47     MOCK_METHOD2(GetWindowsByDisplayId, RetError(const uint64_t displayId,
48         std::vector<AccessibilityWindowInfo> &windows));
49     MOCK_METHOD2(SetOnKeyPressEventResult, void(const bool handled, const int32_t sequence));
50     MOCK_METHOD1(SendSimulateGesture, RetError(const std::shared_ptr<AccessibilityGestureInjectPath>& gesturePath));
51 };
52 
53 class MockAccessibleAbilityConnection : public AccessibleAbilityConnection {
54 public:
55     MockAccessibleAbilityConnection(int32_t accountId, int32_t connectionId, AccessibilityAbilityInfo& abilityInfo);
56 
57     virtual ~MockAccessibleAbilityConnection();
58 
59     MOCK_METHOD3(OnAbilityConnectDone,
60         void(const AppExecFwk::ElementName& element, const sptr<IRemoteObject>& remoteObject, int32_t resultCode));
61     MOCK_METHOD2(OnAbilityDisconnectDone, void(const AppExecFwk::ElementName& element, int32_t resultCode));
62     // For AccessibleAbilityClientProxy
63     MOCK_METHOD1(OnAccessibilityEvent, void(AccessibilityEventInfo& eventInfo));
64     MOCK_METHOD2(OnKeyPressEvent, bool(const MMI::KeyEvent& keyEvent, const int32_t sequence));
65     MOCK_METHOD2(OnGestureSimulateResult, bool(const int32_t sequence, const bool completedSuccessfully));
66     // Get Attribution
67     AccessibilityAbilityInfo abilityInfo_;
68     AppExecFwk::ElementName elementName_;
GetAbilityInfo()69     inline AccessibilityAbilityInfo& GetAbilityInfo()
70     {
71         return abilityInfo_;
72     }
73 
GetElementName()74     inline AppExecFwk::ElementName& GetElementName()
75     {
76         return elementName_;
77     }
78 
GetAccountData()79     inline wptr<AccessibilityAccountData> GetAccountData()
80     {
81         return nullptr;
82     }
83 
GetProxy()84     inline sptr<IAccessibleAbilityClient> GetProxy()
85     {
86         return nullptr;
87     }
88 
89     MOCK_METHOD0(Disconnect, void());
90     MOCK_METHOD1(Connect, void(const AppExecFwk::ElementName& element));
91     MOCK_METHOD0(GetChannelId, int32_t());
92 };
93 } // namespace Accessibility
94 } // namespace OHOS
95 #endif // MOCK_ACCESSIBLE_ABILITY_CONNECTION_H