1 /* 2 * Copyright (c) 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 16 #ifndef FRAMEWORKS_WM_TEST_UT_MOCK_DISPLAY_MANAGER_ADAPTER_LITE_H 17 #define FRAMEWORKS_WM_TEST_UT_MOCK_DISPLAY_MANAGER_ADAPTER_LITE_H 18 #include <gmock/gmock.h> 19 20 #include "display_manager_adapter_lite.h" 21 22 namespace OHOS { 23 namespace Rosen { 24 class MockDisplayManagerAdapterLite : public DisplayManagerAdapterLite { 25 public: 26 MOCK_METHOD0(Clear, void()); 27 MOCK_METHOD2(RegisterDisplayManagerAgent, DMError(const sptr<IDisplayManagerAgent>& displayManagerAgent, 28 DisplayManagerAgentType type)); 29 MOCK_METHOD2(UnregisterDisplayManagerAgent, DMError(const sptr<IDisplayManagerAgent>& displayManagerAgent, 30 DisplayManagerAgentType type)); 31 MOCK_METHOD0(GetDefaultDisplayInfo, sptr<DisplayInfo>()); 32 MOCK_METHOD1(GetDisplayInfo, sptr<DisplayInfo>(DisplayId displayId)); 33 34 MOCK_METHOD1(WakeUpBegin, bool(PowerStateChangeReason reason)); 35 MOCK_METHOD0(WakeUpEnd, bool()); 36 MOCK_METHOD1(SuspendBegin, bool(PowerStateChangeReason reason)); 37 MOCK_METHOD0(SuspendEnd, bool()); 38 MOCK_METHOD1(SetDisplayState, bool(DisplayState state)); 39 MOCK_METHOD1(GetDisplayState, DisplayState(DisplayId displayId)); 40 }; 41 42 class MockScreenManagerAdapterLite : public ScreenManagerAdapterLite { 43 public: 44 MOCK_METHOD0(Clear, void()); 45 MOCK_METHOD2(RegisterDisplayManagerAgent, DMError(const sptr<IDisplayManagerAgent>& displayManagerAgent, 46 DisplayManagerAgentType type)); 47 MOCK_METHOD2(UnregisterDisplayManagerAgent, DMError(const sptr<IDisplayManagerAgent>& displayManagerAgent, 48 DisplayManagerAgentType type)); 49 MOCK_METHOD3(SetSpecifiedScreenPower, bool(ScreenId screenId, ScreenPowerState state, 50 PowerStateChangeReason reason)); 51 MOCK_METHOD2(SetScreenPowerForAll, bool(ScreenPowerState state, PowerStateChangeReason reason)); 52 MOCK_METHOD1(GetScreenPower, ScreenPowerState(ScreenId dmsScreenId)); 53 }; 54 } 55 } // namespace OHOS 56 57 #endif // FRAMEWORKS_WM_TEST_UT_MOCK_DISPLAY_MANAGER_ADAPTER_LITE_H