1 // Copyright 2013 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include "content/shell/renderer/test_runner/MockWebSpeechRecognizer.h"
6
7 #include "base/logging.h"
8 #include "content/shell/renderer/test_runner/WebTestDelegate.h"
9 #include "third_party/WebKit/public/web/WebSpeechRecognitionResult.h"
10 #include "third_party/WebKit/public/web/WebSpeechRecognizerClient.h"
11
12 using namespace blink;
13 using namespace std;
14
15 namespace content {
16
17 namespace {
18
19 // Task class for calling a client function that does not take any parameters.
20 typedef void (WebSpeechRecognizerClient::*ClientFunctionPointer)(const WebSpeechRecognitionHandle&);
21 class ClientCallTask : public MockWebSpeechRecognizer::Task {
22 public:
ClientCallTask(MockWebSpeechRecognizer * mock,ClientFunctionPointer function)23 ClientCallTask(MockWebSpeechRecognizer* mock, ClientFunctionPointer function)
24 : MockWebSpeechRecognizer::Task(mock)
25 , m_function(function)
26 {
27 }
28
run()29 virtual void run() OVERRIDE { (m_recognizer->client()->*m_function)(m_recognizer->handle()); }
30
31 private:
32 ClientFunctionPointer m_function;
33 };
34
35 // Task for delivering a result event.
36 class ResultTask : public MockWebSpeechRecognizer::Task {
37 public:
ResultTask(MockWebSpeechRecognizer * mock,const WebString transcript,float confidence)38 ResultTask(MockWebSpeechRecognizer* mock, const WebString transcript, float confidence)
39 : MockWebSpeechRecognizer::Task(mock)
40 , m_transcript(transcript)
41 , m_confidence(confidence)
42 {
43 }
44
run()45 virtual void run() OVERRIDE
46 {
47 WebVector<WebString> transcripts(static_cast<size_t>(1));
48 WebVector<float> confidences(static_cast<size_t>(1));
49 transcripts[0] = m_transcript;
50 confidences[0] = m_confidence;
51 WebVector<WebSpeechRecognitionResult> finalResults(static_cast<size_t>(1));
52 WebVector<WebSpeechRecognitionResult> interimResults;
53 finalResults[0].assign(transcripts, confidences, true);
54
55 m_recognizer->client()->didReceiveResults(m_recognizer->handle(), finalResults, interimResults);
56 }
57
58 private:
59 WebString m_transcript;
60 float m_confidence;
61 };
62
63 // Task for delivering a nomatch event.
64 class NoMatchTask : public MockWebSpeechRecognizer::Task {
65 public:
NoMatchTask(MockWebSpeechRecognizer * mock)66 NoMatchTask(MockWebSpeechRecognizer* mock) : MockWebSpeechRecognizer::Task(mock) { }
run()67 virtual void run() OVERRIDE { m_recognizer->client()->didReceiveNoMatch(m_recognizer->handle(), WebSpeechRecognitionResult()); }
68 };
69
70 // Task for delivering an error event.
71 class ErrorTask : public MockWebSpeechRecognizer::Task {
72 public:
ErrorTask(MockWebSpeechRecognizer * mock,WebSpeechRecognizerClient::ErrorCode code,const WebString & message)73 ErrorTask(MockWebSpeechRecognizer* mock, WebSpeechRecognizerClient::ErrorCode code, const WebString& message)
74 : MockWebSpeechRecognizer::Task(mock)
75 , m_code(code)
76 , m_message(message)
77 {
78 }
79
run()80 virtual void run() OVERRIDE { m_recognizer->client()->didReceiveError(m_recognizer->handle(), m_message, m_code); }
81
82 private:
83 WebSpeechRecognizerClient::ErrorCode m_code;
84 WebString m_message;
85 };
86
87 } // namespace
88
MockWebSpeechRecognizer()89 MockWebSpeechRecognizer::MockWebSpeechRecognizer()
90 : m_wasAborted(false)
91 , m_taskQueueRunning(false)
92 , m_delegate(0)
93 {
94 }
95
~MockWebSpeechRecognizer()96 MockWebSpeechRecognizer::~MockWebSpeechRecognizer()
97 {
98 clearTaskQueue();
99 }
100
setDelegate(WebTestDelegate * delegate)101 void MockWebSpeechRecognizer::setDelegate(WebTestDelegate* delegate)
102 {
103 m_delegate = delegate;
104 }
105
start(const WebSpeechRecognitionHandle & handle,const WebSpeechRecognitionParams & params,WebSpeechRecognizerClient * client)106 void MockWebSpeechRecognizer::start(const WebSpeechRecognitionHandle& handle, const WebSpeechRecognitionParams& params, WebSpeechRecognizerClient* client)
107 {
108 m_wasAborted = false;
109 m_handle = handle;
110 m_client = client;
111
112 m_taskQueue.push_back(new ClientCallTask(this, &WebSpeechRecognizerClient::didStart));
113 m_taskQueue.push_back(new ClientCallTask(this, &WebSpeechRecognizerClient::didStartAudio));
114 m_taskQueue.push_back(new ClientCallTask(this, &WebSpeechRecognizerClient::didStartSound));
115
116 if (!m_mockTranscripts.empty()) {
117 DCHECK_EQ(m_mockTranscripts.size(), m_mockConfidences.size());
118
119 for (size_t i = 0; i < m_mockTranscripts.size(); ++i)
120 m_taskQueue.push_back(new ResultTask(this, m_mockTranscripts[i], m_mockConfidences[i]));
121
122 m_mockTranscripts.clear();
123 m_mockConfidences.clear();
124 } else
125 m_taskQueue.push_back(new NoMatchTask(this));
126
127 m_taskQueue.push_back(new ClientCallTask(this, &WebSpeechRecognizerClient::didEndSound));
128 m_taskQueue.push_back(new ClientCallTask(this, &WebSpeechRecognizerClient::didEndAudio));
129 m_taskQueue.push_back(new ClientCallTask(this, &WebSpeechRecognizerClient::didEnd));
130
131 startTaskQueue();
132 }
133
stop(const WebSpeechRecognitionHandle & handle,WebSpeechRecognizerClient * client)134 void MockWebSpeechRecognizer::stop(const WebSpeechRecognitionHandle& handle, WebSpeechRecognizerClient* client)
135 {
136 m_handle = handle;
137 m_client = client;
138
139 // FIXME: Implement.
140 NOTREACHED();
141 }
142
abort(const WebSpeechRecognitionHandle & handle,WebSpeechRecognizerClient * client)143 void MockWebSpeechRecognizer::abort(const WebSpeechRecognitionHandle& handle, WebSpeechRecognizerClient* client)
144 {
145 m_handle = handle;
146 m_client = client;
147
148 clearTaskQueue();
149 m_wasAborted = true;
150 m_taskQueue.push_back(new ClientCallTask(this, &WebSpeechRecognizerClient::didEnd));
151 startTaskQueue();
152 }
153
addMockResult(const WebString & transcript,float confidence)154 void MockWebSpeechRecognizer::addMockResult(const WebString& transcript, float confidence)
155 {
156 m_mockTranscripts.push_back(transcript);
157 m_mockConfidences.push_back(confidence);
158 }
159
setError(const WebString & error,const WebString & message)160 void MockWebSpeechRecognizer::setError(const WebString& error, const WebString& message)
161 {
162 WebSpeechRecognizerClient::ErrorCode code;
163 if (error == "OtherError")
164 code = WebSpeechRecognizerClient::OtherError;
165 else if (error == "NoSpeechError")
166 code = WebSpeechRecognizerClient::NoSpeechError;
167 else if (error == "AbortedError")
168 code = WebSpeechRecognizerClient::AbortedError;
169 else if (error == "AudioCaptureError")
170 code = WebSpeechRecognizerClient::AudioCaptureError;
171 else if (error == "NetworkError")
172 code = WebSpeechRecognizerClient::NetworkError;
173 else if (error == "NotAllowedError")
174 code = WebSpeechRecognizerClient::NotAllowedError;
175 else if (error == "ServiceNotAllowedError")
176 code = WebSpeechRecognizerClient::ServiceNotAllowedError;
177 else if (error == "BadGrammarError")
178 code = WebSpeechRecognizerClient::BadGrammarError;
179 else if (error == "LanguageNotSupportedError")
180 code = WebSpeechRecognizerClient::LanguageNotSupportedError;
181 else
182 return;
183
184 clearTaskQueue();
185 m_taskQueue.push_back(new ErrorTask(this, code, message));
186 m_taskQueue.push_back(new ClientCallTask(this, &WebSpeechRecognizerClient::didEnd));
187 startTaskQueue();
188 }
189
startTaskQueue()190 void MockWebSpeechRecognizer::startTaskQueue()
191 {
192 if (m_taskQueueRunning)
193 return;
194 m_delegate->postTask(new StepTask(this));
195 m_taskQueueRunning = true;
196 }
197
clearTaskQueue()198 void MockWebSpeechRecognizer::clearTaskQueue()
199 {
200 while (!m_taskQueue.empty()) {
201 delete m_taskQueue.front();
202 m_taskQueue.pop_front();
203 }
204 m_taskQueueRunning = false;
205 }
206
runIfValid()207 void MockWebSpeechRecognizer::StepTask::runIfValid()
208 {
209 if (m_object->m_taskQueue.empty()) {
210 m_object->m_taskQueueRunning = false;
211 return;
212 }
213
214 Task* task = m_object->m_taskQueue.front();
215 m_object->m_taskQueue.pop_front();
216 task->run();
217 delete task;
218
219 if (m_object->m_taskQueue.empty()) {
220 m_object->m_taskQueueRunning = false;
221 return;
222 }
223
224 m_object->m_delegate->postTask(new StepTask(m_object));
225 }
226
227 } // namespace content
228