1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "third_party/blink/renderer/modules/speech/speech_recognition_controller.h"
#include "third_party/blink/renderer/core/frame/local_frame.h"
#include "third_party/blink/renderer/core/testing/page_test_base.h"
#include "third_party/blink/renderer/modules/speech/speech_grammar_list.h"
namespace blink {
class SpeechRecognitionControllerTest : public PageTestBase {
public:
SpeechRecognitionControllerTest() = default;
SpeechRecognitionControllerTest(const SpeechRecognitionControllerTest&) =
delete;
SpeechRecognitionControllerTest& operator=(
const SpeechRecognitionControllerTest&) = delete;
~SpeechRecognitionControllerTest() override = default;
void SetUp() override {
PageTestBase::SetUp();
HeapVector<Member<SpeechRecognitionPhrase>> phrases;
phrases_ = MakeGarbageCollected<SpeechRecognitionPhraseList>(phrases);
controller_ = SpeechRecognitionController::From(*GetFrame().DomWindow());
}
void TearDown() override {
phrases_ = nullptr;
controller_ = nullptr;
PageTestBase::TearDown();
}
media::mojom::blink::StartSpeechRecognitionRequestParamsPtr BuildParams() {
SpeechGrammarList* grammars = MakeGarbageCollected<SpeechGrammarList>();
return controller_->BuildStartSpeechRecognitionRequestParams(
remote_.InitWithNewPipeAndPassReceiver(),
receiver_.InitWithNewPipeAndPassRemote(), *grammars, phrases_, "en-US",
/*continuous=*/true, /*interim_results=*/true, /*max_alternatives=*/5,
/*on_device=*/true, /*allow_cloud_fallback=*/true);
}
void SetSpeechRecognitionPhrases() {
HeapVector<Member<SpeechRecognitionPhrase>> phrases;
phrases.push_back(
MakeGarbageCollected<SpeechRecognitionPhrase>("text", 2.0));
phrases_ = MakeGarbageCollected<SpeechRecognitionPhraseList>(phrases);
}
private:
Persistent<SpeechRecognitionPhraseList> phrases_;
Persistent<SpeechRecognitionController> controller_;
mojo::PendingRemote<media::mojom::blink::SpeechRecognitionSession> remote_;
mojo::PendingReceiver<media::mojom::blink::SpeechRecognitionSessionClient>
receiver_;
};
TEST_F(SpeechRecognitionControllerTest, BuildParams) {
auto params = BuildParams();
EXPECT_TRUE(params->recognition_context.is_null());
EXPECT_EQ(params->language, "en-US");
EXPECT_EQ(params->max_hypotheses, (unsigned int)5);
EXPECT_TRUE(params->continuous);
EXPECT_TRUE(params->interim_results);
EXPECT_TRUE(params->on_device);
EXPECT_TRUE(params->allow_cloud_fallback);
EXPECT_TRUE(params->client.is_valid());
EXPECT_TRUE(params->session_receiver.is_valid());
EXPECT_FALSE(params->audio_forwarder.is_valid());
}
TEST_F(SpeechRecognitionControllerTest,
BuildParamsWithSpeechRecognitionPhrases) {
SetSpeechRecognitionPhrases();
auto params = BuildParams();
EXPECT_FALSE(params->recognition_context.is_null());
EXPECT_EQ(params->recognition_context->phrases.size(), (unsigned int)1);
EXPECT_EQ(params->recognition_context->phrases.at(0)->phrase, "text");
EXPECT_EQ(params->recognition_context->phrases.at(0)->boost, 2.0);
}
} // namespace blink
|