1
    2
    3
    4
    5
    6
    7
    8
    9
   10
   11
   12
   13
   14
   15
   16
   17
   18
   19
   20
   21
   22
   23
   24
   25
   26
   27
   28
   29
   30
   31
   32
   33
   34
   35
   36
   37
   38
   39
   40
   41
   42
   43
   44
   45
   46
   47
   48
   49
   50
   51
   52
   53
   54
   55
   56
   57
   58
   59
   60
   61
   62
   63
   64
   65
   66
   67
   68
   69
   70
   71
   72
   73
   74
   75
   76
   77
   78
   79
   80
   81
   82
   83
   84
   85
   86
   87
   88
   89
   90
   91
   92
   93
   94
   95
   96
   97
   98
   99
  100
  101
  102
  103
  104
  105
  106
  107
  108
  109
  110
  111
  112
  113
  114
  115
  116
  117
  118
  119
  120
  121
  122
  123
  124
  125


media / base / data_buffer_unittest.cc [blame]

// Copyright 2012 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#include "media/base/data_buffer.h"

#include <stdint.h>

#include <memory>
#include <utility>

#include "base/strings/string_util.h"
#include "base/time/time.h"
#include "testing/gtest/include/gtest/gtest.h"

namespace media {

TEST(DataBufferTest, Constructor_ZeroCapacity) {
  // Zero-sized buffers are valid. In practice they aren't used very much but it
  // eliminates clients from worrying about null data pointers.
  scoped_refptr<DataBuffer> buffer = base::MakeRefCounted<DataBuffer>(0);
  EXPECT_TRUE(buffer->data().empty());
  EXPECT_TRUE(buffer->writable_data().empty());
  EXPECT_EQ(0u, buffer->size());
  EXPECT_EQ(0u, buffer->capacity());
  EXPECT_EQ(0u, buffer->data().size());
  EXPECT_EQ(0u, buffer->writable_data().size());
  EXPECT_FALSE(buffer->end_of_stream());
}

TEST(DataBufferTest, Constructor_NonZeroCapacity) {
  // Buffer size should be set.
  scoped_refptr<DataBuffer> buffer = base::MakeRefCounted<DataBuffer>(10);
  EXPECT_FALSE(buffer->data().empty());
  EXPECT_FALSE(buffer->writable_data().empty());
  EXPECT_EQ(0u, buffer->size());
  EXPECT_EQ(10u, buffer->capacity());
  EXPECT_EQ(10u, buffer->data().size());
  EXPECT_EQ(10u, buffer->writable_data().size());
  EXPECT_FALSE(buffer->end_of_stream());
}

TEST(DataBufferTest, Constructor_ScopedArray) {
  constexpr std::array<const uint8_t, 8> kTestData = {0x00, 0x11, 0x22, 0x33,
                                                      0x44, 0x55, 0x66, 0x77};

  // Data should be passed and both data and buffer size should be set.
  auto data = base::HeapArray<uint8_t>::CopiedFrom(kTestData);
  auto buffer = base::MakeRefCounted<DataBuffer>(std::move(data));

  EXPECT_EQ(kTestData.size(), buffer->size());
  EXPECT_EQ(kTestData.size(), buffer->capacity());
  EXPECT_EQ(kTestData, buffer->data());
  EXPECT_EQ(buffer->writable_data(), buffer->data());
  EXPECT_FALSE(buffer->end_of_stream());
}

TEST(DataBufferTest, CopyFrom) {
  constexpr const uint8_t kTestData[] = {0x00, 0x11, 0x22, 0x33,
                                         0x44, 0x55, 0x66, 0x77};
  const base::span<const uint8_t> kTestDataSpan(kTestData);

  scoped_refptr<DataBuffer> buffer = DataBuffer::CopyFrom(kTestDataSpan);
  EXPECT_EQ(kTestDataSpan.size(), buffer->size());
  EXPECT_FALSE(buffer->end_of_stream());

  // Ensure we are copying the data, not just pointing to the original data.
  EXPECT_EQ(buffer->data(), kTestDataSpan);
  buffer->writable_data()[0] = 0xFF;
  EXPECT_NE(buffer->data(), kTestDataSpan);
}

TEST(DataBufferTest, CreateEOSBuffer) {
  scoped_refptr<DataBuffer> buffer = DataBuffer::CreateEOSBuffer();
  EXPECT_TRUE(buffer->end_of_stream());
}

TEST(DataBufferTest, Timestamp) {
  constexpr base::TimeDelta kTimestampA = base::Microseconds(1337);
  constexpr base::TimeDelta kTimestampB = base::Microseconds(1234);

  auto buffer = base::MakeRefCounted<DataBuffer>(0);
  EXPECT_EQ(buffer->timestamp(), base::TimeDelta{});

  buffer->set_timestamp(kTimestampA);
  EXPECT_EQ(buffer->timestamp(), kTimestampA);

  buffer->set_timestamp(kTimestampB);
  EXPECT_EQ(buffer->timestamp(), kTimestampB);
}

TEST(DataBufferTest, Duration) {
  constexpr base::TimeDelta kDurationA = base::Microseconds(1337);
  constexpr base::TimeDelta kDurationB = base::Microseconds(1234);

  scoped_refptr<DataBuffer> buffer = base::MakeRefCounted<DataBuffer>(0);
  EXPECT_EQ(buffer->duration(), base::TimeDelta{});

  buffer->set_duration(kDurationA);
  EXPECT_EQ(buffer->duration(), kDurationA);

  buffer->set_duration(kDurationB);
  EXPECT_EQ(buffer->duration(), kDurationB);
}

TEST(DataBufferTest, DISABLED_ReadingWriting) {
  constexpr const char kData[] = "hello";
  constexpr const char kNewData[] = "chromium";
  const auto kDataSpan = base::byte_span_from_cstring(kData);
  const auto kNewDataSpan = base::byte_span_from_cstring(kNewData);

  // Create a DataBuffer.
  auto buffer = base::MakeRefCounted<DataBuffer>(kDataSpan.size());
  buffer->Append(kDataSpan);
  EXPECT_EQ(buffer->writable_data(), buffer->data());
  EXPECT_EQ(buffer->data(), kDataSpan);

  auto buffer2 = base::MakeRefCounted<DataBuffer>(kNewDataSpan.size() + 10);
  buffer2->Append(kNewDataSpan);
  EXPECT_EQ(kNewDataSpan.size(), buffer2->size());
  EXPECT_EQ(buffer2->writable_data(), buffer2->data());
  EXPECT_EQ(buffer2->data().first(buffer2->size()), kNewDataSpan);
}

}  // namespace media