1 /*
2  * Copyright (C) 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 #ifndef AVSOURCE_CAPI_MOCK_H
17 #define AVSOURCE_CAPI_MOCK_H
18 
19 #include "avsource_mock.h"
20 #include "avcodec_common.h"
21 #include "native_avsource.h"
22 
23 namespace OHOS {
24 namespace MediaAVCodec {
25 class AVSourceCapiMock : public AVSourceMock {
26 public:
AVSourceCapiMock(OH_AVSource * source)27     explicit AVSourceCapiMock(OH_AVSource *source) : source_(source) {}
28     ~AVSourceCapiMock() = default;
29     int32_t Destroy() override;
30     std::shared_ptr<FormatMock> GetSourceFormat() override;
31     std::shared_ptr<FormatMock> GetTrackFormat(uint32_t trackIndex) override;
32     std::shared_ptr<FormatMock> GetUserData() override;
33     OH_AVSource* GetAVSource();
34 private:
35     OH_AVSource* source_ = nullptr;
36 };
37 } // namespace MediaAVCodec
38 } // namespace OHOS
39 #endif // AVSOURCE_CAPI_MOCK_H