Index: chrome/browser/page_load_metrics/observers/loading_predictor_page_load_metrics_observer_unittest.cc |
diff --git a/chrome/browser/page_load_metrics/observers/resource_prefetch_predictor_page_load_metrics_observer_unittest.cc b/chrome/browser/page_load_metrics/observers/loading_predictor_page_load_metrics_observer_unittest.cc |
similarity index 82% |
rename from chrome/browser/page_load_metrics/observers/resource_prefetch_predictor_page_load_metrics_observer_unittest.cc |
rename to chrome/browser/page_load_metrics/observers/loading_predictor_page_load_metrics_observer_unittest.cc |
index 64cbc254766ba6839ad419f4755b2126ba350e0b..cfbd42e162c749c14cf042d036bcdd2e027559da 100644 |
--- a/chrome/browser/page_load_metrics/observers/resource_prefetch_predictor_page_load_metrics_observer_unittest.cc |
+++ b/chrome/browser/page_load_metrics/observers/loading_predictor_page_load_metrics_observer_unittest.cc |
@@ -2,18 +2,20 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
-#include "chrome/browser/page_load_metrics/observers/resource_prefetch_predictor_page_load_metrics_observer.h" |
+#include "chrome/browser/page_load_metrics/observers/loading_predictor_page_load_metrics_observer.h" |
#include <memory> |
#include "base/memory/ptr_util.h" |
#include "chrome/browser/page_load_metrics/observers/page_load_metrics_observer_test_harness.h" |
+#include "chrome/browser/predictors/loading_data_collector.h" |
#include "chrome/browser/predictors/resource_prefetch_common.h" |
#include "chrome/browser/predictors/resource_prefetch_predictor.h" |
#include "chrome/test/base/testing_profile.h" |
#include "content/public/test/test_browser_thread_bundle.h" |
#include "testing/gmock/include/gmock/gmock.h" |
+using predictors::LoadingDataCollector; |
using predictors::ResourcePrefetchPredictor; |
class MockResourcePrefetchPredictor : public ResourcePrefetchPredictor { |
@@ -28,7 +30,7 @@ class MockResourcePrefetchPredictor : public ResourcePrefetchPredictor { |
~MockResourcePrefetchPredictor() override {} |
}; |
-class ResourcePrefetchPredictorPageLoadMetricsObserverTest |
+class LoadingPredictorPageLoadMetricsObserverTest |
: public page_load_metrics::PageLoadMetricsObserverTestHarness { |
protected: |
void SetUp() override { |
@@ -39,6 +41,7 @@ class ResourcePrefetchPredictorPageLoadMetricsObserverTest |
base::MakeUnique<testing::StrictMock<MockResourcePrefetchPredictor>>( |
config, profile()); |
page_load_metrics::InitPageLoadTimingForTest(&timing_); |
+ collector_ = base::MakeUnique<LoadingDataCollector>(predictor_.get()); |
timing_.navigation_start = base::Time::FromDoubleT(1); |
timing_.paint_timing->first_paint = base::TimeDelta::FromSeconds(2); |
timing_.paint_timing->first_contentful_paint = |
@@ -50,17 +53,17 @@ class ResourcePrefetchPredictorPageLoadMetricsObserverTest |
void RegisterObservers(page_load_metrics::PageLoadTracker* tracker) override { |
tracker->AddObserver( |
- base::MakeUnique<ResourcePrefetchPredictorPageLoadMetricsObserver>( |
- predictor_.get(), web_contents())); |
+ base::MakeUnique<LoadingPredictorPageLoadMetricsObserver>( |
+ predictor_.get(), collector_.get(), web_contents())); |
} |
std::unique_ptr<testing::StrictMock<MockResourcePrefetchPredictor>> |
predictor_; |
page_load_metrics::mojom::PageLoadTiming timing_; |
+ std::unique_ptr<LoadingDataCollector> collector_; |
}; |
-TEST_F(ResourcePrefetchPredictorPageLoadMetricsObserverTest, |
- PrefetchableIsRecorded) { |
+TEST_F(LoadingPredictorPageLoadMetricsObserverTest, PrefetchableIsRecorded) { |
const GURL main_frame_url("https://www.google.com"); |
EXPECT_CALL(*predictor_.get(), IsUrlPrefetchable(main_frame_url)) |
.WillOnce(testing::Return(true)); |
@@ -74,7 +77,7 @@ TEST_F(ResourcePrefetchPredictorPageLoadMetricsObserverTest, |
internal::kHistogramResourcePrefetchPredictorFirstMeaningfulPaint, 1); |
} |
-TEST_F(ResourcePrefetchPredictorPageLoadMetricsObserverTest, |
+TEST_F(LoadingPredictorPageLoadMetricsObserverTest, |
NotPrefetchableIsNotRecorded) { |
const GURL main_frame_url("https://www.google.com"); |
EXPECT_CALL(*predictor_.get(), IsUrlPrefetchable(main_frame_url)) |