1 /*
2 * Copyright (c) 2014 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 #include <string.h>
12
13 #include "testing/gtest/include/gtest/gtest.h"
14 #include "webrtc/base/scoped_ptr.h"
15 #include "webrtc/test/rtp_file_reader.h"
16 #include "webrtc/test/rtp_file_writer.h"
17 #include "webrtc/test/testsupport/fileutils.h"
18
19 namespace webrtc {
20
21 class RtpFileWriterTest : public ::testing::Test {
22 public:
Init(const std::string & filename)23 void Init(const std::string& filename) {
24 filename_ = test::OutputPath() + filename;
25 rtp_writer_.reset(
26 test::RtpFileWriter::Create(test::RtpFileWriter::kRtpDump, filename_));
27 }
28
WriteRtpPackets(int num_packets)29 void WriteRtpPackets(int num_packets) {
30 ASSERT_TRUE(rtp_writer_.get() != NULL);
31 test::RtpPacket packet;
32 for (int i = 1; i <= num_packets; ++i) {
33 packet.length = i;
34 packet.original_length = i;
35 packet.time_ms = i;
36 memset(packet.data, i, packet.length);
37 EXPECT_TRUE(rtp_writer_->WritePacket(&packet));
38 }
39 }
40
CloseOutputFile()41 void CloseOutputFile() { rtp_writer_.reset(); }
42
VerifyFileContents(int expected_packets)43 void VerifyFileContents(int expected_packets) {
44 ASSERT_TRUE(rtp_writer_.get() == NULL)
45 << "Must call CloseOutputFile before VerifyFileContents";
46 rtc::scoped_ptr<test::RtpFileReader> rtp_reader(
47 test::RtpFileReader::Create(test::RtpFileReader::kRtpDump, filename_));
48 ASSERT_TRUE(rtp_reader.get() != NULL);
49 test::RtpPacket packet;
50 int i = 0;
51 while (rtp_reader->NextPacket(&packet)) {
52 ++i;
53 EXPECT_EQ(static_cast<size_t>(i), packet.length);
54 EXPECT_EQ(static_cast<size_t>(i), packet.original_length);
55 EXPECT_EQ(static_cast<uint32_t>(i), packet.time_ms);
56 for (int j = 0; j < i; ++j) {
57 EXPECT_EQ(i, packet.data[j]);
58 }
59 }
60 EXPECT_EQ(expected_packets, i);
61 }
62
63 private:
64 rtc::scoped_ptr<test::RtpFileWriter> rtp_writer_;
65 std::string filename_;
66 };
67
TEST_F(RtpFileWriterTest,WriteToRtpDump)68 TEST_F(RtpFileWriterTest, WriteToRtpDump) {
69 Init("test_rtp_file_writer.rtp");
70 WriteRtpPackets(10);
71 CloseOutputFile();
72 VerifyFileContents(10);
73 }
74
75 } // namespace webrtc
76