| 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..4b162598e5db2f7ff0d348dbcdc7c887541bd772 100644
|
| --- a/content/browser/speech/speech_recognizer_unittest.cc
|
| +++ b/content/browser/speech/speech_recognizer_unittest.cc
|
| @@ -62,8 +62,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 +84,15 @@ class SpeechRecognizerTest : public SpeechRecognizerDelegate,
|
| audio_packet_[i] = static_cast<uint8>(i);
|
| }
|
|
|
| + void FillPacketWithNoise() {
|
| + int value = 0;
|
| + int factor = 175;
|
| + for (size_t i = 0; i < audio_packet_.size(); ++i) {
|
| + value += factor;
|
| + audio_packet_[i] = value % 100;
|
| + }
|
| + }
|
| +
|
| protected:
|
| MessageLoopForIO message_loop_;
|
| BrowserThread io_thread_;
|
| @@ -95,6 +105,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 +283,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());
|
| @@ -283,13 +295,14 @@ TEST_F(SpeechRecognizerTest, SetInputVolumeCallback) {
|
| controller->event_handler()->OnData(controller, &audio_packet_[0],
|
| audio_packet_.size());
|
| MessageLoop::current()->RunAllPending();
|
| - EXPECT_EQ(0, volume_);
|
| + EXPECT_FLOAT_EQ(0.51877826f, volume_);
|
|
|
| FillPacketWithTestWaveform();
|
| 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.52143687f, noise_volume_);
|
|
|
| EXPECT_EQ(SpeechRecognizer::RECOGNIZER_NO_ERROR, error_);
|
| EXPECT_FALSE(recording_complete_);
|
|
|