Improve test coverage for ContentFaviconDriver

content/test/test_web_contents.h

 // Copyright (c) 2012 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 CONTENT_TEST_TEST_WEB_CONTENTS_H_
 #define CONTENT_TEST_TEST_WEB_CONTENTS_H_
 
 #include <stdint.h>
 
+#include <list>
+#include <map>
 #include <string>
 
 #include "content/browser/web_contents/web_contents_impl.h"
 #include "content/public/test/web_contents_tester.h"
 #include "content/test/test_render_frame_host.h"
 #include "content/test/test_render_view_host.h"
 #include "ui/base/page_transition_types.h"
 
 class GURL;
 class Referrer;
+class SkBitmap;
+
+namespace gfx {
+class Size;
+}
 
 namespace content {
 
 class NavigationData;
 class NavigationHandle;
 class RenderViewHost;
 class TestRenderViewHost;
 class WebContentsTester;
 
 // Subclass WebContentsImpl to ensure it creates TestRenderViewHosts
@@ skipping 36 matching lines @@
   void TestDidNavigateWithSequenceNumber(RenderFrameHost* render_frame_host,
                                          int nav_entry_id,
                                          bool did_create_new_entry,
                                          const GURL& url,
                                          const Referrer& referrer,
                                          ui::PageTransition transition,
                                          bool was_within_same_page,
                                          int item_sequence_number,
                                          int document_sequence_number);
   const std::string& GetSaveFrameHeaders() override;
+  bool HasPendingDownloadImage(const GURL& url) override;
+  bool TestDidDownloadImage(
+      const GURL& url,
+      int http_status_code,
+      const std::vector<SkBitmap>& bitmaps,
+      const std::vector<gfx::Size>& original_bitmap_sizes) override;
 
   // True if a cross-site navigation is pending.
   bool CrossProcessNavigationPending();
 
   // Prevent interaction with views.
   bool CreateRenderViewForRenderManager(
       RenderViewHost* render_view_host,
       int opener_frame_routing_id,
       int proxy_routing_id,
       const FrameReplicationState& replicated_frame_state) override;
@@ skipping 68 matching lines @@
                             const Referrer& referrer,
                             const std::string& headers) override;
 
   RenderViewHostDelegateView* delegate_view_override_;
 
   // Expectations for arguments of |SetHistoryOffsetAndLength()|.
   bool expect_set_history_offset_and_length_;
   int expect_set_history_offset_and_length_history_offset_;
   int expect_set_history_offset_and_length_history_length_;
   std::string save_frame_headers_;
+  // Map keyed by image URL. Values are <id, callback> pairs.
+  std::map<GURL, std::list<std::pair<int, ImageDownloadCallback>>>
+      pending_image_downloads_;
 };
 
 }  // namespace content
 
 #endif  // CONTENT_TEST_TEST_WEB_CONTENTS_H_