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
|
// Copyright 2022 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#import "components/translate/ios/browser/ios_translate_driver.h"
#import "components/language/core/browser/language_model.h"
#import "components/language/ios/browser/ios_language_detection_tab_helper.h"
#import "components/sync_preferences/testing_pref_service_syncable.h"
#import "components/translate/core/browser/mock_translate_client.h"
#import "components/translate/core/browser/mock_translate_driver.h"
#import "components/translate/core/browser/mock_translate_ranker.h"
#import "components/translate/core/browser/translate_manager.h"
#import "components/translate/core/browser/translate_pref_names.h"
#import "components/translate/ios/browser/translate_java_script_feature.h"
#import "ios/web/public/test/fakes/fake_browser_state.h"
#import "ios/web/public/test/fakes/fake_web_frames_manager.h"
#import "ios/web/public/test/fakes/fake_web_state.h"
#import "ios/web/public/test/web_task_environment.h"
#import "testing/platform_test.h"
using testing::_;
namespace {
class MockLanguageModel : public language::LanguageModel {
std::vector<language::LanguageModel::LanguageDetails> GetLanguages()
override {
return {language::LanguageModel::LanguageDetails("en", 1.0)};
}
};
} // namespace
namespace translate {
class IOSTranslateDriverTest : public PlatformTest {
protected:
IOSTranslateDriverTest()
: fake_browser_state_(std::make_unique<web::FakeBrowserState>()),
fake_web_state_(std::make_unique<web::FakeWebState>()) {
pref_service_.registry()->RegisterBooleanPref(prefs::kOfferTranslateEnabled,
true);
auto web_frames_manager = std::make_unique<web::FakeWebFramesManager>();
web::ContentWorld content_world =
TranslateJavaScriptFeature::GetInstance()->GetSupportedContentWorld();
fake_web_state_->SetWebFramesManager(content_world,
std::move(web_frames_manager));
language::IOSLanguageDetectionTabHelper::CreateForWebState(
fake_web_state_.get(), nullptr, nullptr, &pref_service_);
driver_ =
std::make_unique<IOSTranslateDriver>(fake_web_state_.get(), nullptr);
mock_translate_client_ =
std::make_unique<::testing::NiceMock<testing::MockTranslateClient>>(
driver_.get(), &pref_service_);
translate_manager_ = std::make_unique<TranslateManager>(
mock_translate_client_.get(), &mock_translate_ranker_,
&mock_language_model_);
driver_->Initialize(nullptr, translate_manager_.get());
}
protected:
web::WebTaskEnvironment task_environment_{
base::test::TaskEnvironment::TimeSource::MOCK_TIME};
std::unique_ptr<web::FakeBrowserState> fake_browser_state_;
sync_preferences::TestingPrefServiceSyncable pref_service_;
std::unique_ptr<web::FakeWebState> fake_web_state_;
std::unique_ptr<IOSTranslateDriver> driver_;
std::unique_ptr<::testing::NiceMock<testing::MockTranslateClient>>
mock_translate_client_;
MockLanguageModel mock_language_model_;
testing::MockTranslateRanker mock_translate_ranker_;
std::unique_ptr<TranslateManager> translate_manager_;
};
// Test that if there is a timeout, only the error is shown.
TEST_F(IOSTranslateDriverTest, TestTimeout) {
driver_->PrepareToTranslatePage(0, "en", "fr", true);
EXPECT_CALL(*mock_translate_client_,
ShowTranslateUI(TRANSLATE_STEP_AFTER_TRANSLATE, _, _,
TranslateErrors::TRANSLATION_TIMEOUT, _))
.WillOnce(::testing::Return(true));
EXPECT_CALL(*mock_translate_client_,
ShowTranslateUI(TRANSLATE_STEP_AFTER_TRANSLATE, _, _,
TranslateErrors::NONE, _))
.Times(0);
task_environment_.AdvanceClock(base::Seconds(10));
task_environment_.RunUntilIdle();
driver_->OnTranslateComplete(TranslateErrors::NONE, "en", 1.0);
}
// Test that if translation is a success, timeout is not triggered.
TEST_F(IOSTranslateDriverTest, TestNoTimeout) {
driver_->PrepareToTranslatePage(0, "en", "fr", true);
EXPECT_CALL(*mock_translate_client_,
ShowTranslateUI(TRANSLATE_STEP_AFTER_TRANSLATE, _, _,
TranslateErrors::NONE, _))
.WillOnce(::testing::Return(true));
EXPECT_CALL(*mock_translate_client_,
ShowTranslateUI(TRANSLATE_STEP_AFTER_TRANSLATE, _, _,
TranslateErrors::TRANSLATION_TIMEOUT, _))
.Times(0);
driver_->OnTranslateComplete(TranslateErrors::NONE, "en", 1.0);
task_environment_.AdvanceClock(base::Seconds(10));
task_environment_.RunUntilIdle();
}
} // namespace translate
|
