1 files changed, 126 insertions, 0 deletions
diff --git a/media/base/buffer_queue_unittest.cc b/media/base/buffer_queue_unittest.cc
new file mode 100644
index 0000000..7f64b43
--- /dev/null
+++ b/media/base/buffer_queue_unittest.cc
@@ -0,0 +1,126 @@
+// Copyright (c) 2009 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 "base/scoped_ptr.h"
+#include "base/string_util.h"
+#include "media/base/buffer_queue.h"
+#include "media/base/data_buffer.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace media {
+
+class BufferQueueTest : public testing::Test {
+ protected:
+  BufferQueueTest() {
+    buffer1 = new DataBuffer();
+    data1 = buffer1->GetWritableData(kDataSize);
+
+    buffer2 = new DataBuffer();
+    data2 = buffer2->GetWritableData(kNewDataSize);
+
+    bufferBig = new DataBuffer();
+    dataBig = bufferBig->GetWritableData(2 * (kDataSize + kNewDataSize));
+
+    memcpy(data1, kData, kDataSize);
+    memcpy(data2, kNewData, kNewDataSize);
+  }
+
+  BufferQueue queue_;
+
+  static const char kData[];
+  static const size_t kDataSize;
+  static const char kNewData[];
+  static const size_t kNewDataSize;
+  scoped_refptr<DataBuffer> buffer1;
+  scoped_refptr<DataBuffer> buffer2;
+  scoped_refptr<DataBuffer> bufferBig;
+  uint8* data1;
+  uint8* data2;
+  uint8* dataBig;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(BufferQueueTest);
+};
+
+const char BufferQueueTest::kData[] = "hello";
+const size_t BufferQueueTest::kDataSize = arraysize(BufferQueueTest::kData);
+const char BufferQueueTest::kNewData[] = "chromium";
+const size_t BufferQueueTest::kNewDataSize =
+    arraysize(BufferQueueTest::kNewData);
+
+TEST_F(BufferQueueTest, ValidTestData) {
+  ASSERT_TRUE(kNewDataSize > kDataSize);
+  ASSERT_TRUE(buffer1.get());
+  ASSERT_TRUE(data1);
+  ASSERT_TRUE(buffer2.get());
+  ASSERT_TRUE(data2);
+  ASSERT_TRUE(bufferBig.get());
+  ASSERT_TRUE(dataBig);
+}
+
+TEST_F(BufferQueueTest, Ctor) {
+  EXPECT_TRUE(queue_.IsEmpty());
+  EXPECT_EQ(0u, queue_.SizeInBytes());
+}
+
+TEST_F(BufferQueueTest, Enqueue) {
+  queue_.Enqueue(buffer1.get());
+  EXPECT_FALSE(queue_.IsEmpty());
+  EXPECT_EQ(kDataSize, queue_.SizeInBytes());
+}
+
+TEST_F(BufferQueueTest, CopyWithOneBuffer) {
+  queue_.Enqueue(buffer1.get());
+
+  EXPECT_EQ(kDataSize, queue_.Copy(data2, kDataSize));
+  EXPECT_EQ(0, memcmp(data2, kData, kDataSize));
+}
+
+TEST_F(BufferQueueTest, Clear) {
+  queue_.Enqueue(buffer1.get());
+
+  queue_.Clear();
+  EXPECT_TRUE(queue_.IsEmpty());
+  EXPECT_EQ(0u, queue_.SizeInBytes());
+}
+
+TEST_F(BufferQueueTest, MultipleEnqueues) {
+  queue_.Enqueue(buffer2.get());
+  queue_.Enqueue(buffer1.get());
+  EXPECT_EQ(kDataSize + kNewDataSize, queue_.SizeInBytes());
+}
+
+TEST_F(BufferQueueTest, CopyWithMultipleBuffers) {
+  queue_.Enqueue(buffer2.get());
+  queue_.Enqueue(buffer1.get());
+
+  EXPECT_EQ(kDataSize + kNewDataSize,
+            queue_.Copy(dataBig, kDataSize + kNewDataSize));
+  EXPECT_EQ(0, memcmp(dataBig, kNewData, kNewDataSize));
+  dataBig += kNewDataSize;
+  EXPECT_EQ(0, memcmp(dataBig, kData, kDataSize));
+}
+
+TEST_F(BufferQueueTest, Consume) {
+  queue_.Enqueue(buffer2.get());
+  queue_.Enqueue(buffer1.get());
+
+  queue_.Consume(kDataSize);
+  EXPECT_EQ(kNewDataSize, queue_.SizeInBytes());
+}
+
+TEST_F(BufferQueueTest, CopyFromMiddleOfBuffer) {
+  queue_.Enqueue(buffer2.get());
+  queue_.Enqueue(buffer1.get());
+  queue_.Consume(kDataSize);
+
+  EXPECT_EQ(kNewDataSize, queue_.Copy(dataBig, kNewDataSize));
+  EXPECT_EQ(0, memcmp(dataBig,
+                      kNewData + kDataSize,
+                      kNewDataSize - kDataSize));
+  dataBig += (kNewDataSize - kDataSize);
+  EXPECT_EQ(0, memcmp(dataBig, kData, kDataSize));
+}
+
+}  // namespace media