--- /dev/null
+/*
+ * Copyright (c) 2022 Samsung Electronics Co., Ltd All Rights Reserved
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ * http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+#include <gst/gst.h>
+#include "ut_base.hpp"
+
+static GMainLoop* mainloop;
+
+using namespace std::chrono_literals;
+
+static void* _run_rtsp_sender(void* ptr) {
+ mtpr_h mtpr;
+ unsigned int src_id = 0;
+ mtpr_create(MTPR_CONNECTION_TYPE_RTSP_SENDER, &mtpr);
+ mtpr_set_sender_address(mtpr, "rtsp://127.0.0.1:8554/test");
+ mtpr_add_media_source(mtpr, MTPR_SOURCE_TYPE_VIDEOTEST, &src_id);
+ mtpr_start(mtpr);
+
+ mainloop = g_main_loop_new(NULL, FALSE);
+
+ LOGD("rtsp sender ready");
+ g_main_loop_run(mainloop);
+
+ LOGD("rtsp sender terminated");
+ g_main_loop_unref(mainloop);
+
+ mtpr_stop(mtpr);
+ mtpr_destroy(mtpr);
+
+ return NULL;
+}
+
+static void dataPacketCb(mtpr_h mtpr, mtpr_media_type_e type,
+ unsigned int id, media_packet_h packet, void *user_data) {
+ LOGD("callback is invoked. type: %d, id: %d", type, id);
+
+ static std::once_flag flag;
+
+ std::call_once(flag, [&](){
+ auto p = static_cast<std::promise<bool>*>(user_data);
+ assert(p);
+ p->set_value(true);
+ });
+}
+
+static bool waitDataPacket(std::future<bool>& f) {
+ LOGI("start waiting for packet arrival...");
+
+ if (f.wait_for(10s) != std::future_status::ready) {
+ LOGW("timeout(10s)!!!");
+ return false;
+ }
+
+ LOGI("ready to get the result!");
+ return f.get();
+}
+
+class MediaTransporterTestRtspReceiverWithSender : public MediaTransporterTestBase {
+public:
+ MediaTransporterTestRtspReceiverWithSender() = default;
+ ~MediaTransporterTestRtspReceiverWithSender() = default;
+
+ void SetUp() override {
+ LOGD("Enter");
+ start_rtsp_sender();
+
+ int ret = mtpr_create(MTPR_CONNECTION_TYPE_RTSP_RECEIVER, &_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ LOGD("Leave");
+ }
+
+ void TearDown() override {
+ int ret = MTPR_ERROR_NONE;
+ mtpr_state_e state = MTPR_STATE_IDLE;
+
+ LOGD("Enter");
+
+ if (_mtpr) {
+ ret = mtpr_get_state(_mtpr, &state);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ if (state != MTPR_STATE_IDLE) {
+ ret = mtpr_stop(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ }
+ ret = mtpr_destroy(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ }
+
+ stop_rtsp_sender();
+ LOGD("Leave");
+ }
+
+protected:
+ const std::string _senderPath = "rtsp://127.0.0.1:8554/test";
+
+ void start_rtsp_sender() {
+ _rtspSender = g_thread_try_new("rtsp_sender", _run_rtsp_sender, NULL, NULL);
+ std::this_thread::sleep_for(std::chrono::seconds(1));
+ }
+
+ void stop_rtsp_sender() {
+ if (mainloop)
+ g_main_loop_quit(mainloop);
+ if (_rtspSender)
+ g_thread_join(_rtspSender);
+ }
+
+ GThread* _rtspSender {nullptr};
+};
+
+class MediaTransporterTestRtspReceiver : public MediaTransporterTestBase {
+public:
+ MediaTransporterTestRtspReceiver() = default;
+ ~MediaTransporterTestRtspReceiver() = default;
+
+ void SetUp() override {
+ LOGD("Enter");
+
+ int ret = mtpr_create(MTPR_CONNECTION_TYPE_RTSP_RECEIVER, &_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ LOGD("Leave");
+ }
+
+ void TearDown() override {
+ int ret = MTPR_ERROR_NONE;
+ mtpr_state_e state = MTPR_STATE_IDLE;
+ LOGD("Enter");
+
+ if (_mtpr) {
+ ret = mtpr_get_state(_mtpr, &state);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ if (state != MTPR_STATE_IDLE) {
+ ret = mtpr_stop(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ }
+ ret = mtpr_destroy(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ }
+ LOGD("Leave");
+ }
+
+protected:
+ const std::string _senderPath = "rtsp://127.0.0.1:8554/test";
+};
+
+// MediaTransporterTestRtspReceiver
+
+TEST_F(MediaTransporterTestRtspReceiver, set_address_p)
+{
+ int ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, set_address_n1)
+{
+ int ret = MTPR_ERROR_NONE;
+
+ ret = mtpr_set_sender_address(NULL, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+
+ ret = mtpr_set_sender_address(_mtpr, NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+
+ ret = mtpr_set_sender_address(NULL, NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, set_address_n2)
+{
+ int ret = mtpr_set_receiver_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_OPERATION);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, get_address_p)
+{
+ int ret = MTPR_ERROR_NONE;
+ char *address = NULL;
+
+ ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_get_sender_address(_mtpr, &address);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ ASSERT_TRUE(address);
+ ASSERT_STREQ(_senderPath.c_str(), address);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, get_address_n1)
+{
+ int ret = MTPR_ERROR_NONE;
+ char *address = NULL;
+
+ ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_get_sender_address(NULL, &address);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+
+ ret = mtpr_get_sender_address(_mtpr, NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+
+ ret = mtpr_get_sender_address(NULL, NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, get_address_n2)
+{
+ char *address = NULL;
+
+ int ret = mtpr_get_sender_address(_mtpr, &address);
+ ASSERT_EQ(ret, MTPR_ERROR_NO_DATA);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, get_address_n3)
+{
+ int ret = MTPR_ERROR_NONE;
+ char *address = NULL;
+
+ ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_get_receiver_address(_mtpr, &address);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_OPERATION);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, get_connection_type_p)
+{
+ mtpr_connection_type_e type;
+
+ int ret = mtpr_get_connection_type(_mtpr, &type);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ ASSERT_EQ(type, MTPR_CONNECTION_TYPE_RTSP_RECEIVER);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, get_connection_type_n1)
+{
+ int ret = MTPR_ERROR_NONE;
+ mtpr_connection_type_e type;
+
+ ret = mtpr_get_connection_type(NULL, &type);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+
+ ret = mtpr_get_connection_type(_mtpr, NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+
+ ret = mtpr_get_connection_type(NULL, NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, start_n1)
+{
+ int ret = MTPR_ERROR_NONE;
+
+ ret = mtpr_start(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_OPERATION);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, start_n2)
+{
+ int ret = MTPR_ERROR_NONE;
+
+ ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_start(NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, stop_n)
+{
+ int ret = mtpr_stop(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_STATE);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, get_state_p)
+{
+ mtpr_state_e state = MTPR_STATE_IDLE;
+
+ int ret = mtpr_get_state(_mtpr, &state);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ ASSERT_EQ(state, MTPR_STATE_IDLE);
+}
+
+TEST_F(MediaTransporterTestRtspReceiver, get_state_n)
+{
+ int ret = MTPR_ERROR_NONE;
+ mtpr_state_e state = MTPR_STATE_IDLE;
+
+ ret = mtpr_get_state(NULL, &state);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+
+ ret = mtpr_get_state(_mtpr, NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+
+ ret = mtpr_get_state(NULL, NULL);
+ ASSERT_EQ(ret, MTPR_ERROR_INVALID_PARAMETER);
+}
+
+
+// MediaTransporterTestRtspReceiverWithSender
+
+TEST_F(MediaTransporterTestRtspReceiverWithSender, start_p)
+{
+ int ret = MTPR_ERROR_NONE;
+
+ ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ std::promise<bool> p;
+ std::future<bool> f = p.get_future();
+ ret = mtpr_set_video_packet_cb(_mtpr, dataPacketCb, &p);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_start(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ASSERT_TRUE(waitDataPacket(f));
+
+ ret = mtpr_unset_video_packet_cb(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+}
+
+TEST_F(MediaTransporterTestRtspReceiverWithSender, stop_p)
+{
+ int ret = MTPR_ERROR_NONE;
+
+ ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_start(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ std::this_thread::sleep_for(std::chrono::seconds(1));
+
+ ret = mtpr_stop(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+}
+
+TEST_F(MediaTransporterTestRtspReceiverWithSender, get_state_p1)
+{
+ int ret = MTPR_ERROR_NONE;
+ mtpr_state_e state = MTPR_STATE_IDLE;
+
+ ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_start(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_get_state(_mtpr, &state);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ ASSERT_EQ(state, MTPR_STATE_PLAYING);
+}
+
+TEST_F(MediaTransporterTestRtspReceiverWithSender, get_state_p2)
+{
+ int ret = MTPR_ERROR_NONE;
+ mtpr_state_e state = MTPR_STATE_IDLE;
+
+ ret = mtpr_set_sender_address(_mtpr, _senderPath.c_str());
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_start(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_stop(_mtpr);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+
+ ret = mtpr_get_state(_mtpr, &state);
+ ASSERT_EQ(ret, MTPR_ERROR_NONE);
+ ASSERT_EQ(state, MTPR_STATE_IDLE);
+}