Tests for "Save Image As..." when Data Saver is enabled.

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 <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 SiteInstanceImpl;
 
 namespace content {
 
 class RenderViewHost;
 class TestRenderViewHost;
 class WebContentsTester;
 
 // Subclass WebContentsImpl to ensure it creates TestRenderViewHosts
 // and does not do anything involving views.
@@ skipping 20 matching lines @@
                        bool did_create_new_entry,
                        const GURL& url,
                        ui::PageTransition transition) override;
   void TestDidNavigateWithReferrer(RenderFrameHost* render_frame_host,
                                    int page_id,
                                    int nav_entry_id,
                                    bool did_create_new_entry,
                                    const GURL& url,
                                    const Referrer& referrer,
                                    ui::PageTransition transition) override;
+  const std::string& GetSaveFrameHeaders() override;
 
   // True if a cross-site navigation is pending.
   bool CrossProcessNavigationPending();
 
   // Prevent interaction with views.
   bool CreateRenderViewForRenderManager(RenderViewHost* render_view_host,
                                         int opener_route_id,
                                         int proxy_routing_id,
                                         bool for_main_frame) override;
   void UpdateRenderViewSizeForRenderManager() override {}
@@ skipping 45 matching lines @@
   void CreateNewWidget(int render_process_id,
                        int route_id,
                        blink::WebPopupType popup_type) override;
   void CreateNewFullscreenWidget(int render_process_id, int route_id) override;
   void ShowCreatedWindow(int route_id,
                          WindowOpenDisposition disposition,
                          const gfx::Rect& initial_rect,
                          bool user_gesture) override;
   void ShowCreatedWidget(int route_id, const gfx::Rect& initial_rect) override;
   void ShowCreatedFullscreenWidget(int route_id) override;
+  void SaveFrameWithHeaders(const GURL& url,
+                            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_;
 };
 
 }  // namespace content
 
 #endif  // CONTENT_TEST_TEST_WEB_CONTENTS_H_