1 /*
2 * Copyright (c) 2023-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 #include "gtest/gtest.h"
17 #include "plugin/plugins/sink/video_surface_sink/surface_sink_plugin.h"
18 #include "plugin_utils.h"
19 #include "ui/rs_surface_node.h"
20 #include "wm/window.h"
21 #include "window_option.h"
22 #include "display/composer/v1_2/display_composer_type.h"
23
24 using namespace testing::ext;
25 using namespace OHOS::Media::Plugin;
26 using namespace OHOS::Media::Plugin::VidSurfaceSinkPlugin;
27 constexpr uint32_t DEFAULT_WIDTH = 640;
28 constexpr uint32_t DEFAULT_HEIGHT = 480;
29 constexpr uint32_t DEFAULT_BUFFER_NUM = 32;
30
31 namespace OHOS {
32 namespace Media {
33 namespace Test {
VideoSinkPluginCreate(const std::string & name)34 std::shared_ptr<VideoSinkPlugin> VideoSinkPluginCreate(const std::string& name)
35 {
36 return std::make_shared<SurfaceSinkPlugin>(name);
37 }
38
GetVideoSurface()39 sptr<Surface> GetVideoSurface()
40 {
41 sptr<Rosen::WindowOption> option = new Rosen::WindowOption();
42 option->SetWindowRect({ 0, 0, 640, 480 });
43 option->SetWindowType(Rosen::WindowType::WINDOW_TYPE_APP_LAUNCHING);
44 option->SetWindowMode(Rosen::WindowMode::WINDOW_MODE_FLOATING);
45 sptr<Rosen::Window> window = Rosen::Window::Create("surface plugin unittest", option);
46 if (window == nullptr || window->GetSurfaceNode() == nullptr) {
47 return nullptr;
48 }
49 window->Show();
50 return window->GetSurfaceNode()->GetSurface();
51 }
52
53 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_create, TestSize.Level1)
54 {
55 std::shared_ptr<VideoSinkPlugin> plugin = VideoSinkPluginCreate("create");
56 ASSERT_TRUE(plugin != nullptr);
57 }
58
59 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_process, TestSize.Level1)
60 {
61 std::shared_ptr<VideoSinkPlugin> plugin = std::make_shared<SurfaceSinkPlugin>("process");
62 ASSERT_TRUE(plugin != nullptr);
63 sptr<Surface> surface = GetVideoSurface();
64 ASSERT_TRUE(surface != nullptr);
65 surface->SetUserData("SURFACE_FORMAT",
66 std::to_string(OHOS::HDI::Display::Composer::V1_2::PixelFormat::PIXEL_FMT_RGBA_8888));
67
68 plugin->SetParameter(Tag::VIDEO_SURFACE, surface);
69 auto initStatus = plugin->Init();
70 ASSERT_TRUE(initStatus == Status::OK);
71
72 auto prepareStatus = plugin->Prepare();
73 ASSERT_TRUE(prepareStatus == Status::OK);
74
75 auto allocator = plugin->GetAllocator();
76 ASSERT_TRUE(allocator != nullptr);
77
78 auto stopStatus = plugin->Stop();
79 ASSERT_TRUE(stopStatus == Status::OK);
80
81 auto freeStatus = plugin->Deinit();
82 ASSERT_TRUE(freeStatus == Status::OK);
83 }
84
85 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_reset, TestSize.Level1)
86 {
87 std::shared_ptr<VideoSinkPlugin> plugin = std::make_shared<SurfaceSinkPlugin>("reset");
88 ASSERT_TRUE(plugin != nullptr);
89 auto resetStatus = plugin->Reset();
90 ASSERT_TRUE(resetStatus == Status::OK);
91 }
92
93 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_start, TestSize.Level1)
94 {
95 std::shared_ptr<VideoSinkPlugin> plugin = std::make_shared<SurfaceSinkPlugin>("start");
96 ASSERT_TRUE(plugin != nullptr);
97 auto startStatus = plugin->Start();
98 ASSERT_TRUE(startStatus == Status::OK);
99 }
100
101 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_pause, TestSize.Level1)
102 {
103 std::shared_ptr<VideoSinkPlugin> plugin = std::make_shared<SurfaceSinkPlugin>("pause");
104 ASSERT_TRUE(plugin != nullptr);
105 auto pauseStatus = plugin->Pause();
106 ASSERT_TRUE(pauseStatus == Status::OK);
107 }
108
109 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_resume, TestSize.Level1)
110 {
111 std::shared_ptr<VideoSinkPlugin> plugin = std::make_shared<SurfaceSinkPlugin>("resume");
112 ASSERT_TRUE(plugin != nullptr);
113 auto resumeStatus = plugin->Resume();
114 ASSERT_TRUE(resumeStatus == Status::OK);
115 }
116
117 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_flush, TestSize.Level1)
118 {
119 std::shared_ptr<VideoSinkPlugin> plugin = std::make_shared<SurfaceSinkPlugin>("flush");
120 ASSERT_TRUE(plugin != nullptr);
121 auto flushStatus = plugin->Flush();
122 ASSERT_TRUE(flushStatus == Status::OK);
123 }
124
125 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_get_latency, TestSize.Level1)
126 {
127 std::shared_ptr<VideoSinkPlugin> plugin = std::make_shared<SurfaceSinkPlugin>("get latency");
128 ASSERT_TRUE(plugin != nullptr);
129 uint64_t nanoSec;
130 auto latencyStatus = plugin->GetLatency(nanoSec);
131 ASSERT_TRUE(latencyStatus == Status::OK);
132 }
133
134 HWTEST(TestSurfaceSinkPlugin, find_surface_sink_plugins_set_parameter, TestSize.Level1)
135 {
136 std::shared_ptr<VideoSinkPlugin> plugin = std::make_shared<SurfaceSinkPlugin>("set parameter");
137 ASSERT_TRUE(plugin != nullptr);
138 auto widthStatus = plugin->SetParameter(Tag::VIDEO_WIDTH, DEFAULT_WIDTH);
139 ASSERT_TRUE(widthStatus == Status::OK);
140 auto heightStatus = plugin->SetParameter(Tag::VIDEO_HEIGHT, DEFAULT_HEIGHT);
141 ASSERT_TRUE(heightStatus == Status::OK);
142 auto formatStatus = plugin->SetParameter(Tag::VIDEO_PIXEL_FORMAT, VideoPixelFormat::NV21);
143 ASSERT_TRUE(formatStatus == Status::OK);
144 auto surfaceNumStatus = plugin->SetParameter(Tag::VIDEO_MAX_SURFACE_NUM, DEFAULT_BUFFER_NUM);
145 ASSERT_TRUE(surfaceNumStatus == Status::OK);
146 }
147
148 } // namespace Test
149 } // namespace Media
150 } // namespace OHOS
151