predictors: Ignore repeating subresources while checking.

chrome/browser/predictors/resource_prefetch_predictor_browsertest.cc

 // 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.
 
 #include "base/command_line.h"
 #include "chrome/browser/predictors/resource_prefetch_predictor.h"
 #include "chrome/browser/predictors/resource_prefetch_predictor_factory.h"
 #include "chrome/browser/predictors/resource_prefetch_predictor_test_util.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/ui/browser.h"
 #include "chrome/browser/ui/tabs/tab_strip_model.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/test/base/in_process_browser_test.h"
 #include "chrome/test/base/ui_test_utils.h"
 #include "content/public/browser/render_frame_host.h"
 #include "content/public/browser/render_process_host.h"
 #include "net/test/embedded_test_server/http_request.h"
 #include "net/test/embedded_test_server/http_response.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 
 namespace predictors {
 
+namespace {
+
 static const char kImagePath[] = "/predictors/image.png";
 static const char kStylePath[] = "/predictors/style.css";
 static const char kScriptPath[] = "/predictors/script.js";
 static const char kFontPath[] = "/predictors/font.ttf";
 static const char kHtmlSubresourcesPath[] =
     "/predictors/html_subresources.html";
 static const char kRedirectPath[] = "/predictors/redirect.html";
 static const char kRedirectPath2[] = "/predictors/redirect2.html";
 static const char kRedirectPath3[] = "/predictors/redirect3.html";
 
@@ skipping 20 matching lines @@
   void OnPredictorInitialized() override { run_loop_.Quit(); }
 
   void Wait() { run_loop_.Run(); }
 
  private:
   base::RunLoop run_loop_;
 
   DISALLOW_COPY_AND_ASSIGN(InitializationObserver);
 };
 
+using PageRequestSummary = ResourcePrefetchPredictor::PageRequestSummary;
+using URLRequestSummary = ResourcePrefetchPredictor::URLRequestSummary;
+
+std::vector<URLRequestSummary> GetUniqueSubresources(
+    const PageRequestSummary& summary) {
+  std::vector<URLRequestSummary> subresources(summary.subresource_requests);
+  std::stable_sort(subresources.begin(), subresources.end(),
+                   [](const URLRequestSummary& x, const URLRequestSummary& y) {
+                     return x.resource_url < y.resource_url;
+                   });
+  subresources.erase(
+      std::unique(subresources.begin(), subresources.end(),
+                  [](const URLRequestSummary& x, const URLRequestSummary& y) {
+                    return x.resource_url == y.resource_url;
+                  }),
+      subresources.end());
+  return subresources;
+}
+
+}  // namespace
+
 // Helper class to track and allow waiting for ResourcePrefetchPredictor events.
 // These events are also used to verify that ResourcePrefetchPredictor works as
 // expected.
 class ResourcePrefetchPredictorTestObserver : public TestObserver {
  public:
   using PageRequestSummary = ResourcePrefetchPredictor::PageRequestSummary;
 
   explicit ResourcePrefetchPredictorTestObserver(
       ResourcePrefetchPredictor* predictor,
       const size_t expected_url_visit_count,
       const PageRequestSummary& expected_summary)
       : TestObserver(predictor),
         url_visit_count_(expected_url_visit_count),
         summary_(expected_summary) {}
 
   // TestObserver:
   void OnNavigationLearned(size_t url_visit_count,
                            const PageRequestSummary& summary) override {
     EXPECT_EQ(url_visit_count, url_visit_count_);
     EXPECT_EQ(summary.main_frame_url, summary_.main_frame_url);
     EXPECT_EQ(summary.initial_url, summary_.initial_url);
-    EXPECT_THAT(
-        summary.subresource_requests,
-        testing::UnorderedElementsAreArray(summary_.subresource_requests));
+    // Duplicate resources can be observed in a single navigation but
+    // ResourcePrefetchPredictor only cares about the first occurrence of each.
+    std::vector<ResourcePrefetchPredictor::URLRequestSummary> subresources =
+        GetUniqueSubresources(summary);
+    EXPECT_THAT(subresources, testing::UnorderedElementsAreArray(
+                                  summary_.subresource_requests));
     run_loop_.Quit();
   }
 
   void Wait() { run_loop_.Run(); }
 
  private:
   base::RunLoop run_loop_;
   size_t url_visit_count_;
   PageRequestSummary summary_;
 
@@ skipping 242 matching lines @@
   AddResource(https_server()->GetURL(kStylePath),
               content::RESOURCE_TYPE_STYLESHEET, net::HIGHEST);
   AddResource(https_server()->GetURL(kScriptPath),
               content::RESOURCE_TYPE_SCRIPT, net::MEDIUM);
   AddResource(https_server()->GetURL(kFontPath),
               content::RESOURCE_TYPE_FONT_RESOURCE, net::HIGHEST);
   NavigateToURLAndCheckSubresources(GetURL(kRedirectPath));
 }
 
 }  // namespace predictors