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