Checking in media::DataBuffer, a simple implementation of WritableBufferInterface.

media/base/data_buffer_unittest.cc

+// Copyright (c) 2006-2008 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/string_util.h"
+#include "media/base/data_buffer.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using media::DataBuffer;
+
+TEST(DataBufferTest, Basic) {
+  const size_t kBufferSize = 32;
+  const char kData[] = "hello";
+  const size_t kDataSize = arraysize(kData);
+  const char kNewData[] = "chromium";
+  const size_t kNewDataSize = arraysize(kNewData);
+
+  // Create our buffer and copy some data into it.
+  char* data = new char[kBufferSize];
+  ASSERT_TRUE(data);
+  size_t copied = base::strlcpy(data, kData, kBufferSize);
+  EXPECT_EQ(kDataSize, copied + 1);
+
+  // Create a DataBuffer.
+  scoped_refptr<DataBuffer> buffer;
+  buffer = new DataBuffer(data, kBufferSize, kDataSize, 1337, 1667);  
+  ASSERT_TRUE(buffer.get());
+
+  // Test StreamSampleInterface.
+  EXPECT_EQ(1337, buffer->GetTimestamp());
+  EXPECT_EQ(1667, buffer->GetDuration());
+  buffer->SetTimestamp(1234);
+  buffer->SetDuration(5678);
+  EXPECT_EQ(1234, buffer->GetTimestamp());
+  EXPECT_EQ(5678, buffer->GetDuration());
+
+  // Test BufferInterface.
+  ASSERT_EQ(data, buffer->GetData());
+  EXPECT_EQ(kDataSize, buffer->GetDataSize());
+  EXPECT_STREQ(kData, buffer->GetData());
+
+  // Test WritableBufferInterface.
+  ASSERT_EQ(data, buffer->GetWritableData());
+  EXPECT_EQ(kBufferSize, buffer->GetBufferSize());
+  copied = base::strlcpy(data, kNewData, kBufferSize);
+  EXPECT_EQ(kNewDataSize, copied + 1);
+  buffer->SetDataSize(kNewDataSize);
+  EXPECT_EQ(kNewDataSize, buffer->GetDataSize());
+}