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 #include <gtest/gtest.h>
17 #include "display_cutout_controller.h"
18 #include "display_manager_service_inner.h"
19 #include "display_manager_service.h"
20 #include "dm_common.h"
21 
22 using namespace testing;
23 using namespace testing::ext;
24 
25 namespace OHOS {
26 namespace Rosen {
27 class DisplayCutoutControllerTest : public testing::Test {
28 public:
29     static void SetUpTestCase();
30     static void TearDownTestCase();
31     void SetUp() override;
32     void TearDown() override;
33 };
34 
SetUpTestCase()35 void DisplayCutoutControllerTest::SetUpTestCase()
36 {
37 }
38 
TearDownTestCase()39 void DisplayCutoutControllerTest::TearDownTestCase()
40 {
41 }
42 
SetUp()43 void DisplayCutoutControllerTest::SetUp()
44 {
45 }
46 
TearDown()47 void DisplayCutoutControllerTest::TearDown()
48 {
49 }
50 
51 namespace {
52 /**
53  * @tc.name: SetCurvedScreenBoundary
54  * @tc.desc: SetCurvedScreenBoundary size < 4
55  * @tc.type: FUNC
56  */
57 HWTEST_F(DisplayCutoutControllerTest, SetCurvedScreenBoundary, Function | SmallTest | Level1)
58 {
59     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
60     std::vector<int> curvedScreenBoundary;
61     curvedScreenBoundary.emplace_back(2);
62     controller->SetCurvedScreenBoundary(curvedScreenBoundary);
63     ASSERT_EQ(controller->curvedScreenBoundary_.size(), 4);
64 }
65 
66 /**
67  * @tc.name: SetCutoutSvgPath
68  * @tc.desc: SetCutoutSvgPath
69  * @tc.type: FUNC
70  */
71 HWTEST_F(DisplayCutoutControllerTest, SetCutoutSvgPath, Function | SmallTest | Level1)
72 {
73     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
74     DisplayId displayId = 1;
75     std::string svgPath = "m10";
76     controller->SetCutoutSvgPath(displayId, svgPath);
77     controller->SetCutoutSvgPath(displayId, svgPath);
78     ASSERT_EQ(controller->svgPaths_.size(), 1);
79     ASSERT_EQ(controller->svgPaths_[displayId].size(), 2);
80 }
81 
82 /**
83  * @tc.name: GetCutoutInfo
84  * @tc.desc: GetCutoutInfo
85  * @tc.type: FUNC
86  */
87 HWTEST_F(DisplayCutoutControllerTest, GetCutoutInfo, Function | SmallTest | Level1)
88 {
89     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
90     DisplayId displayId = DisplayManagerServiceInner::GetInstance().GetDefaultDisplayId();
91     std::string svgPath = "m10";
92     controller->SetCutoutSvgPath(displayId, svgPath);
93     controller->SetIsWaterfallDisplay(true);
94     sptr<CutoutInfo> cutoutInfo = controller->GetCutoutInfo(displayId);
95     ASSERT_NE(cutoutInfo, nullptr);
96 }
97 
98 
99 /**
100  * @tc.name: CalcBuiltInDisplayWaterfallRects
101  * @tc.desc: CalcBuiltInDisplayWaterfallRects
102  * @tc.type: FUNC
103  */
104 HWTEST_F(DisplayCutoutControllerTest, CalcBuiltInDisplayWaterfallRects, Function | SmallTest | Level1)
105 {
106     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
107     controller->SetIsWaterfallDisplay(true);
108     std::vector<int> curvedScreenBoundary;
109     controller->SetCurvedScreenBoundary(curvedScreenBoundary);
110     controller->CalcBuiltInDisplayWaterfallRects();
111     ASSERT_TRUE(controller->waterfallDisplayAreaRects_.isUninitialized());
112     curvedScreenBoundary.emplace_back(1);
113     curvedScreenBoundary.emplace_back(2);
114     curvedScreenBoundary.emplace_back(3);
115     curvedScreenBoundary.emplace_back(4);
116     controller->SetCurvedScreenBoundary(curvedScreenBoundary);
117     controller->CalcBuiltInDisplayWaterfallRects();
118     ASSERT_TRUE(controller->waterfallDisplayAreaRects_.isUninitialized());
119 }
120 
121 
122 /**
123  * @tc.name: CalcBuiltInDisplayWaterfallRectsByRotation
124  * @tc.desc: CalcBuiltInDisplayWaterfallRectsByRotation
125  * @tc.type: FUNC
126  */
127 HWTEST_F(DisplayCutoutControllerTest, CalcBuiltInDisplayWaterfallRectsByRotation, Function | SmallTest | Level1)
128 {
129     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
130     uint32_t displayHeight = 300;
131     uint32_t displayWidth = 300;
132     std::vector<int> curvedScreenBoundary;
133     curvedScreenBoundary.emplace_back(0);
134     curvedScreenBoundary.emplace_back(0);
135     curvedScreenBoundary.emplace_back(500);
136     curvedScreenBoundary.emplace_back(500);
137     controller->SetCurvedScreenBoundary(curvedScreenBoundary);
138     controller->CalcBuiltInDisplayWaterfallRectsByRotation(Rotation::ROTATION_270, displayHeight, displayWidth);
139     controller->CalcBuiltInDisplayWaterfallRectsByRotation(Rotation::ROTATION_180, displayHeight, displayWidth);
140     controller->CalcBuiltInDisplayWaterfallRectsByRotation(Rotation::ROTATION_90, displayHeight, displayWidth);
141     controller->CalcBuiltInDisplayWaterfallRectsByRotation(Rotation::ROTATION_0, displayHeight, displayWidth);
142     controller->CalcBuiltInDisplayWaterfallRectsByRotation(static_cast<Rotation>(10), displayHeight, displayWidth);
143     ASSERT_FALSE(controller->waterfallDisplayAreaRects_.isUninitialized());
144 }
145 
146 /**
147  * @tc.name: CheckBoundingRectsBoundary
148  * @tc.desc: CheckBoundingRectsBoundary
149  * @tc.type: FUNC
150  */
151 HWTEST_F(DisplayCutoutControllerTest, CheckBoundingRectsBoundary, Function | SmallTest | Level1)
152 {
153     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
154     DisplayId displayId = DisplayManagerServiceInner::GetInstance().GetDefaultDisplayId();
155     std::vector<DMRect> boundingRects;
156     controller->CheckBoundingRectsBoundary(displayId, boundingRects);
157     ASSERT_TRUE(boundingRects.empty());
158 }
159 
160 /**
161  * @tc.name: CalcCutoutBoundingRect
162  * @tc.desc: CalcCutoutBoundingRect success
163  * @tc.type: FUNC
164  */
165 HWTEST_F(DisplayCutoutControllerTest, CalcCutoutBoundingRect, Function | SmallTest | Level1)
166 {
167     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
168     std::string svgPath = "M 100,100 m -75,0 a 75,75 0 1,0 150,0 a 75,75 0 1,0 -150,0 z";
169     DMRect rect = controller->CalcCutoutBoundingRect(svgPath);
170     DMRect emptyRect = {0, 0, 0, 0};
171     ASSERT_NE(rect, emptyRect);
172 }
173 
174 /**
175  * @tc.name: TransferBoundingRectsByRotation01
176  * @tc.desc: TransferBoundingRectsByRotation empty
177  * @tc.type: FUNC
178  */
179 HWTEST_F(DisplayCutoutControllerTest, TransferBoundingRectsByRotation01, Function | SmallTest | Level1)
180 {
181     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
182     DisplayId id = 10;
183     std::vector<DMRect> boundingRects;
184     controller->TransferBoundingRectsByRotation(id, boundingRects);
185     ASSERT_TRUE(boundingRects.empty());
186 }
187 
188 /**
189  * @tc.name: TransferBoundingRectsByRotation02
190  * @tc.desc: TransferBoundingRectsByRotation empty
191  * @tc.type: FUNC
192  */
193 HWTEST_F(DisplayCutoutControllerTest, TransferBoundingRectsByRotation02, Function | SmallTest | Level1)
194 {
195     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
196     DisplayId id = 10;
197     std::vector<DMRect> emptyRects;
198     controller->boundingRects_[id] = emptyRects;
199     std::vector<DMRect> boundingRects;
200     controller->TransferBoundingRectsByRotation(id, boundingRects);
201     ASSERT_TRUE(boundingRects.empty());
202 }
203 
204 /**
205  * @tc.name: TransferBoundingRectsByRotation03
206  * @tc.desc: TransferBoundingRectsByRotation success
207  * @tc.type: FUNC
208  */
209 HWTEST_F(DisplayCutoutControllerTest, TransferBoundingRectsByRotation03, Function | SmallTest | Level1)
210 {
211     DisplayId id = 11;
212     ScreenId sid = 12;
213     std::string name = "abstract_display_test";
214     SupportedScreenModes modesInfo;
215     modesInfo.width_ = 2160;
216     modesInfo.height_ = 1600;
217     modesInfo.refreshRate_ = 60;
218     sptr<SupportedScreenModes> info = new SupportedScreenModes(modesInfo);
219     sptr<AbstractScreenController> absScreenController;
220     sptr<AbstractScreen> absScreen;
221     absScreenController = nullptr;
222     absScreen = new AbstractScreen(absScreenController, name, sid, 1);
223     auto displayController = DisplayManagerService::GetInstance().abstractDisplayController_;
224     ASSERT_NE(displayController, nullptr);
225 
226     sptr<AbstractDisplay> absDisplay = new AbstractDisplay(id, info, absScreen);
227     absDisplay->RequestRotation(Rotation::ROTATION_0);
228     displayController->abstractDisplayMap_.insert((std::make_pair(id, absDisplay)));
229 
230     sptr<DisplayCutoutController> controller = new DisplayCutoutController();
231     std::vector<DMRect> dmRects;
232     DMRect rect = {1, 1, 100, 100};
233     dmRects.emplace_back(rect);
234     controller->boundingRects_[id] = dmRects;
235     ASSERT_FALSE(controller->boundingRects_.count(id) == 0);
236     ASSERT_FALSE(controller->boundingRects_[id].empty());
237     ASSERT_NE(DisplayManagerServiceInner::GetInstance().GetDisplayById(id), nullptr);
238     std::vector<DMRect> boundingRects;
239     controller->TransferBoundingRectsByRotation(id, boundingRects);
240 
241     absScreenController = DisplayManagerService::GetInstance().abstractScreenController_;
242     ASSERT_NE(absScreenController, nullptr);
243     absScreenController->dmsScreenMap_.insert(std::make_pair(sid, absScreen));
244     absScreen->modes_.emplace_back(info);
245     absDisplay->RequestRotation(Rotation::ROTATION_180);
246     controller->TransferBoundingRectsByRotation(id, boundingRects);
247 
248     absDisplay->RequestRotation(Rotation::ROTATION_90);
249     controller->TransferBoundingRectsByRotation(id, boundingRects);
250 
251     absDisplay->RequestRotation(Rotation::ROTATION_270);
252     controller->TransferBoundingRectsByRotation(id, boundingRects);
253     ASSERT_FALSE(boundingRects.empty());
254     displayController->abstractDisplayMap_.clear();
255     absScreenController->dmsScreenMap_.clear();
256 }
257 }
258 } // Rosen
259 } // OHOS
260