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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
|
// Copyright 2021 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "media/mojo/mojom/speech_recognition_result_mojom_traits.h"
#include <vector>
#include "base/time/time.h"
#include "media/mojo/mojom/speech_recognition_result.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace media {
namespace {
base::TimeDelta kZeroTime = base::Seconds(0);
}
TEST(SpeechRecognitionResultStructTraitsTest, NoTimingInformation) {
media::SpeechRecognitionResult result("hello world", true);
std::vector<uint8_t> data =
media::mojom::SpeechRecognitionResult::Serialize(&result);
media::SpeechRecognitionResult output;
EXPECT_TRUE(media::mojom::SpeechRecognitionResult::Deserialize(
std::move(data), &output));
EXPECT_EQ(result, output);
}
TEST(SpeechRecognitionResultStructTraitsTest, WithTimingInformation) {
media::SpeechRecognitionResult invalid_result("hello world", true);
invalid_result.timing_information = media::TimingInformation();
invalid_result.timing_information->audio_start_time = kZeroTime;
invalid_result.timing_information->audio_end_time = base::Seconds(-1);
std::vector<uint8_t> data =
media::mojom::SpeechRecognitionResult::Serialize(&invalid_result);
media::SpeechRecognitionResult output;
EXPECT_FALSE(media::mojom::SpeechRecognitionResult::Deserialize(
std::move(data), &output));
media::SpeechRecognitionResult valid_result("hello world", true);
valid_result.timing_information = media::TimingInformation();
valid_result.timing_information->audio_start_time = kZeroTime;
valid_result.timing_information->audio_end_time = base::Seconds(1);
std::vector<uint8_t> valid_data =
media::mojom::SpeechRecognitionResult::Serialize(&valid_result);
media::SpeechRecognitionResult valid_output;
EXPECT_TRUE(media::mojom::SpeechRecognitionResult::Deserialize(
std::move(valid_data), &valid_output));
EXPECT_EQ(valid_result, valid_output);
}
TEST(SpeechRecognitionResultStructTraitsTest,
PartialResultWithTimingInformation) {
media::SpeechRecognitionResult invalid_result("hello world", false);
invalid_result.timing_information = media::TimingInformation();
invalid_result.timing_information->audio_start_time = kZeroTime;
invalid_result.timing_information->audio_end_time = base::Seconds(1);
std::vector<uint8_t> invalid_data =
media::mojom::SpeechRecognitionResult::Serialize(&invalid_result);
media::SpeechRecognitionResult invalid_output;
// Partial results shouldn't have timing information.
EXPECT_FALSE(media::mojom::SpeechRecognitionResult::Deserialize(
std::move(invalid_data), &invalid_output));
}
TEST(SpeechRecognitionResultStructTraitsTest, WithInvalidHypothesisParts) {
media::SpeechRecognitionResult invalid_result("hello world", true);
invalid_result.timing_information = media::TimingInformation();
invalid_result.timing_information->audio_start_time = kZeroTime;
invalid_result.timing_information->audio_end_time = base::Seconds(1);
invalid_result.timing_information->hypothesis_parts =
std::vector<media::HypothesisParts>();
auto& hypothesis_parts =
invalid_result.timing_information->hypothesis_parts.value();
// Invalid hypothesis parts (outside the range audio_start_time to
// audio_end_time):
hypothesis_parts.emplace_back(std::vector<std::string>({"hello"}),
base::Seconds(-1));
hypothesis_parts.emplace_back(std::vector<std::string>({"world"}),
base::Seconds(2));
std::vector<uint8_t> data =
media::mojom::SpeechRecognitionResult::Serialize(&invalid_result);
media::SpeechRecognitionResult output;
EXPECT_FALSE(media::mojom::SpeechRecognitionResult::Deserialize(
std::move(data), &output));
}
TEST(SpeechRecognitionResultStructTraitsTest, WithValidHypothesisParts) {
media::SpeechRecognitionResult valid_result("hello world", true);
valid_result.timing_information = media::TimingInformation();
valid_result.timing_information->audio_start_time = kZeroTime;
valid_result.timing_information->audio_end_time = base::Seconds(1);
valid_result.timing_information->hypothesis_parts =
std::vector<media::HypothesisParts>();
auto& hypothesis_parts =
valid_result.timing_information->hypothesis_parts.value();
hypothesis_parts.emplace_back(std::vector<std::string>({"hello"}),
base::Seconds(0));
hypothesis_parts.emplace_back(std::vector<std::string>({"world"}),
base::Seconds(1));
std::vector<uint8_t> data =
media::mojom::SpeechRecognitionResult::Serialize(&valid_result);
media::SpeechRecognitionResult output;
EXPECT_TRUE(media::mojom::SpeechRecognitionResult::Deserialize(
std::move(data), &output));
EXPECT_EQ(valid_result, output);
}
} // namespace media
|
