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 AVMUXER_CAPI_MOCK_H
17 #define AVMUXER_CAPI_MOCK_H
18 
19 #include "avmuxer_mock.h"
20 #include "avformat_capi_mock.h"
21 #include "native_avmuxer.h"
22 
23 namespace OHOS {
24 namespace MediaAVCodec {
25 class AVMuxerCapiMock : public AVMuxerMock {
26 public:
AVMuxerCapiMock(OH_AVMuxer * muxer)27     explicit AVMuxerCapiMock(OH_AVMuxer *muxer) : muxer_(muxer) {}
28     ~AVMuxerCapiMock() = default;
29     int32_t Destroy() override;
30     int32_t Start() override;
31     int32_t Stop() override;
32     int32_t AddTrack(int32_t &trackIndex, std::shared_ptr<FormatMock> &trackFormat) override;
33     int32_t WriteSample(uint32_t trackIndex, const uint8_t *sample, const OH_AVCodecBufferAttr &info) override;
34     int32_t WriteSampleBuffer(uint32_t trackIndex, const OH_AVBuffer *sample) override;
35     int32_t SetRotation(int32_t rotation) override;
36     int32_t SetTimedMetadata() override;
37 private:
38     OH_AVMuxer *muxer_ = nullptr;
39 };
40 } // namespace MediaAVCodec
41 } // namespace OHOS
42 #endif // AVMUXER_CAPI_MOCK_H