From d38ab2fab8509b77ade0f696cfc44b77397c7cbf Mon Sep 17 00:00:00 2001
From: liujie129 <liujie129@huawei.com>
Date: Thu, 17 Aug 2023 21:45:34 +0800
Subject: [PATCH] =?UTF-8?q?ut=E6=B5=8B=E8=AF=95=E7=94=A8=E4=BE=8B=E8=A1=A5?=
 =?UTF-8?q?=E5=85=85=20downloadRequest?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Signed-off-by: liujie129 <liujie129@huawei.com>
---
 test/unittest/BUILD.gn                 |  2 +
 test/unittest/TestHttpSourcePlugin.cpp | 64 ++++++++++++++++++++++++++
 2 files changed, 66 insertions(+)
 create mode 100644 test/unittest/TestHttpSourcePlugin.cpp

diff --git a/test/unittest/BUILD.gn b/test/unittest/BUILD.gn
index 8520be9c..934db774 100644
--- a/test/unittest/BUILD.gn
+++ b/test/unittest/BUILD.gn
@@ -51,6 +51,7 @@ ohos_unittest("histreamer_unit_test") {
     "$histreamer_root_dir/engine/plugin/plugins/ffmpeg_adapter:ffmpeg_video_encoders",
     "$histreamer_root_dir/engine/plugin/plugins/source/audio_capture:histreamer_plugin_StdAudioCapture",
     "$histreamer_root_dir/engine/plugin/plugins/source/file_source:filesource",
+    "$histreamer_root_dir/engine/plugin/plugins/source/http_source:httpsource",
     "$histreamer_root_dir/engine/scene:std_engine_factory",
     "$histreamer_root_dir/engine/scene/player:histreamer_player",
     "//third_party/googletest:gtest_rtti",
@@ -86,6 +87,7 @@ ohos_unittest("histreamer_unit_test") {
     "./TestFileSourcePlugin.cpp",
     "./TestFilter.cpp",
     "./TestHiPlayer.cpp",
+    "./TestHttpSourcePlugin.cpp",
     "./TestMeta.cpp",
     "./TestMimeDefs.cpp",
     "./TestPipline.cpp",
diff --git a/test/unittest/TestHttpSourcePlugin.cpp b/test/unittest/TestHttpSourcePlugin.cpp
new file mode 100644
index 00000000..8ef74bd4
--- /dev/null
+++ b/test/unittest/TestHttpSourcePlugin.cpp
@@ -0,0 +1,64 @@
+/*
+ * Copyright (c) 2023-2023 Huawei Device Co., Ltd.
+ * 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 <string>
+#include "gtest/gtest.h"
+#include "plugin/plugins/source/http_source/download/downloader.h"
+
+namespace OHOS {
+namespace Media {
+namespace Test {
+using namespace OHOS::Media::Plugin::HttpPlugin;
+using namespace testing::ext;
+
+HWTEST(HttpSourcePluginTest, test_download_request_save_header, TestSize.Level1)
+{
+    std::shared_ptr<HeaderInfo> headerInfo = std::make_shared<HeaderInfo>();
+
+    const size_t fileContentLen = 20;
+    headerInfo->fileContentLen = fileContentLen;
+
+    const char contentType[] = "audio";
+    strcpy_s(headerInfo->contentType, sizeof(headerInfo->contentType), contentType);
+
+    headerInfo->isChunked = true;
+    headerInfo->isClosed = true;
+
+    DataSaveFunc dataSaveFunc = [](uint8_t*, uint32_t) {
+        return false;
+    };
+
+    StatusCallbackFunc statusCallbackFunc = [](DownloadStatus, std::shared_ptr<Downloader>&,
+        std::shared_ptr<DownloadRequest>&) {};
+
+    const std::string url = "http://test";
+    DownloadRequest downloadRequest{url, dataSaveFunc, statusCallbackFunc};
+    downloadRequest.SaveHeader(headerInfo.get());
+
+    EXPECT_EQ(fileContentLen, downloadRequest.GetFileContentLength());
+
+    EXPECT_EQ(url, downloadRequest.GetUrl());
+
+    EXPECT_EQ(true, downloadRequest.IsChunked());
+
+    EXPECT_EQ(false, downloadRequest.IsClosed());
+
+    downloadRequest.Close();
+
+    EXPECT_EQ(true, downloadRequest.IsClosed());
+}
+} // namespace Test
+} // namespace Media
+} // namespace OHOS
-- 
Gitee