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) 2011 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 "testsupport/packet_reader.h"
#include "gtest/gtest.h"
#include "testsupport/unittest_utils.h"
namespace webrtc {
namespace test {
class PacketReaderTest: public PacketRelatedTest {
protected:
PacketReaderTest() {}
virtual ~PacketReaderTest() {}
void SetUp() {
reader_ = new PacketReader();
}
void TearDown() {
delete reader_;
}
void VerifyPacketData(int expected_length,
int actual_length,
WebRtc_UWord8* original_data_pointer,
WebRtc_UWord8* new_data_pointer) {
EXPECT_EQ(expected_length, actual_length);
EXPECT_EQ(*original_data_pointer, *new_data_pointer);
EXPECT_EQ(0, memcmp(original_data_pointer, new_data_pointer,
actual_length));
}
PacketReader* reader_;
};
// Test lack of initialization
TEST_F(PacketReaderTest, Uninitialized) {
WebRtc_UWord8* data_pointer = NULL;
EXPECT_EQ(-1, reader_->NextPacket(&data_pointer));
EXPECT_EQ(NULL, data_pointer);
}
TEST_F(PacketReaderTest, InitializeZeroLengthArgument) {
reader_->InitializeReading(packet_data_, 0, kPacketSizeInBytes);
ASSERT_EQ(0, reader_->NextPacket(&packet_data_pointer_));
}
// Test with something smaller than one packet
TEST_F(PacketReaderTest, NormalSmallData) {
const int kDataLengthInBytes = 1499;
WebRtc_UWord8 data[kDataLengthInBytes];
WebRtc_UWord8* data_pointer = data;
memset(data, 1, kDataLengthInBytes);
reader_->InitializeReading(data, kDataLengthInBytes, kPacketSizeInBytes);
int length_to_read = reader_->NextPacket(&data_pointer);
VerifyPacketData(kDataLengthInBytes, length_to_read, data, data_pointer);
EXPECT_EQ(0, data_pointer - data); // pointer hasn't moved
// Reading another one shall result in 0 bytes:
length_to_read = reader_->NextPacket(&data_pointer);
EXPECT_EQ(0, length_to_read);
EXPECT_EQ(kDataLengthInBytes, data_pointer - data);
}
// Test with data length that exactly matches one packet
TEST_F(PacketReaderTest, NormalOnePacketData) {
WebRtc_UWord8 data[kPacketSizeInBytes];
WebRtc_UWord8* data_pointer = data;
memset(data, 1, kPacketSizeInBytes);
reader_->InitializeReading(data, kPacketSizeInBytes, kPacketSizeInBytes);
int length_to_read = reader_->NextPacket(&data_pointer);
VerifyPacketData(kPacketSizeInBytes, length_to_read, data, data_pointer);
EXPECT_EQ(0, data_pointer - data); // pointer hasn't moved
// Reading another one shall result in 0 bytes:
length_to_read = reader_->NextPacket(&data_pointer);
EXPECT_EQ(0, length_to_read);
EXPECT_EQ(kPacketSizeInBytes, data_pointer - data);
}
// Test with data length that will result in 3 packets
TEST_F(PacketReaderTest, NormalLargeData) {
reader_->InitializeReading(packet_data_, kPacketDataLength,
kPacketSizeInBytes);
int length_to_read = reader_->NextPacket(&packet_data_pointer_);
VerifyPacketData(kPacketSizeInBytes, length_to_read,
packet1_, packet_data_pointer_);
length_to_read = reader_->NextPacket(&packet_data_pointer_);
VerifyPacketData(kPacketSizeInBytes, length_to_read,
packet2_, packet_data_pointer_);
length_to_read = reader_->NextPacket(&packet_data_pointer_);
VerifyPacketData(1u, length_to_read,
packet3_, packet_data_pointer_);
// Reading another one shall result in 0 bytes:
length_to_read = reader_->NextPacket(&packet_data_pointer_);
EXPECT_EQ(0, length_to_read);
EXPECT_EQ(kPacketDataLength, packet_data_pointer_ - packet_data_);
}
// Test with empty data.
TEST_F(PacketReaderTest, EmptyData) {
const int kDataLengthInBytes = 0;
WebRtc_UWord8* data = new WebRtc_UWord8[kDataLengthInBytes];
reader_->InitializeReading(data, kDataLengthInBytes, kPacketSizeInBytes);
EXPECT_EQ(kDataLengthInBytes, reader_->NextPacket(&data));
// Do it again to make sure nothing changes
EXPECT_EQ(kDataLengthInBytes, reader_->NextPacket(&data));
delete[] data;
}
} // namespace test
} // namespace webrtc
|