1/*
2 * Copyright (c) 2021-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#include <gtest/gtest.h>
17#include "display_manager.h"
18#include "pixel_map.h"
19
20#include <securec.h>
21
22#include "mock_display_manager_adapter.h"
23#include "singleton_mocker.h"
24#include "common_test_utils.h"
25
26using namespace testing;
27using namespace testing::ext;
28
29namespace OHOS {
30namespace Rosen {
31using Mocker = SingletonMocker<DisplayManagerAdapter, MockDisplayManagerAdapter>;
32class ScreenshotTest : public testing::Test {
33public:
34    static void SetUpTestCase();
35    static void TearDownTestCase();
36    virtual void SetUp() override;
37    virtual void TearDown() override;
38};
39void ScreenshotTest::SetUpTestCase()
40{
41    CommonTestUtils::InjectTokenInfoByHapName(0, "com.ohos.systemui", 0);
42    const char** perms = new const char *[1];
43    perms[0] = "ohos.permission.CAPTURE_SCREEN";
44    CommonTestUtils::SetAceessTokenPermission("DisplayManagerServiceTest", perms, 1);
45}
46
47void ScreenshotTest::TearDownTestCase()
48{
49}
50
51void ScreenshotTest::SetUp()
52{
53}
54
55void ScreenshotTest::TearDown()
56{
57}
58
59namespace {
60/**
61 * @tc.name: GetScreenshot_default
62 * @tc.desc: SetWindowRect/GetWindowRect
63 * @tc.type: FUNC
64 */
65HWTEST_F(ScreenshotTest, GetScreenshot_default, Function | SmallTest | Level2)
66{
67    std::unique_ptr<Mocker> m = std::make_unique<Mocker>();
68
69    EXPECT_CALL(m->Mock(), GetDefaultDisplayInfo()).Times(1).WillOnce(Return(nullptr));
70
71    DisplayManager::GetInstance().GetDefaultDisplayId();
72
73    EXPECT_CALL(m->Mock(), GetDisplaySnapshot(_, _)).Times(1).WillOnce(Return(nullptr));
74
75    ASSERT_EQ(nullptr, DisplayManager::GetInstance().GetScreenshot(0));
76}
77
78HWTEST_F(ScreenshotTest, GetScreenshot_01, Function | MediumTest | Level2)
79{
80    std::unique_ptr<Mocker> m = std::make_unique<Mocker>();
81
82    EXPECT_CALL(m->Mock(), GetDefaultDisplayInfo()).Times(1).WillOnce(Return(nullptr));
83    DisplayManager::GetInstance().GetDefaultDisplayId();
84
85    EXPECT_CALL(m->Mock(), GetDisplaySnapshot(_, _)).Times(1).WillOnce(Return(CommonTestUtils::CreatePixelMap()));
86    auto screenshot = DisplayManager::GetInstance().GetScreenshot(0);
87    ASSERT_NE(nullptr, screenshot);
88
89    uint32_t width = screenshot->GetWidth();
90    uint32_t height = screenshot->GetHeight();
91    ASSERT_EQ(width, CommonTestUtils::TEST_IMAGE_WIDTH);
92    ASSERT_EQ(height, CommonTestUtils::TEST_IMAGE_HEIGHT);
93}
94}
95} // namespace Rosen
96} // namespace OHOS
97