predictors: Don't call IsUrlPrefetchable twice.

chrome/browser/predictors/resource_prefetch_predictor_test_util.h

 // Copyright 2016 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.
 #ifndef CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCH_PREDICTOR_TEST_UTIL_H_
 #define CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCH_PREDICTOR_TEST_UTIL_H_
 
-#include <set>
 #include <string>
 #include <vector>
 
 #include "chrome/browser/predictors/resource_prefetch_predictor.h"
 #include "chrome/browser/predictors/resource_prefetch_predictor_tables.h"
 #include "components/sessions/core/session_id.h"
 #include "testing/gmock/include/gmock/gmock.h"
 
 namespace predictors {
 
 // Does nothing, controls which URLs are prefetchable.
 class MockResourcePrefetchPredictor : public ResourcePrefetchPredictor {
  public:
   MockResourcePrefetchPredictor(const LoadingPredictorConfig& config,
                                 Profile* profile);
   ~MockResourcePrefetchPredictor();
 
-  bool IsUrlPrefetchable(const GURL& main_frame_url) const override {
-    return prefetchable_urls_.find(main_frame_url) != prefetchable_urls_.end();
-  }
-
-  void AddPrefetchableUrl(const GURL& url) { prefetchable_urls_.insert(url); }
-
   MOCK_CONST_METHOD2(GetPrefetchData,
-                     bool(const GURL& main_frame_url, Prediction* prediction));
+                     bool(const GURL&, ResourcePrefetchPredictor::Prediction*));
   MOCK_METHOD0(StartInitialization, void());
   MOCK_METHOD0(Shutdown, void());
-  MOCK_METHOD1(StartPrefetching, void(const GURL&));
+  MOCK_METHOD2(StartPrefetching,
+               void(const GURL&, const ResourcePrefetchPredictor::Prediction&));
   MOCK_METHOD1(StopPrefeching, void(const GURL&));
-
- private:
-  std::set<GURL> prefetchable_urls_;
 };
 
 void InitializeResourceData(ResourceData* resource,
                             const std::string& resource_url,
                             content::ResourceType resource_type,
                             int number_of_hits,
                             int number_of_misses,
                             int consecutive_misses,
                             double average_position,
                             net::RequestPriority priority,
@@ skipping 94 matching lines @@
                          const PrecacheManifest& manifest);
 std::ostream& operator<<(std::ostream& stream,
                          const PrecacheResource& resource);
 
 bool operator==(const PrecacheManifest& lhs, const PrecacheManifest& rhs);
 bool operator==(const PrecacheResource& lhs, const PrecacheResource& rhs);
 
 }  // namespace precache
 
 #endif  // CHROME_BROWSER_PREDICTORS_RESOURCE_PREFETCH_PREDICTOR_TEST_UTIL_H_