[translate] Add translate ranker model loader.

components/translate/core/browser/translate_manager_unittest.cc

 // Copyright 2015 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "components/translate/core/browser/translate_manager.h"
 
 #include "base/json/json_reader.h"
 #include "base/memory/ptr_util.h"
 #include "base/run_loop.h"
 #include "base/test/histogram_tester.h"
 #include "base/test/scoped_feature_list.h"
 #include "build/build_config.h"
 #include "components/infobars/core/infobar.h"
 #include "components/pref_registry/pref_registry_syncable.h"
 #include "components/sync_preferences/testing_pref_service_syncable.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_browser_metrics.h"
 #include "components/translate/core/browser/translate_client.h"
 #include "components/translate/core/browser/translate_download_manager.h"
 #include "components/translate/core/browser/translate_prefs.h"
 #include "components/translate/core/common/translate_pref_names.h"
 #include "components/variations/variations_associated_data.h"
 #include "net/base/network_change_notifier.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
@@ skipping 130 matching lines @@
 
   void TearDown() override {
     manager_->ResetForTesting();
     variations::testing::ClearAllVariationParams();
   }
 
   // Utility function to prepare translate_manager_ for testing.
   void PrepareTranslateManager() {
     TranslateManager::SetIgnoreMissingKeyForTesting(true);
     translate_manager_.reset(new translate::TranslateManager(
-        &mock_translate_client_, kAcceptLanguages));
+        &mock_translate_client_, &mock_translate_ranker_, kAcceptLanguages));
   }
 
   // Prepare the test for ULP related tests.
   // Put the ulp json into profile.
   void PrepareULPTest(const char* ulp_json, bool turn_on_feature) {
     PrepareTranslateManager();
     std::unique_ptr<base::Value> profile(CreateProfileFromJSON(ulp_json));
     prefs_.SetUserPref(TranslatePrefs::kPrefLanguageProfile, profile.release());
     if (turn_on_feature)
       TurnOnTranslateByULP();
@@ skipping 45 matching lines @@
   }
 
   sync_preferences::TestingPrefServiceSyncable prefs_;
 
   // TODO(groby): request TranslatePrefs from |mock_translate_client_| instead.
   TranslatePrefs translate_prefs_;
   TranslateDownloadManager* manager_;
 
   TestNetworkChangeNotifier network_notifier_;
   translate::testing::MockTranslateDriver driver_;
+  translate::testing::MockTranslateRanker mock_translate_ranker_;
   ::testing::NiceMock<MockTranslateClient> mock_translate_client_;
   std::unique_ptr<TranslateManager> translate_manager_;
   std::unique_ptr<base::FieldTrialList> field_trial_list_;
   base::test::ScopedFeatureList scoped_feature_list_;
 };
 
 
 // Target language comes from application locale if the locale's language
 // is supported.
 TEST_F(TranslateManagerTest, GetTargetLanguageDefaultsToAppLocale) {
@@ skipping 31 matching lines @@
   // First supported language is the fallback language.
   ASSERT_TRUE(TranslateDownloadManager::IsSupportedLanguage("en"));
   accept_language_list.push_back("en");
   translate_prefs_.UpdateLanguageList(accept_language_list);
   EXPECT_EQ("en", TranslateManager::GetTargetLanguage(&translate_prefs_));
 }
 
 TEST_F(TranslateManagerTest, DontTranslateOffline) {
   TranslateManager::SetIgnoreMissingKeyForTesting(true);
   translate_manager_.reset(new translate::TranslateManager(
-      &mock_translate_client_, kAcceptLanguages));
+      &mock_translate_client_, &mock_translate_ranker_, kAcceptLanguages));
 
   // The test measures that the "Translate was disabled" exit can only be
   // reached after the early-out tests including IsOffline() passed.
   const char kMetricName[] = "Translate.InitiationStatus.v2";
   base::HistogramTester histogram_tester;
 
   prefs_.SetBoolean(prefs::kEnableTranslate, false);
 
   translate_manager_->GetLanguageState().LanguageDetermined("de", true);
 
@@ skipping 11 matching lines @@
       translate::TranslateBrowserMetrics::INITIATION_STATUS_DISABLED_BY_PREFS,
       1);
 }
 
 // The test measures that Translate is not triggered for a zh-TW page for a
 // zh-CN user.
 TEST_F(TranslateManagerTest,
        DontTranslateZhTraditionalPageForZhSimplifiedLocale) {
   TranslateManager::SetIgnoreMissingKeyForTesting(true);
   translate_manager_.reset(new translate::TranslateManager(
-      &mock_translate_client_, kAcceptLanguages));
+      &mock_translate_client_, &mock_translate_ranker_, kAcceptLanguages));
 
   const char kMetricName[] = "Translate.InitiationStatus.v2";
   base::HistogramTester histogram_tester;
 
   const std::string locale = "zh-TW";
   const std::string page_lang = "zh-CN";
 
   network_notifier_.SimulateOnline();
   manager_->set_application_locale(locale);
   ON_CALL(mock_translate_client_, IsTranslatableURL(_))
       .WillByDefault(Return(true));
 
   EXPECT_EQ("zh-TW", translate_manager_->GetTargetLanguage(&translate_prefs_));
   translate_manager_->GetLanguageState().LanguageDetermined(page_lang, true);
   translate_manager_->InitiateTranslation(page_lang);
 
   histogram_tester.ExpectUniqueSample(
       kMetricName,
       translate::TranslateBrowserMetrics::INITIATION_STATUS_SIMILAR_LANGUAGES,
       1);
 }
 
 // The test measures that Translate is not triggered for a zh-CN page for a
 // zh-TW user.
 TEST_F(TranslateManagerTest,
        DontTranslateZhSimplifiedPageForZhTraditionalLocale) {
   TranslateManager::SetIgnoreMissingKeyForTesting(true);
   translate_manager_.reset(new translate::TranslateManager(
-      &mock_translate_client_, kAcceptLanguages));
+      &mock_translate_client_, &mock_translate_ranker_, kAcceptLanguages));
 
   const char kMetricName[] = "Translate.InitiationStatus.v2";
   base::HistogramTester histogram_tester;
 
   const std::string locale = "zh-CN";
   const std::string page_lang = "zh-TW";
 
   network_notifier_.SimulateOnline();
   manager_->set_application_locale(locale);
   ON_CALL(mock_translate_client_, IsTranslatableURL(_))
@@ skipping 147 matching lines @@
   // 0.79 and lower than 0.8 and the probability threshold is lower than both
   // the one with "fr" (0.6) and "pt-PT" (0.4).
   EXPECT_TRUE(CallLanguageInULP("fr"));
   EXPECT_TRUE(CallLanguageInULP("pt"));
   EXPECT_FALSE(CallLanguageInULP("zh-TW"));
 }
 
 }  // namespace testing
 
 }  // namespace translate