Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(91)

Side by Side Diff: components/translate/core/browser/mock_translate_ranker.h

Issue 2565873002: [translate] Add translate ranker model loader. (Closed)
Patch Set: fix builders? Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
OLDNEW
(Empty)
1 // Copyright 2016 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 #ifndef COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_
6 #define COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_
7
8 #include <memory>
9 #include <string>
10 #include <vector>
11
12 #include "components/translate/core/browser/translate_ranker.h"
13
14 namespace metrics {
15 class TranslateEventProto;
16 }
17
18 namespace translate {
19
20 class TranslatePrefs;
21
22 namespace testing {
23
24 class MockTranslateRanker : public TranslateRanker {
25 public:
26 MockTranslateRanker();
27 ~MockTranslateRanker() override;
28
29 void set_is_logging_enabled(bool val) { is_logging_enabled_ = val; }
30 void set_is_query_enabled(bool val) { is_query_enabled_ = val; }
31 void set_is_enforcement_enabled(bool val) { is_enforcement_enabled_ = val; }
32 void set_model_version(int val) { model_version_ = val; }
33 void set_should_offer_translation(bool val) {
34 should_offer_translation_ = val;
35 }
36
37 // TranslateRanker...
38 bool IsEnabled() override;
39 bool IsLoggingEnabled() override;
40 bool IsQueryEnabled() override;
41 bool IsEnforcementEnabled() override;
42 int GetModelVersion() const override;
43 bool ShouldOfferTranslation(const TranslatePrefs& translate_prefs,
44 const std::string& src_lang,
45 const std::string& dst_lang) override;
46 void AddTranslateEvent(
47 std::unique_ptr<metrics::TranslateEventProto> translate_event) override;
48 void FlushTranslateEvents(
49 std::vector<std::unique_ptr<metrics::TranslateEventProto>>* events)
50 override;
51
52 private:
53 std::vector<std::unique_ptr<metrics::TranslateEventProto>> event_cache_;
54
55 bool is_logging_enabled_ = false;
56 bool is_query_enabled_ = false;
57 bool is_enforcement_enabled_ = false;
58 bool model_version_ = 0;
59 bool should_offer_translation_ = true;
60
61 DISALLOW_COPY_AND_ASSIGN(MockTranslateRanker);
62 };
63
64 } // namespace testing
65 } // namespace translate
66
67 #endif // COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698