predictors: Enable origin learning with a flag.

chrome/browser/predictors/resource_prefetch_common_unittest.cc

 // Copyright 2014 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 <memory>
 #include <string>
 
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "chrome/browser/net/prediction_options.h"
@@ skipping 55 matching lines @@
     EXPECT_GT(config.min_resource_hits_to_trigger_prefetch, 1U);
   }
 
   void TestIsDefaultExtraConfig(const ResourcePrefetchPredictorConfig& config) {
     EXPECT_FALSE(config.IsLowConfidenceForTest());
     EXPECT_FALSE(config.IsHighConfidenceForTest());
     EXPECT_FALSE(config.IsMoreResourcesEnabledForTest());
     EXPECT_FALSE(config.IsSmallDBEnabledForTest());
     EXPECT_FALSE(config.is_url_learning_enabled);
     EXPECT_FALSE(config.is_manifests_enabled);
-    EXPECT_FALSE(config.is_origin_prediction_enabled);
+    EXPECT_FALSE(config.is_origin_learning_enabled);
     EXPECT_GT(config.min_resource_hits_to_trigger_prefetch, 1U);
   }
 
  protected:
   base::MessageLoop loop_;
   content::TestBrowserThread ui_thread_;
   std::unique_ptr<TestingProfile> profile_;
 };
 
 ResourcePrefetchCommonTest::ResourcePrefetchCommonTest()
@@ skipping 68 matching lines @@
       "dummy-trial",
       {{kModeParamName, kLearningMode}, {kEnableManifestsParamName, "true"}},
       {kSpeculativeResourcePrefetchingFeatureName});
 
   ResourcePrefetchPredictorConfig config;
   EXPECT_TRUE(IsSpeculativeResourcePrefetchingEnabled(profile_.get(), &config));
   TestIsPrefetchLearning(config);
   EXPECT_TRUE(config.is_manifests_enabled);
 }
 
+TEST_F(ResourcePrefetchCommonTest, EnableOriginLearning) {
+  variations::testing::VariationParamsManager params_manager(
+      "dummy-trial",
+      {{kModeParamName, kLearningMode},
+       {kEnableOriginLearningParamName, "true"}},
+      {kSpeculativeResourcePrefetchingFeatureName});
+
+  ResourcePrefetchPredictorConfig config;
+  EXPECT_TRUE(IsSpeculativeResourcePrefetchingEnabled(profile_.get(), &config));
+  TestIsPrefetchLearning(config);
+  EXPECT_TRUE(config.is_origin_learning_enabled);
+}
+
 // Verifies whether prefetching is disabled according to the network type. But
 // learning should not be disabled by network.
 TEST_F(ResourcePrefetchCommonTest, RespectsNetworkSettings) {
   variations::testing::VariationParamsManager params_manager(
       "dummy-trial", {{kModeParamName, kPrefetchingMode}},
       {kSpeculativeResourcePrefetchingFeatureName});
 
   ResourcePrefetchPredictorConfig config;
   EXPECT_TRUE(IsSpeculativeResourcePrefetchingEnabled(profile_.get(), &config));
   TestIsPrefetchEnabledForOrigin(config, PrefetchOrigin::EXTERNAL);
@@ skipping 20 matching lines @@
     TestIsPrefetchLearning(config);
   }
   {
     std::unique_ptr<NetworkChangeNotifier> mock(
         new MockNetworkChangeNotifier4G);
     TestIsPrefetchLearning(config);
   }
 }
 
 }  // namespace predictors