1 /* 2 * Copyright (c) 2025 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 #ifndef DASH_SEGMENT_DOWNLOADER_UNITTEST_H 17 #define DASH_SEGMENT_DOWNLOADER_UNITTEST_H 18 19 #include "gtest/gtest.h" 20 #include "gmock/gmock.h" 21 #include "dash_segment_downloader.h" 22 23 namespace OHOS { 24 namespace Media { 25 namespace Plugins { 26 namespace HttpPlugin { 27 class DashSegmentDownloaderUnitTest : public testing::Test { 28 public: 29 // SetUpTestCase: Called before all test cases 30 static void SetUpTestCase(void); 31 // TearDownTestCase: Called after all test case 32 static void TearDownTestCase(void); 33 // SetUp: Called before each test cases 34 void SetUp(void); 35 // TearDown: Called after each test cases 36 void TearDown(void); 37 38 private: 39 std::shared_ptr<DashSegmentDownloader> segmentDownloader_ = nullptr; 40 }; 41 42 class MockCallback : public Callback { 43 public: 44 virtual ~MockCallback() = default; 45 46 MOCK_METHOD(void, OnEvent, (const PluginEvent &event), (override)); 47 MOCK_METHOD(void, OnDfxEvent, (const PluginDfxEvent &event), (override)); 48 MOCK_METHOD(void, SetSelectBitRateFlag, (bool flag, uint32_t desBitRate), (override)); 49 MOCK_METHOD(bool, CanAutoSelectBitRate, (), (override)); 50 }; 51 } // HttpPlugin 52 } // Plugins 53 } // Media 54 } // OHOS 55 #endif //DASH_SEGMENT_DOWNLOADER_UNITTEST_H 56