src/ppapi/shared_impl/media_stream_buffer_manager_unittest.cc

   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 "base/memory/scoped_ptr.h"
   6 #include "base/memory/shared_memory.h"
   7 #include "ppapi/c/pp_errors.h"
   8 #include "ppapi/shared_impl/media_stream_buffer_manager.h"
   9 #include "testing/gtest/include/gtest/gtest.h"
  10
  11 using base::SharedMemory;
  12 using base::SharedMemoryCreateOptions;
  13
  14 namespace {
  15
  16 scoped_ptr<SharedMemory> CreateSharedMemory(int32_t buffer_size,
  17                                             int32_t number_of_buffers) {
  18   scoped_ptr<SharedMemory> shared_memory(new SharedMemory());
  19   SharedMemoryCreateOptions options;
  20   options.size = buffer_size * number_of_buffers;
  21   options.executable = false;
  22   EXPECT_TRUE(shared_memory->Create(options));
  23   return shared_memory.Pass();
  24 }
  25
  26 }  // namespace
  27
  28 namespace ppapi {
  29
  30 class MockDelegate : public MediaStreamBufferManager::Delegate {
  31  public:
  32   MockDelegate() : new_buffer_enqueue_counter_(0) {}
  33   virtual void OnNewBufferEnqueued() override {
  34     new_buffer_enqueue_counter_++;
  35   }
  36
  37   int32_t new_buffer_enqueue_counter_;
  38 };
  39
  40 TEST(MediaStreamBufferManager, General) {
  41   {
  42     const int32_t kNumberOfBuffers = 5;
  43     const int32_t kBufferSize = 128;
  44     MockDelegate delegate;
  45     MediaStreamBufferManager manager(&delegate);
  46     scoped_ptr<SharedMemory> shared_memory =
  47         CreateSharedMemory(kBufferSize, kNumberOfBuffers);
  48     // SetBuffers with enqueue_all_buffers = true;
  49     EXPECT_TRUE(manager.SetBuffers(kNumberOfBuffers,
  50                                    kBufferSize,
  51                                    shared_memory.Pass(),
  52                                    true));
  53
  54     int8_t* memory = reinterpret_cast<int8_t*>(manager.GetBufferPointer(0));
  55     EXPECT_NE(static_cast<int8_t*>(NULL), memory);
  56
  57     EXPECT_EQ(kNumberOfBuffers, manager.number_of_buffers());
  58     EXPECT_EQ(kBufferSize, manager.buffer_size());
  59
  60     // Test DequeueBuffer() and GetBufferPointer()
  61     for (int32_t i = 0; i < kNumberOfBuffers; ++i) {
  62       EXPECT_EQ(i, manager.DequeueBuffer());
  63       EXPECT_EQ(reinterpret_cast<MediaStreamBuffer*>(memory + i * kBufferSize),
  64                 manager.GetBufferPointer(i));
  65     }
  66
  67     manager.EnqueueBuffer(0);
  68     manager.EnqueueBuffer(4);
  69     manager.EnqueueBuffer(3);
  70     manager.EnqueueBuffer(1);
  71     manager.EnqueueBuffer(2);
  72     EXPECT_EQ(5, delegate.new_buffer_enqueue_counter_);
  73
  74     EXPECT_EQ(0, manager.DequeueBuffer());
  75     EXPECT_EQ(4, manager.DequeueBuffer());
  76     EXPECT_EQ(3, manager.DequeueBuffer());
  77     EXPECT_EQ(1, manager.DequeueBuffer());
  78     EXPECT_EQ(2, manager.DequeueBuffer());
  79     EXPECT_EQ(PP_ERROR_FAILED, manager.DequeueBuffer());
  80     EXPECT_EQ(PP_ERROR_FAILED, manager.DequeueBuffer());
  81
  82     // Returns NULL for invalid index to GetBufferPointer()
  83     EXPECT_EQ(NULL, manager.GetBufferPointer(-1));
  84     EXPECT_EQ(NULL, manager.GetBufferPointer(kNumberOfBuffers));
  85
  86     // Test crash for passing invalid index to EnqueueBuffer().
  87     EXPECT_DEATH(manager.EnqueueBuffer(-1),
  88                  ".*Check failed: index >= 0.*");
  89     EXPECT_DEATH(manager.EnqueueBuffer(kNumberOfBuffers),
  90                  ".*Check failed: index < number_of_buffers_.*");
  91   }
  92
  93   {
  94     const int32_t kNumberOfBuffers = 5;
  95     const int32_t kBufferSize = 128;
  96     MockDelegate delegate;
  97     MediaStreamBufferManager manager(&delegate);
  98     scoped_ptr<SharedMemory> shared_memory =
  99         CreateSharedMemory(kBufferSize, kNumberOfBuffers);
 100     // SetBuffers with enqueue_all_buffers = false;
 101     EXPECT_TRUE(manager.SetBuffers(kNumberOfBuffers,
 102                                    kBufferSize,
 103                                    shared_memory.Pass(),
 104                                    false));
 105
 106     int8_t* memory = reinterpret_cast<int8_t*>(manager.GetBufferPointer(0));
 107     EXPECT_NE(static_cast<int8_t*>(NULL), memory);
 108
 109     EXPECT_EQ(kNumberOfBuffers, manager.number_of_buffers());
 110     EXPECT_EQ(kBufferSize, manager.buffer_size());
 111
 112     // Test DequeueBuffer() and GetBufferPointer()
 113     for (int32_t i = 0; i < kNumberOfBuffers; ++i) {
 114       EXPECT_EQ(PP_ERROR_FAILED, manager.DequeueBuffer());
 115       EXPECT_EQ(reinterpret_cast<MediaStreamBuffer*>(memory + i * kBufferSize),
 116                 manager.GetBufferPointer(i));
 117     }
 118   }
 119 }
 120
 121 TEST(MediaStreamBufferManager, ResetBuffers) {
 122   const int32_t kNumberOfBuffers1 = 5;
 123   const int32_t kBufferSize1 = 128;
 124   const int32_t kNumberOfBuffers2 = 8;
 125   const int32_t kBufferSize2 = 256;
 126   MockDelegate delegate;
 127   MediaStreamBufferManager manager(&delegate);
 128   {
 129     scoped_ptr<SharedMemory> shared_memory(new SharedMemory());
 130     SharedMemoryCreateOptions options;
 131     options.size = kBufferSize1 * kNumberOfBuffers1;
 132     options.executable = false;
 133
 134     EXPECT_TRUE(shared_memory->Create(options));
 135
 136     // SetBuffers with enqueue_all_buffers = true;
 137     EXPECT_TRUE(manager.SetBuffers(kNumberOfBuffers1,
 138                                    kBufferSize1,
 139                                    shared_memory.Pass(),
 140                                    true));
 141
 142     int8_t* memory = reinterpret_cast<int8_t*>(manager.GetBufferPointer(0));
 143     EXPECT_NE(static_cast<int8_t*>(NULL), memory);
 144
 145     EXPECT_EQ(kNumberOfBuffers1, manager.number_of_buffers());
 146     EXPECT_EQ(kBufferSize1, manager.buffer_size());
 147
 148     // Test DequeueBuffer() and GetBufferPointer()
 149     for (int32_t i = 0; i < kNumberOfBuffers1; ++i) {
 150       EXPECT_EQ(i, manager.DequeueBuffer());
 151       EXPECT_EQ(reinterpret_cast<MediaStreamBuffer*>(memory + i * kBufferSize1),
 152                 manager.GetBufferPointer(i));
 153     }
 154   }
 155
 156   {
 157     scoped_ptr<SharedMemory> shared_memory =
 158         CreateSharedMemory(kBufferSize2, kNumberOfBuffers2);
 159     // SetBuffers with enqueue_all_buffers = true;
 160     EXPECT_TRUE(manager.SetBuffers(kNumberOfBuffers2,
 161                                    kBufferSize2,
 162                                    shared_memory.Pass(),
 163                                    true));
 164
 165     int8_t* memory = reinterpret_cast<int8_t*>(manager.GetBufferPointer(0));
 166     EXPECT_NE(static_cast<int8_t*>(NULL), memory);
 167
 168     EXPECT_EQ(kNumberOfBuffers2, manager.number_of_buffers());
 169     EXPECT_EQ(kBufferSize2, manager.buffer_size());
 170
 171     // Test DequeueBuffer() and GetBufferPointer()
 172     for (int32_t i = 0; i < kNumberOfBuffers2; ++i) {
 173       EXPECT_EQ(i, manager.DequeueBuffer());
 174       EXPECT_EQ(reinterpret_cast<MediaStreamBuffer*>(memory + i * kBufferSize2),
 175                 manager.GetBufferPointer(i));
 176     }
 177   }
 178 }
 179
 180 }  // namespace ppapi