1 files changed, 68 insertions, 0 deletions
diff --git a/dom/media/gtest/TestAudioBuffers.cpp b/dom/media/gtest/TestAudioBuffers.cpp
new file mode 100644
index 0000000000..b9f85a8194
--- /dev/null
+++ b/dom/media/gtest/TestAudioBuffers.cpp
@@ -0,0 +1,68 @@
+/* -*- Mode: C++; tab-width: 2; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
+/* This Source Code Form is subject to the terms of the Mozilla Public
+ * License, v. 2.0. If a copy of the MPL was not distributed with this file,
+ * You can obtain one at http://mozilla.org/MPL/2.0/. */
+
+#include <stdint.h>
+#include "AudioBufferUtils.h"
+#include "gtest/gtest.h"
+#include <vector>
+
+const uint32_t FRAMES = 256;
+
+void test_for_number_of_channels(const uint32_t channels) {
+  const uint32_t samples = channels * FRAMES;
+
+  mozilla::AudioCallbackBufferWrapper<float> mBuffer(channels);
+  mozilla::SpillBuffer<float, 128> b(channels);
+  std::vector<float> fromCallback(samples, 0.0);
+  std::vector<float> other(samples, 1.0);
+  mozilla::AudioChunk chunk;
+  chunk.mBufferFormat = mozilla::AUDIO_FORMAT_FLOAT32;
+  chunk.mChannelData.SetLength(channels);
+  for (uint32_t i = 0; i < channels; ++i) {
+    chunk.mChannelData[i] = other.data() + i * channels;
+  }
+
+  // Set the buffer in the wrapper from the callback
+  mBuffer.SetBuffer(fromCallback.data(), FRAMES);
+
+  // Fill the SpillBuffer with data.
+  chunk.mDuration = 15;
+  ASSERT_TRUE(b.Fill(chunk) == 15);
+  chunk.mDuration = 17;
+  ASSERT_TRUE(b.Fill(chunk) == 17);
+  for (uint32_t i = 0; i < 32 * channels; i++) {
+    other[i] = 0.0;
+  }
+
+  // Empty it in the AudioCallbackBufferWrapper
+  ASSERT_TRUE(b.Empty(mBuffer) == 32);
+
+  // Check available return something reasonnable
+  ASSERT_TRUE(mBuffer.Available() == FRAMES - 32);
+
+  // Fill the buffer with the rest of the data
+  mBuffer.WriteFrames(other.data() + 32 * channels, FRAMES - 32);
+
+  // Check the buffer is now full
+  ASSERT_TRUE(mBuffer.Available() == 0);
+
+  for (uint32_t i = 0; i < samples; i++) {
+    ASSERT_TRUE(fromCallback[i] == 1.0)
+    << "Difference at " << i << " (" << fromCallback[i] << " != " << 1.0
+    << ")\n";
+  }
+
+  chunk.mDuration = FRAMES;
+  ASSERT_TRUE(b.Fill(chunk) == 128);
+  ASSERT_TRUE(b.Fill(chunk) == 0);
+  ASSERT_TRUE(b.Empty(mBuffer) == 0);
+}
+
+TEST(AudioBuffers, Test)
+{
+  for (uint32_t ch = 1; ch <= 8; ++ch) {
+    test_for_number_of_channels(ch);
+  }
+}