1 /*
2 * Copyright 2015 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 "rtc_base/buffer_queue.h"
12
13 #include <string.h>
14
15 #include "test/gtest.h"
16
17 namespace rtc {
18
TEST(BufferQueueTest,TestAll)19 TEST(BufferQueueTest, TestAll) {
20 const size_t kSize = 16;
21 const char in[kSize * 2 + 1] = "0123456789ABCDEFGHIJKLMNOPQRSTUV";
22 char out[kSize * 2];
23 size_t bytes;
24 BufferQueue queue1(1, kSize);
25 BufferQueue queue2(2, kSize);
26
27 // The queue is initially empty.
28 EXPECT_EQ(0u, queue1.size());
29 EXPECT_FALSE(queue1.ReadFront(out, kSize, &bytes));
30
31 // A write should succeed.
32 EXPECT_TRUE(queue1.WriteBack(in, kSize, &bytes));
33 EXPECT_EQ(kSize, bytes);
34 EXPECT_EQ(1u, queue1.size());
35
36 // The queue is full now (only one buffer allowed).
37 EXPECT_FALSE(queue1.WriteBack(in, kSize, &bytes));
38 EXPECT_EQ(1u, queue1.size());
39
40 // Reading previously written buffer.
41 EXPECT_TRUE(queue1.ReadFront(out, kSize, &bytes));
42 EXPECT_EQ(kSize, bytes);
43 EXPECT_EQ(0, memcmp(in, out, kSize));
44
45 // The queue is empty again now.
46 EXPECT_FALSE(queue1.ReadFront(out, kSize, &bytes));
47 EXPECT_EQ(0u, queue1.size());
48
49 // Reading only returns available data.
50 EXPECT_TRUE(queue1.WriteBack(in, kSize, &bytes));
51 EXPECT_EQ(kSize, bytes);
52 EXPECT_EQ(1u, queue1.size());
53 EXPECT_TRUE(queue1.ReadFront(out, kSize * 2, &bytes));
54 EXPECT_EQ(kSize, bytes);
55 EXPECT_EQ(0, memcmp(in, out, kSize));
56 EXPECT_EQ(0u, queue1.size());
57
58 // Reading maintains buffer boundaries.
59 EXPECT_TRUE(queue2.WriteBack(in, kSize / 2, &bytes));
60 EXPECT_EQ(1u, queue2.size());
61 EXPECT_TRUE(queue2.WriteBack(in + kSize / 2, kSize / 2, &bytes));
62 EXPECT_EQ(2u, queue2.size());
63 EXPECT_TRUE(queue2.ReadFront(out, kSize, &bytes));
64 EXPECT_EQ(kSize / 2, bytes);
65 EXPECT_EQ(0, memcmp(in, out, kSize / 2));
66 EXPECT_EQ(1u, queue2.size());
67 EXPECT_TRUE(queue2.ReadFront(out, kSize, &bytes));
68 EXPECT_EQ(kSize / 2, bytes);
69 EXPECT_EQ(0, memcmp(in + kSize / 2, out, kSize / 2));
70 EXPECT_EQ(0u, queue2.size());
71
72 // Reading truncates buffers.
73 EXPECT_TRUE(queue2.WriteBack(in, kSize / 2, &bytes));
74 EXPECT_EQ(1u, queue2.size());
75 EXPECT_TRUE(queue2.WriteBack(in + kSize / 2, kSize / 2, &bytes));
76 EXPECT_EQ(2u, queue2.size());
77 // Read first packet partially in too-small buffer.
78 EXPECT_TRUE(queue2.ReadFront(out, kSize / 4, &bytes));
79 EXPECT_EQ(kSize / 4, bytes);
80 EXPECT_EQ(0, memcmp(in, out, kSize / 4));
81 EXPECT_EQ(1u, queue2.size());
82 // Remainder of first packet is truncated, reading starts with next packet.
83 EXPECT_TRUE(queue2.ReadFront(out, kSize, &bytes));
84 EXPECT_EQ(kSize / 2, bytes);
85 EXPECT_EQ(0, memcmp(in + kSize / 2, out, kSize / 2));
86 EXPECT_EQ(0u, queue2.size());
87 }
88
89 } // namespace rtc
90