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_); |