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 SOURCE_TEST_H 17 #define SOURCE_TEST_H 18 19 #include "gtest/gtest.h" 20 #include "source_plugin_mock.h" 21 #include "media_source_mock.h" 22 #include "callback_mock.h" 23 #include "source.h" 24 25 namespace OHOS { 26 namespace Media { 27 28 class MockMediaSource; 29 class SourceTest : public testing::Test { 30 public: 31 // SetUpTestCase: Called before all test cases SetUpTestCase(void)32 static void SetUpTestCase(void) { } 33 // TearDownTestCase: Called after all test case TearDownTestCase(void)34 static void TearDownTestCase(void) { } 35 // SetUp: Called before each test cases 36 void SetUp(void); 37 // TearDown: Called after each test cases 38 void TearDown(void); 39 protected: 40 std::shared_ptr<Source> source_ = nullptr; 41 std::shared_ptr<MockCallback> mockCallback_ = nullptr; 42 std::shared_ptr<MockMediaSource> mockMediaSource_ = nullptr; 43 std::shared_ptr<MockSourcePlugin> mockSourcePlugin_ = nullptr; 44 }; 45 } // namespace Media 46 } // namespace OHOS 47 #endif // SOURCE_TEST_H