Index: chrome/browser/page_load_metrics/observers/resource_tracking_page_load_metrics_observer_unittest.cc |
diff --git a/chrome/browser/page_load_metrics/observers/resource_tracking_page_load_metrics_observer_unittest.cc b/chrome/browser/page_load_metrics/observers/resource_tracking_page_load_metrics_observer_unittest.cc |
index 73611a0e1356f36cb57b75fb400145631bba66c3..6239ff99685b3702acbc9e9fe48aca45a581958d 100644 |
--- a/chrome/browser/page_load_metrics/observers/resource_tracking_page_load_metrics_observer_unittest.cc |
+++ b/chrome/browser/page_load_metrics/observers/resource_tracking_page_load_metrics_observer_unittest.cc |
@@ -4,7 +4,9 @@ |
#include "chrome/browser/page_load_metrics/observers/resource_tracking_page_load_metrics_observer.h" |
+#include "chrome/browser/offline_pages/offliner_user_data.h" |
#include "chrome/browser/page_load_metrics/observers/page_load_metrics_observer_test_harness.h" |
+#include "components/offline_pages/core/background/offliner_stub.h" |
namespace { |
const char kFakeUrl[] = "http://www.google.com/nothingotseehere.html"; |
@@ -54,6 +56,12 @@ TEST_F(ResourceTrackingPageLoadMetricsObserverTest, StartAndFinish) { |
content::ResourceType::RESOURCE_TYPE_IMAGE, |
}; |
+ // Add some OfflinerUserData to the web_contents so that our tracker will |
+ // be called. |
+ offline_pages::OfflinerStub offliner_stub; |
+ offline_pages::OfflinerUserData::AddToWebContents(web_contents(), |
+ &offliner_stub); |
+ |
// Start the navigation. This will create the page load tracker and register |
// the observers. |
NavigateAndCommit(GURL(kFakeUrl)); |
@@ -63,13 +71,13 @@ TEST_F(ResourceTrackingPageLoadMetricsObserverTest, StartAndFinish) { |
SimulateStartedResource(start_info_2); |
SimulateLoadedResource(done_info); |
- int64_t started = -1; |
- int64_t completed = -1; |
- EXPECT_NE(nullptr, observer()); |
- observer()->GetCountsForTypeForTesting( |
- content::ResourceType::RESOURCE_TYPE_IMAGE, &started, &completed); |
- EXPECT_EQ(2, started); |
- EXPECT_EQ(1, completed); |
+ int64_t resource_watcher_started = -1; |
+ int64_t resource_watcher_completed = -1; |
+ offliner_stub.GetResourcePercentageCounts( |
+ offline_pages::Offliner::ResourceDataType::IMAGE, |
+ &resource_watcher_started, &resource_watcher_completed); |
+ EXPECT_EQ(2, resource_watcher_started); |
+ EXPECT_EQ(1, resource_watcher_completed); |
} |
} // namespace page_load_metrics |