Improve input handling for WaveAudioHandler.

media/audio/simple_sources_unittest.cc

 // Copyright (c) 2012 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 <limits>
 
+#include "base/files/file_util.h"
 #include "base/logging.h"
 #include "base/basictypes.h"
 #include "base/memory/scoped_ptr.h"
 #include "media/audio/audio_parameters.h"
 #include "media/audio/simple_sources.h"
+#include "media/audio/sounds/test_data.h"
 #include "media/base/audio_bus.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace media {
 
 // Validate that the SineWaveAudioSource writes the expected values.
 TEST(SimpleSources, SineWaveAudioSource) {
   static const uint32 samples = 1024;
   static const uint32 bytes_per_sample = 2;
   static const int freq = 200;
@@ skipping 46 matching lines @@
 }
 
 TEST(SimpleSources, OnError) {
   SineWaveAudioSource source(1, 200, AudioParameters::kTelephoneSampleRate);
   source.OnError(NULL);
   EXPECT_EQ(1, source.errors());
   source.OnError(NULL);
   EXPECT_EQ(2, source.errors());
 }
 
+TEST(SimpleSources, FileSourceTestData) {
+  const int kNumFrames = 10;
+
+  // Create a temporary file filled with WAV data.
+  base::FilePath temp_path;
+  ASSERT_TRUE(base::CreateTemporaryFile(&temp_path));
+  base::File temp(temp_path,
+                  base::File::FLAG_WRITE | base::File::FLAG_OPEN_ALWAYS);
+  temp.WriteAtCurrentPos(kTestAudioData, kTestAudioDataSize);
+  ASSERT_EQ(kTestAudioDataSize, static_cast<size_t>(temp.GetLength()));
+  temp.Close();
+
+  // Create AudioParameters which match those in the WAV data.
+  AudioParameters params(AudioParameters::AUDIO_PCM_LINEAR,
+                         CHANNEL_LAYOUT_STEREO, 48000, 16, kNumFrames);
+  scoped_ptr<AudioBus> audio_bus = AudioBus::Create(2, kNumFrames);
+  audio_bus->Zero();
+
+  // Create a FileSource that reads this file.
+  FileSource source(params, temp_path);
+  EXPECT_EQ(kNumFrames, source.OnMoreData(audio_bus.get(), 0));
+
+  // Convert the test data (little-endian) into floats and compare.
+  const int kFirstSampleIndex = 12 + 8 + 16 + 8;
+  int16_t data[2];
+  data[0] = kTestAudioData[kFirstSampleIndex];
+  data[0] |= (kTestAudioData[kFirstSampleIndex + 1] << 8);
+  data[1] = kTestAudioData[kFirstSampleIndex + 2];
+  data[1] |= (kTestAudioData[kFirstSampleIndex + 3] << 8);
+
+  // The first frame should hold the WAV data.
+  EXPECT_FLOAT_EQ(static_cast<float>(data[0]) / ((1 << 15) - 1),
+                  audio_bus->channel(0)[0]);
+  EXPECT_FLOAT_EQ(static_cast<float>(data[1]) / ((1 << 15) - 1),
+                  audio_bus->channel(1)[0]);
+
+  // All other frames should be zero-padded.
+  for (int channel = 0; channel < audio_bus->channels(); ++channel) {
+    for (int frame = 1; frame < audio_bus->frames(); ++frame) {
+      EXPECT_FLOAT_EQ(0.0, audio_bus->channel(channel)[frame]);
+    }
+  }
+}
+
+TEST(SimpleSources, BadFilePathFails) {
+  AudioParameters params(AudioParameters::AUDIO_PCM_LINEAR,
+                         CHANNEL_LAYOUT_STEREO, 48000, 16, 10);
+  scoped_ptr<AudioBus> audio_bus = AudioBus::Create(2, 10);
+  audio_bus->Zero();
+
+  // Create a FileSource that reads this file.
+  base::FilePath path;
+  path = path.Append(FILE_PATH_LITERAL("does"))
+             .Append(FILE_PATH_LITERAL("not"))
+             .Append(FILE_PATH_LITERAL("exist"));
+  FileSource source(params, path);
+  EXPECT_EQ(0, source.OnMoreData(audio_bus.get(), 0));
+
+  // Confirm all frames are zero-padded.
+  for (int channel = 0; channel < audio_bus->channels(); ++channel) {
+    for (int frame = 0; frame < audio_bus->frames(); ++frame) {
+      EXPECT_FLOAT_EQ(0.0, audio_bus->channel(channel)[frame]);
+    }
+  }
+}
+
+TEST(SimpleSources, FileSourceCorruptTestDataFails) {
+  const int kNumFrames = 10;
+
+  // Create a temporary file filled with WAV data.
+  base::FilePath temp_path;
+  ASSERT_TRUE(base::CreateTemporaryFile(&temp_path));
+  base::File temp(temp_path,
+                  base::File::FLAG_WRITE | base::File::FLAG_OPEN_ALWAYS);
+  temp.WriteAtCurrentPos(kTestAudioData, kTestAudioDataSize);
+
+  // Corrupt the header.
+  temp.Write(3, "0x00", 1);
+
+  ASSERT_EQ(kTestAudioDataSize, static_cast<size_t>(temp.GetLength()));
+  temp.Close();
+
+  // Create AudioParameters which match those in the WAV data.
+  AudioParameters params(AudioParameters::AUDIO_PCM_LINEAR,
+                         CHANNEL_LAYOUT_STEREO, 48000, 16, kNumFrames);
+  scoped_ptr<AudioBus> audio_bus = AudioBus::Create(2, kNumFrames);
+  audio_bus->Zero();
+
+  // Create a FileSource that reads this file.
+  FileSource source(params, temp_path);
+  EXPECT_EQ(0, source.OnMoreData(audio_bus.get(), 0));
+
+  // Confirm all frames are zero-padded.
+  for (int channel = 0; channel < audio_bus->channels(); ++channel) {
+    for (int frame = 0; frame < audio_bus->frames(); ++frame) {
+      EXPECT_FLOAT_EQ(0.0, audio_bus->channel(channel)[frame]);
+    }
+  }
+}
+
 }  // namespace media