Print Preview: Making margin selection sticky (part 2/2)

chrome/browser/ui/webui/print_preview_ui.h

 // Copyright (c) 2011 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 CHROME_BROWSER_UI_WEBUI_PRINT_PREVIEW_UI_H_
 #define CHROME_BROWSER_UI_WEBUI_PRINT_PREVIEW_UI_H_
 #pragma once
 
 #include <string>
 
@@ skipping 32 matching lines @@
 
   // Returns the available draft page count.
   int GetAvailableDraftPageCount();
 
   // Setters
   void SetInitiatorTabURLAndTitle(const std::string& initiator_url,
                                   const string16& initiator_tab_title);
 
   string16 initiator_tab_title() { return initiator_tab_title_; }
 
+  bool source_is_modifiable() { return source_is_modifiable_; }
+
+  void SetSourceIsModifiable(bool source_is_modifiable);
+
   // Determines whether to cancel a print preview request based on
   // |preview_ui_addr| and |request_id|.
   // Can be called from any thread.
   static void GetCurrentPrintPreviewStatus(const std::string& preview_ui_addr,
                                            int request_id,
                                            bool* cancel);
 
   // Returns a string to uniquely identify this PrintPreviewUI.
   std::string GetPrintPreviewUIAddress() const;
 
@@ skipping 62 matching lines @@
   // Store the PrintPreviewUI address string.
   std::string preview_ui_addr_str_;
 
   // Weak pointer to the WebUI handler.
   PrintPreviewHandler* handler_;
 
   // Store the |initiator_url| in order to display an accurate error message
   // when the initiator tab is closed/crashed.
   std::string initiator_url_;
 
+  // Indicates whether the source document can be modified.
+  bool source_is_modifiable_;
+
   // Store the initiator tab title, used for populating the print preview tab
   // title.
   string16 initiator_tab_title_;
 
+  friend class PrintPreviewHandlerTest;
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest, StickyMarginsCustom);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest, StickyMarginsDefault);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest,
+                           StickyMarginsCustomThenDefault);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest,
+                           TestGetLastUsedMarginSettingsCustom);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest,
+                           TestGetLastUsedMarginSettingsDefault);
   DISALLOW_COPY_AND_ASSIGN(PrintPreviewUI);
 };
 
 #endif  // CHROME_BROWSER_UI_WEBUI_PRINT_PREVIEW_UI_H_