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
124
125
126
127
|
// Copyright 2014 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.
#include "media/cast/net/rtp/packet_storage.h"
#include <stdint.h>
#include <algorithm>
#include <vector>
#include "base/test/simple_test_tick_clock.h"
#include "base/time/time.h"
#include "media/cast/cast_defines.h"
#include "testing/gmock/include/gmock/gmock.h"
namespace media {
namespace cast {
static const size_t kStoredFrames = 10;
// Generate |number_of_frames| and store into |*storage|.
// First frame has 1 packet, second frame has 2 packets, etc.
static void StoreFrames(size_t number_of_frames,
uint32 first_frame_id,
PacketStorage* storage) {
const base::TimeTicks kTicks;
const int kSsrc = 1;
for (size_t i = 0; i < number_of_frames; ++i) {
SendPacketVector packets;
// First frame has 1 packet, second frame has 2 packets, etc.
const size_t kNumberOfPackets = i + 1;
for (size_t j = 0; j < kNumberOfPackets; ++j) {
Packet test_packet(1, 0);
packets.push_back(
std::make_pair(
PacedPacketSender::MakePacketKey(kTicks, kSsrc, j),
new base::RefCountedData<Packet>(test_packet)));
}
storage->StoreFrame(first_frame_id, packets);
++first_frame_id;
}
}
TEST(PacketStorageTest, NumberOfStoredFrames) {
PacketStorage storage;
uint32 frame_id = 0;
frame_id = ~frame_id; // The maximum value of uint32.
StoreFrames(kMaxUnackedFrames / 2, frame_id, &storage);
EXPECT_EQ(static_cast<size_t>(kMaxUnackedFrames / 2),
storage.GetNumberOfStoredFrames());
}
TEST(PacketStorageTest, GetFrameWrapAround8bits) {
PacketStorage storage;
const uint32 kFirstFrameId = 250;
StoreFrames(kStoredFrames, kFirstFrameId, &storage);
EXPECT_EQ(std::min<size_t>(kMaxUnackedFrames, kStoredFrames),
storage.GetNumberOfStoredFrames());
// Expect we get the correct frames by looking at the number of
// packets.
uint32 frame_id = kFirstFrameId;
for (size_t i = 0; i < kStoredFrames; ++i) {
ASSERT_TRUE(storage.GetFrame8(frame_id));
EXPECT_EQ(i + 1, storage.GetFrame8(frame_id)->size());
++frame_id;
}
}
TEST(PacketStorageTest, GetFrameWrapAround32bits) {
PacketStorage storage;
// First frame ID is close to the maximum value of uint32.
uint32 first_frame_id = 0xffffffff - 5;
StoreFrames(kStoredFrames, first_frame_id, &storage);
EXPECT_EQ(std::min<size_t>(kMaxUnackedFrames, kStoredFrames),
storage.GetNumberOfStoredFrames());
// Expect we get the correct frames by looking at the number of
// packets.
uint32 frame_id = first_frame_id;
for (size_t i = 0; i < kStoredFrames; ++i) {
ASSERT_TRUE(storage.GetFrame8(frame_id));
EXPECT_EQ(i + 1, storage.GetFrame8(frame_id)->size());
++frame_id;
}
}
TEST(PacketStorageTest, FramesReleased) {
PacketStorage storage;
const uint32 kFirstFrameId = 0;
StoreFrames(5, kFirstFrameId, &storage);
EXPECT_EQ(std::min<size_t>(kMaxUnackedFrames, 5),
storage.GetNumberOfStoredFrames());
for (uint32 frame_id = kFirstFrameId; frame_id < kFirstFrameId + 5;
++frame_id) {
EXPECT_TRUE(storage.GetFrame8(frame_id));
}
storage.ReleaseFrame(kFirstFrameId + 2);
EXPECT_EQ(4u, storage.GetNumberOfStoredFrames());
EXPECT_FALSE(storage.GetFrame8(kFirstFrameId + 2));
storage.ReleaseFrame(kFirstFrameId + 0);
EXPECT_EQ(3u, storage.GetNumberOfStoredFrames());
EXPECT_FALSE(storage.GetFrame8(kFirstFrameId + 0));
storage.ReleaseFrame(kFirstFrameId + 3);
EXPECT_EQ(2u, storage.GetNumberOfStoredFrames());
EXPECT_FALSE(storage.GetFrame8(kFirstFrameId + 3));
storage.ReleaseFrame(kFirstFrameId + 4);
EXPECT_EQ(1u, storage.GetNumberOfStoredFrames());
EXPECT_FALSE(storage.GetFrame8(kFirstFrameId + 4));
storage.ReleaseFrame(kFirstFrameId + 1);
EXPECT_EQ(0u, storage.GetNumberOfStoredFrames());
EXPECT_FALSE(storage.GetFrame8(kFirstFrameId + 1));
}
} // namespace cast
} // namespace media
|
