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.
7 #include "base/basictypes.h"
8 #include "base/memory/scoped_ptr.h"
9 #include "media/formats/common/offset_byte_queue.h"
10 #include "testing/gtest/include/gtest/gtest.h"
14 class OffsetByteQueueTest : public testing::Test {
16 virtual void SetUp() OVERRIDE {
18 for (int i = 0; i < 256; i++) {
21 queue_.reset(new OffsetByteQueue);
22 queue_->Push(buf, sizeof(buf));
23 queue_->Push(buf, sizeof(buf));
26 // Queue will start with 128 bytes of data and an offset of 384 bytes.
27 // These values are used throughout the test.
31 scoped_ptr<OffsetByteQueue> queue_;
34 TEST_F(OffsetByteQueueTest, SetUp) {
35 EXPECT_EQ(384, queue_->head());
36 EXPECT_EQ(512, queue_->tail());
41 queue_->Peek(&buf, &size);
43 EXPECT_EQ(128, buf[0]);
44 EXPECT_EQ(255, buf[size-1]);
47 TEST_F(OffsetByteQueueTest, PeekAt) {
51 queue_->PeekAt(400, &buf, &size);
52 EXPECT_EQ(queue_->tail() - 400, size);
53 EXPECT_EQ(400 - 256, buf[0]);
55 queue_->PeekAt(512, &buf, &size);
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());
66 EXPECT_TRUE(queue_->Trim(400));
67 EXPECT_EQ(400, queue_->head());
68 EXPECT_EQ(512, queue_->tail());
72 queue_->PeekAt(400, &buf, &size);
73 EXPECT_EQ(queue_->tail() - 400, size);
74 EXPECT_EQ(400 - 256, buf[0]);
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);
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));
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));