Chromium Code Reviews| Index: content/browser/speech/speech_recognizer_unittest.cc |
| diff --git a/content/browser/speech/speech_recognizer_unittest.cc b/content/browser/speech/speech_recognizer_unittest.cc |
| index 836539686a8a90901ff13d14d0f3e1605bf36e8f..1871383faa710ebc902c989b1443780c949f724c 100644 |
| --- a/content/browser/speech/speech_recognizer_unittest.cc |
| +++ b/content/browser/speech/speech_recognizer_unittest.cc |
| @@ -4,6 +4,7 @@ |
| #include <vector> |
| +#include "base/rand_util.h" |
| #include "chrome/common/net/test_url_fetcher_factory.h" |
| #include "content/browser/browser_thread.h" |
| #include "content/browser/speech/speech_recognizer.h" |
| @@ -62,8 +63,9 @@ class SpeechRecognizerTest : public SpeechRecognizerDelegate, |
| error_ = error; |
| } |
| - virtual void SetInputVolume(int caller_id, float volume) { |
| + virtual void SetInputVolume(int caller_id, float volume, float noise_volume) { |
| volume_ = volume; |
| + noise_volume_ = noise_volume; |
| } |
| // testing::Test methods. |
| @@ -83,6 +85,11 @@ class SpeechRecognizerTest : public SpeechRecognizerDelegate, |
| audio_packet_[i] = static_cast<uint8>(i); |
| } |
| + void FillPacketWithNoise() { |
| + for (size_t i = 0; i < audio_packet_.size(); ++i) |
| + audio_packet_[i] = static_cast<uint8>(base::RandInt(0, 50)); |
|
Satish
2011/03/01 16:44:17
Please ignore this part, I should be hardcoding th
bulach
2011/03/01 17:39:04
yay for deterministic tests! :)
|
| + } |
| + |
| protected: |
| MessageLoopForIO message_loop_; |
| BrowserThread io_thread_; |
| @@ -95,6 +102,7 @@ class SpeechRecognizerTest : public SpeechRecognizerDelegate, |
| TestAudioInputControllerFactory audio_input_controller_factory_; |
| std::vector<uint8> audio_packet_; |
| float volume_; |
| + float noise_volume_; |
| }; |
| TEST_F(SpeechRecognizerTest, StopNoData) { |
| @@ -272,6 +280,7 @@ TEST_F(SpeechRecognizerTest, SetInputVolumeCallback) { |
| // Feed some samples to begin with for the endpointer to do noise estimation. |
| int num_packets = SpeechRecognizer::kEndpointerEstimationTimeMs / |
| SpeechRecognizer::kAudioPacketIntervalMs; |
| + FillPacketWithNoise(); |
| for (int i = 0; i < num_packets; ++i) { |
| controller->event_handler()->OnData(controller, &audio_packet_[0], |
| audio_packet_.size()); |
| @@ -289,7 +298,8 @@ TEST_F(SpeechRecognizerTest, SetInputVolumeCallback) { |
| controller->event_handler()->OnData(controller, &audio_packet_[0], |
| audio_packet_.size()); |
| MessageLoop::current()->RunAllPending(); |
| - EXPECT_FLOAT_EQ(0.9f, volume_); |
| + EXPECT_FLOAT_EQ(0.81907868f, volume_); |
| + EXPECT_FLOAT_EQ(0.36962864f, noise_volume_); |
|
hans
2011/03/01 17:25:25
maybe make these magic numbers into constant varia
|
| EXPECT_EQ(SpeechRecognizer::RECOGNIZER_NO_ERROR, error_); |
| EXPECT_FALSE(recording_complete_); |