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

content/test/test_web_contents.cc

 // 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.
 
 #include "content/test/test_web_contents.h"
 
 #include <utility>
 
 #include "base/command_line.h"
 #include "content/browser/browser_url_handler_impl.h"
@@ skipping 13 matching lines @@
 #include "content/public/test/mock_render_process_host.h"
 #include "content/test/test_render_view_host.h"
 #include "ui/base/page_transition_types.h"
 
 namespace content {
 
 TestWebContents::TestWebContents(BrowserContext* browser_context)
     : WebContentsImpl(browser_context, NULL),
       delegate_view_override_(NULL),
       expect_set_history_offset_and_length_(false),
-      expect_set_history_offset_and_length_history_length_(0) {
+      expect_set_history_offset_and_length_history_length_(0),
+      save_frame_headers_(std::string()) {

[Avi (use Gerrit), 2015/05/11 20:52:29]

You don't need to initialize objects that initialize themselves.

[Not at Google. Contact bengr, 2015/05/11 22:43:14]

Done.

 }
 
 TestWebContents* TestWebContents::Create(BrowserContext* browser_context,
                                          SiteInstance* instance) {
   TestWebContents* test_web_contents = new TestWebContents(browser_context);
   test_web_contents->Init(WebContents::CreateParams(browser_context, instance));
   test_web_contents->RenderFrameCreated(test_web_contents->GetMainFrame());
   return test_web_contents;
 }
 
@@ skipping 59 matching lines @@
   params.gesture = NavigationGestureUser;
   params.was_within_same_page = false;
   params.is_post = false;
   params.page_state = PageState::CreateFromURL(url);
 
   RenderFrameHostImpl* rfhi =
       static_cast<RenderFrameHostImpl*>(render_frame_host);
   rfhi->frame_tree_node()->navigator()->DidNavigate(rfhi, params);
 }
 
+const std::string& TestWebContents::GetSaveFrameHeaders() {
+  return save_frame_headers_;
+}
+
 bool TestWebContents::CrossProcessNavigationPending() {
   if (base::CommandLine::ForCurrentProcess()->HasSwitch(
           switches::kEnableBrowserSideNavigation)) {
     return GetRenderManager()->speculative_render_frame_host_ &&
            static_cast<TestRenderFrameHost*>(
                GetRenderManager()->speculative_render_frame_host_.get())
                ->pending_commit();
   }
   return GetRenderManager()->pending_frame_host() != nullptr;
 }
@@ skipping 154 matching lines @@
                                         bool user_gesture) {
 }
 
 void TestWebContents::ShowCreatedWidget(int route_id,
                                         const gfx::Rect& initial_rect) {
 }
 
 void TestWebContents::ShowCreatedFullscreenWidget(int route_id) {
 }
 
+void TestWebContents::SaveFrameWithHeaders(const GURL& url,
+                                           const Referrer& referrer,
+                                           const std::string& headers) {
+  save_frame_headers_ = headers;
+}
+
 }  // namespace content