1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
// Copyright (c) 2009 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef MEDIA_BASE_MOCK_FFMPEG_H_
#define MEDIA_BASE_MOCK_FFMPEG_H_
// TODO(scherkus): See if we can remove ffmpeg_common from this file.
#include "media/filters/ffmpeg_common.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace media {
class MockFFmpeg {
public:
MockFFmpeg();
virtual ~MockFFmpeg();
MOCK_METHOD0(AVCodecInit, void());
MOCK_METHOD1(AVRegisterProtocol, int(URLProtocol* protocol));
MOCK_METHOD0(AVRegisterAll, void());
MOCK_METHOD1(AVCodecFindDecoder, AVCodec*(enum CodecID id));
MOCK_METHOD2(AVCodecOpen, int(AVCodecContext* avctx, AVCodec* codec));
MOCK_METHOD1(AVCodecClose, int(AVCodecContext* avctx));
MOCK_METHOD2(AVCodecThreadInit, int(AVCodecContext* avctx, int threads));
MOCK_METHOD1(AVCodecFlushBuffers, void(AVCodecContext* avctx));
MOCK_METHOD0(AVCodecAllocFrame, AVFrame*());
MOCK_METHOD4(AVCodecDecodeVideo2,
int(AVCodecContext* avctx, AVFrame* picture,
int* got_picture_ptr, AVPacket* avpkt));
MOCK_METHOD5(AVOpenInputFile, int(AVFormatContext** format,
const char* filename,
AVInputFormat* input_format,
int buffer_size,
AVFormatParameters* parameters));
MOCK_METHOD1(AVCloseInputFile, void(AVFormatContext* format));
MOCK_METHOD1(AVFindStreamInfo, int(AVFormatContext* format));
MOCK_METHOD2(AVReadFrame, int(AVFormatContext* format, AVPacket* packet));
MOCK_METHOD4(AVSeekFrame, int(AVFormatContext *format,
int stream_index,
int64_t timestamp,
int flags));
MOCK_METHOD1(AVInitPacket, void(AVPacket* pkt));
MOCK_METHOD2(AVNewPacket, int(AVPacket* packet, int size));
MOCK_METHOD1(AVFreePacket, void(AVPacket* packet));
MOCK_METHOD1(AVFree, void(void* ptr));
// Used for verifying check points during tests.
MOCK_METHOD1(CheckPoint, void(int id));
// Setter/getter for the global instance of MockFFmpeg.
static void set(MockFFmpeg* instance);
static MockFFmpeg* get();
// Returns the URLProtocol registered by the FFmpegGlue singleton.
static URLProtocol* protocol();
// AVPacket destructor for packets allocated by av_new_packet().
static void DestructPacket(AVPacket* packet);
// Modifies the number of outstanding packets.
void inc_outstanding_packets();
void dec_outstanding_packets();
private:
static MockFFmpeg* instance_;
static URLProtocol* protocol_;
// Tracks the number of packets allocated by calls to av_read_frame() and
// av_free_packet(). We crash the unit test if this is not zero at time of
// destruction.
int outstanding_packets_;
};
// Used for simulating av_read_frame().
ACTION_P3(CreatePacket, stream_index, data, size) {
// Confirm we're dealing with AVPacket so we can safely const_cast<>.
::testing::StaticAssertTypeEq<AVPacket*, arg1_type>();
memset(arg1, 0, sizeof(*arg1));
arg1->stream_index = stream_index;
arg1->data = const_cast<uint8*>(data);
arg1->size = size;
// Increment number of packets allocated.
MockFFmpeg::get()->inc_outstanding_packets();
return 0;
}
// Used for simulating av_new_packet().
ACTION(NewPacket) {
::testing::StaticAssertTypeEq<AVPacket*, arg0_type>();
int size = arg1;
memset(arg0, 0, sizeof(*arg0));
arg0->data = new uint8[size];
arg0->size = size;
arg0->destruct = &MockFFmpeg::DestructPacket;
// Increment number of packets allocated.
MockFFmpeg::get()->inc_outstanding_packets();
return 0;
}
// Used for simulating av_free_packet().
ACTION(FreePacket) {
::testing::StaticAssertTypeEq<AVPacket*, arg0_type>();
// Call the destructor if present, such as the one assigned in NewPacket().
if (arg0->destruct) {
arg0->destruct(arg0);
}
// Decrement number of packets allocated.
MockFFmpeg::get()->dec_outstanding_packets();
}
} // namespace media
#endif // MEDIA_BASE_MOCK_FFMPEG_H_
|
