OLD | NEW |
| (Empty) |
1 // Copyright 2014 The Chromium Authors. All rights reserved. | |
2 // Use of this source code is governed by a BSD-style license that can be | |
3 // found in the LICENSE file. | |
4 | |
5 #include <string.h> | |
6 | |
7 #include "base/basictypes.h" | |
8 #include "base/memory/scoped_ptr.h" | |
9 #include "media/formats/mp4/offset_byte_queue.h" | |
10 #include "testing/gtest/include/gtest/gtest.h" | |
11 | |
12 namespace media { | |
13 | |
14 class OffsetByteQueueTest : public testing::Test { | |
15 public: | |
16 virtual void SetUp() OVERRIDE { | |
17 uint8 buf[256]; | |
18 for (int i = 0; i < 256; i++) { | |
19 buf[i] = i; | |
20 } | |
21 queue_.reset(new OffsetByteQueue); | |
22 queue_->Push(buf, sizeof(buf)); | |
23 queue_->Push(buf, sizeof(buf)); | |
24 queue_->Pop(384); | |
25 | |
26 // Queue will start with 128 bytes of data and an offset of 384 bytes. | |
27 // These values are used throughout the test. | |
28 } | |
29 | |
30 protected: | |
31 scoped_ptr<OffsetByteQueue> queue_; | |
32 }; | |
33 | |
34 TEST_F(OffsetByteQueueTest, SetUp) { | |
35 EXPECT_EQ(384, queue_->head()); | |
36 EXPECT_EQ(512, queue_->tail()); | |
37 | |
38 const uint8* buf; | |
39 int size; | |
40 | |
41 queue_->Peek(&buf, &size); | |
42 EXPECT_EQ(128, size); | |
43 EXPECT_EQ(128, buf[0]); | |
44 EXPECT_EQ(255, buf[size-1]); | |
45 } | |
46 | |
47 TEST_F(OffsetByteQueueTest, PeekAt) { | |
48 const uint8* buf; | |
49 int size; | |
50 | |
51 queue_->PeekAt(400, &buf, &size); | |
52 EXPECT_EQ(queue_->tail() - 400, size); | |
53 EXPECT_EQ(400 - 256, buf[0]); | |
54 | |
55 queue_->PeekAt(512, &buf, &size); | |
56 EXPECT_EQ(NULL, buf); | |
57 EXPECT_EQ(0, size); | |
58 } | |
59 | |
60 TEST_F(OffsetByteQueueTest, Trim) { | |
61 EXPECT_TRUE(queue_->Trim(128)); | |
62 EXPECT_TRUE(queue_->Trim(384)); | |
63 EXPECT_EQ(384, queue_->head()); | |
64 EXPECT_EQ(512, queue_->tail()); | |
65 | |
66 EXPECT_TRUE(queue_->Trim(400)); | |
67 EXPECT_EQ(400, queue_->head()); | |
68 EXPECT_EQ(512, queue_->tail()); | |
69 | |
70 const uint8* buf; | |
71 int size; | |
72 queue_->PeekAt(400, &buf, &size); | |
73 EXPECT_EQ(queue_->tail() - 400, size); | |
74 EXPECT_EQ(400 - 256, buf[0]); | |
75 | |
76 // Trimming to the exact end of the buffer should return 'true'. This | |
77 // accomodates EOS cases. | |
78 EXPECT_TRUE(queue_->Trim(512)); | |
79 EXPECT_EQ(512, queue_->head()); | |
80 queue_->Peek(&buf, &size); | |
81 EXPECT_EQ(NULL, buf); | |
82 | |
83 // Trimming past the end of the buffer should return 'false'; we haven't seen | |
84 // the preceeding bytes. | |
85 EXPECT_FALSE(queue_->Trim(513)); | |
86 | |
87 // However, doing that shouldn't affect the EOS case. Only adding new data | |
88 // should alter this behavior. | |
89 EXPECT_TRUE(queue_->Trim(512)); | |
90 } | |
91 | |
92 } // namespace media | |
OLD | NEW |