| 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 |