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
|
// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#ifndef COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_INFOBAR_DELEGATE_H_
#define COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_INFOBAR_DELEGATE_H_
#include <memory>
#include <string>
#include <vector>
#include "components/language/core/browser/language_model.h"
#include "components/language/core/browser/language_prefs.h"
#include "components/sync_preferences/testing_pref_service_syncable.h"
#include "components/translate/core/browser/mock_translate_client.h"
#include "components/translate/core/browser/mock_translate_driver.h"
#include "components/translate/core/browser/mock_translate_ranker.h"
#include "components/translate/core/browser/translate_infobar_delegate.h"
#include "components/translate/core/browser/translate_manager.h"
#include "components/translate/core/browser/translate_pref_names.h"
#include "components/translate/core/browser/translate_prefs.h"
#include "testing/gmock/include/gmock/gmock.h"
using testing::_;
namespace translate {
namespace testing {
class MockLanguageModel : public language::LanguageModel {
std::vector<LanguageDetails> GetLanguages() override;
};
class MockTranslateInfoBarDelegate
: public translate::TranslateInfoBarDelegate {
public:
MockTranslateInfoBarDelegate(
const base::WeakPtr<translate::TranslateManager>& translate_manager,
translate::TranslateStep step,
const std::string& source_language,
const std::string& target_language,
translate::TranslateErrors error_type,
bool triggered_from_menu);
~MockTranslateInfoBarDelegate() override;
MOCK_CONST_METHOD0(num_languages, size_t());
MOCK_CONST_METHOD1(language_code_at, std::string(size_t index));
MOCK_CONST_METHOD1(language_name_at, std::u16string(size_t index));
MOCK_CONST_METHOD0(source_language_name, std::u16string());
MOCK_CONST_METHOD0(ShouldAlwaysTranslate, bool());
MOCK_METHOD1(AddObserver, void(Observer* observer));
MOCK_METHOD1(RemoveObserver, void(Observer* observer));
MOCK_METHOD0(InfoBarDismissed, void());
MOCK_METHOD0(Translate, void());
MOCK_METHOD0(ToggleAlwaysTranslate, void());
MOCK_METHOD0(ToggleTranslatableLanguageByPrefs, void());
MOCK_METHOD0(ToggleNeverPromptSite, void());
MOCK_METHOD0(RevertWithoutClosingInfobar, void());
MOCK_METHOD1(UpdateTargetLanguage, void(const std::string& language_code));
MOCK_METHOD1(UpdateSourceLanguage, void(const std::string& language_code));
void SetTranslateLanguagesForTest(
std::vector<std::pair<std::string, std::u16string>> languages);
void SetContentLanguagesCodesForTest(std::vector<std::string> languages);
private:
std::vector<std::pair<std::string, std::u16string>> languages_;
std::vector<std::string> content_languages_;
};
class MockTranslateInfoBarDelegateFactory {
public:
MockTranslateInfoBarDelegateFactory(const std::string& source_language,
const std::string& target_language);
~MockTranslateInfoBarDelegateFactory();
std::unique_ptr<MockTranslateInfoBarDelegate>
CreateMockTranslateInfoBarDelegate(translate::TranslateStep step);
MockTranslateInfoBarDelegate* GetMockTranslateInfoBarDelegate() {
return delegate_.get();
}
private:
MockTranslateDriver driver_;
std::unique_ptr<sync_preferences::TestingPrefServiceSyncable> pref_service_;
std::unique_ptr<MockTranslateClient> client_;
std::unique_ptr<MockTranslateRanker> ranker_;
std::unique_ptr<MockLanguageModel> language_model_;
std::unique_ptr<translate::TranslateManager> manager_;
std::unique_ptr<MockTranslateInfoBarDelegate> delegate_;
};
} // namespace testing
} // namespace translate
#endif // COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_INFOBAR_DELEGATE_H_
|
