Index: components/translate/core/browser/mock_translate_ranker.h |
diff --git a/components/translate/core/browser/mock_translate_ranker.h b/components/translate/core/browser/mock_translate_ranker.h |
index ea71eb8e31fe9b36d90dc975bc27a4f9fc2b6324..f2e9d446649fdc66b4ca7e9ce6a3d693842a75cb 100644 |
--- a/components/translate/core/browser/mock_translate_ranker.h |
+++ b/components/translate/core/browser/mock_translate_ranker.h |
@@ -10,6 +10,7 @@ |
#include <vector> |
#include "components/translate/core/browser/translate_ranker.h" |
+#include "testing/gmock/include/gmock/gmock.h" |
class GURL; |
@@ -31,23 +32,31 @@ class MockTranslateRanker : public TranslateRanker { |
void set_is_logging_enabled(bool val) { is_logging_enabled_ = val; } |
void set_is_query_enabled(bool val) { is_query_enabled_ = val; } |
void set_is_enforcement_enabled(bool val) { is_enforcement_enabled_ = val; } |
+ void set_is_decision_override_enabled(bool val) { |
+ is_decision_override_enabled_ = val; |
+ } |
void set_model_version(int val) { model_version_ = val; } |
void set_should_offer_translation(bool val) { |
should_offer_translation_ = val; |
} |
// TranslateRanker Implementation: |
- bool IsLoggingEnabled() override; |
- bool IsQueryEnabled() override; |
- bool IsEnforcementEnabled() override; |
- int GetModelVersion() const override; |
- bool ShouldOfferTranslation(const TranslatePrefs& translate_prefs, |
- const std::string& src_lang, |
- const std::string& dst_lang) override; |
- void AddTranslateEvent(const metrics::TranslateEventProto& translate_event, |
- const GURL& url) override; |
+ uint32_t GetModelVersion() const override; |
+ bool ShouldOfferTranslation( |
+ const TranslatePrefs& translate_prefs, |
+ const std::string& src_lang, |
+ const std::string& dst_lang, |
+ metrics::TranslateEventProto* translate_events) override; |
void FlushTranslateEvents( |
std::vector<metrics::TranslateEventProto>* events) override; |
+ MOCK_METHOD3(RecordTranslateEvent, |
+ void(int event_type, |
+ const GURL& url, |
+ metrics::TranslateEventProto* translate_event)); |
+ MOCK_METHOD3(ShouldOverrideDecision, |
+ bool(int event_type, |
+ const GURL& url, |
+ metrics::TranslateEventProto* translate_event)); |
private: |
std::vector<metrics::TranslateEventProto> event_cache_; |
@@ -55,6 +64,7 @@ class MockTranslateRanker : public TranslateRanker { |
bool is_logging_enabled_ = false; |
bool is_query_enabled_ = false; |
bool is_enforcement_enabled_ = false; |
+ bool is_decision_override_enabled_ = false; |
bool model_version_ = 0; |
bool should_offer_translation_ = true; |