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
|
/*
* Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
*
* Use of this source code is governed by a BSD-style license
* that can be found in the LICENSE file in the root of the source
* tree. An additional intellectual property rights grant can be found
* in the file PATENTS. All contributing project authors may
* be found in the AUTHORS file in the root of the source tree.
*/
#include "framedrop_primitives.h"
#include <cstdio>
#include <vector>
#include "gtest/gtest.h"
#include "testsupport/fileutils.h"
#include "testsupport/frame_reader.h"
#include "testsupport/frame_writer.h"
namespace webrtc {
const std::string kOutputFilename = "temp_outputfile.tmp";
const int kFrameLength = 1000;
class FrameDropPrimitivesTest: public testing::Test {
protected:
FrameDropPrimitivesTest() {}
virtual ~FrameDropPrimitivesTest() {}
void SetUp() {
// Cleanup any previous output file.
std::remove(kOutputFilename.c_str());
}
void TearDown() {
// Cleanup the temporary file.
std::remove(kOutputFilename.c_str());
}
};
TEST_F(FrameDropPrimitivesTest, FixOutputFileForComparison) {
// Create test frame objects, where the second and fourth frame is marked
// as dropped at rendering.
std::vector<Frame*> frames;
Frame first_frame(0, kFrameLength);
Frame second_frame(0, kFrameLength);
Frame third_frame(0, kFrameLength);
Frame fourth_frame(0, kFrameLength);
second_frame.dropped_at_render = true;
fourth_frame.dropped_at_render = true;
frames.push_back(&first_frame);
frames.push_back(&second_frame);
frames.push_back(&third_frame);
frames.push_back(&fourth_frame);
// Prepare data for the first and third frames:
uint8_t first_frame_data[kFrameLength];
memset(first_frame_data, 5, kFrameLength); // Fill it with 5's to identify.
uint8_t third_frame_data[kFrameLength];
memset(third_frame_data, 7, kFrameLength); // Fill it with 7's to identify.
// Write the first and third frames to the temporary file. This means the fix
// method should add two frames of data by filling the file with data from
// the first and third frames after executing.
webrtc::test::FrameWriterImpl frame_writer(kOutputFilename, kFrameLength);
EXPECT_TRUE(frame_writer.Init());
EXPECT_TRUE(frame_writer.WriteFrame(first_frame_data));
EXPECT_TRUE(frame_writer.WriteFrame(third_frame_data));
frame_writer.Close();
EXPECT_EQ(2 * kFrameLength,
static_cast<int>(webrtc::test::GetFileSize(kOutputFilename)));
FixOutputFileForComparison(kOutputFilename, kFrameLength, frames);
// Verify that the output file has correct size.
EXPECT_EQ(4 * kFrameLength,
static_cast<int>(webrtc::test::GetFileSize(kOutputFilename)));
webrtc::test::FrameReaderImpl frame_reader(kOutputFilename, kFrameLength);
frame_reader.Init();
uint8_t read_buffer[kFrameLength];
EXPECT_TRUE(frame_reader.ReadFrame(read_buffer));
EXPECT_EQ(0, memcmp(read_buffer, first_frame_data, kFrameLength));
EXPECT_TRUE(frame_reader.ReadFrame(read_buffer));
EXPECT_EQ(0, memcmp(read_buffer, first_frame_data, kFrameLength));
EXPECT_TRUE(frame_reader.ReadFrame(read_buffer));
EXPECT_EQ(0, memcmp(read_buffer, third_frame_data, kFrameLength));
EXPECT_TRUE(frame_reader.ReadFrame(read_buffer));
EXPECT_EQ(0, memcmp(read_buffer, third_frame_data, kFrameLength));
frame_reader.Close();
}
} // namespace webrtc
|
