OLD | NEW |
1 // Copyright 2017 The Chromium Authors. All rights reserved. | 1 // Copyright 2017 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #ifndef COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_ | 5 #ifndef COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_ |
6 #define COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_ | 6 #define COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_ |
7 | 7 |
8 #include <memory> | 8 #include <memory> |
9 #include <string> | 9 #include <string> |
10 #include <vector> | 10 #include <vector> |
11 | 11 |
12 #include "components/translate/core/browser/translate_ranker.h" | 12 #include "components/translate/core/browser/translate_ranker.h" |
13 #include "testing/gmock/include/gmock/gmock.h" | 13 #include "testing/gmock/include/gmock/gmock.h" |
14 | 14 |
15 class GURL; | 15 class GURL; |
16 | 16 |
17 namespace metrics { | 17 namespace metrics { |
18 class TranslateEventProto; | 18 class TranslateEventProto; |
19 } | 19 } |
20 | 20 |
21 namespace translate { | 21 namespace translate { |
22 | 22 |
23 class TranslatePrefs; | |
24 | |
25 namespace testing { | 23 namespace testing { |
26 | 24 |
27 class MockTranslateRanker : public TranslateRanker { | 25 class MockTranslateRanker : public TranslateRanker { |
28 public: | 26 public: |
29 MockTranslateRanker(); | 27 MockTranslateRanker(); |
30 ~MockTranslateRanker() override; | 28 ~MockTranslateRanker() override; |
31 | 29 |
32 void set_is_logging_enabled(bool val) { is_logging_enabled_ = val; } | 30 void set_is_logging_enabled(bool val) { is_logging_enabled_ = val; } |
33 void set_is_query_enabled(bool val) { is_query_enabled_ = val; } | 31 void set_is_query_enabled(bool val) { is_query_enabled_ = val; } |
34 void set_is_enforcement_enabled(bool val) { is_enforcement_enabled_ = val; } | 32 void set_is_enforcement_enabled(bool val) { is_enforcement_enabled_ = val; } |
35 void set_is_decision_override_enabled(bool val) { | 33 void set_is_decision_override_enabled(bool val) { |
36 is_decision_override_enabled_ = val; | 34 is_decision_override_enabled_ = val; |
37 } | 35 } |
38 void set_model_version(int val) { model_version_ = val; } | 36 void set_model_version(int val) { model_version_ = val; } |
39 void set_should_offer_translation(bool val) { | 37 void set_should_offer_translation(bool val) { |
40 should_offer_translation_ = val; | 38 should_offer_translation_ = val; |
41 } | 39 } |
42 | 40 |
43 // TranslateRanker Implementation: | 41 // TranslateRanker Implementation: |
44 uint32_t GetModelVersion() const override; | 42 uint32_t GetModelVersion() const override; |
45 bool ShouldOfferTranslation( | 43 bool ShouldOfferTranslation( |
46 const TranslatePrefs& translate_prefs, | |
47 const std::string& src_lang, | |
48 const std::string& dst_lang, | |
49 metrics::TranslateEventProto* translate_events) override; | 44 metrics::TranslateEventProto* translate_events) override; |
50 void FlushTranslateEvents( | 45 void FlushTranslateEvents( |
51 std::vector<metrics::TranslateEventProto>* events) override; | 46 std::vector<metrics::TranslateEventProto>* events) override; |
52 MOCK_METHOD3(RecordTranslateEvent, | 47 MOCK_METHOD3(RecordTranslateEvent, |
53 void(int event_type, | 48 void(int event_type, |
54 const GURL& url, | 49 const GURL& url, |
55 metrics::TranslateEventProto* translate_event)); | 50 metrics::TranslateEventProto* translate_event)); |
56 MOCK_METHOD3(ShouldOverrideDecision, | 51 MOCK_METHOD3(ShouldOverrideDecision, |
57 bool(int event_type, | 52 bool(int event_type, |
58 const GURL& url, | 53 const GURL& url, |
(...skipping 10 matching lines...) Expand all Loading... |
69 bool should_offer_translation_ = true; | 64 bool should_offer_translation_ = true; |
70 | 65 |
71 DISALLOW_COPY_AND_ASSIGN(MockTranslateRanker); | 66 DISALLOW_COPY_AND_ASSIGN(MockTranslateRanker); |
72 }; | 67 }; |
73 | 68 |
74 } // namespace testing | 69 } // namespace testing |
75 | 70 |
76 } // namespace translate | 71 } // namespace translate |
77 | 72 |
78 #endif // COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_ | 73 #endif // COMPONENTS_TRANSLATE_CORE_BROWSER_MOCK_TRANSLATE_RANKER_H_ |
OLD | NEW |