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 44 matching lines @@
   void OnFileSelectionCancelled();
 
   // Notifies the Web UI that the printer is unavailable or its settings are
   // invalid.
   void OnInvalidPrinterSettings();
 
   // Notifies the Web UI to cancel the pending preview request.
   void OnCancelPendingPreviewRequest();
 
  private:
+  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,
+                           GetLastUsedMarginSettingsCustom);
+  FRIEND_TEST_ALL_PREFIXES(PrintPreviewHandlerTest,
+                           GetLastUsedMarginSettingsDefault);
   FRIEND_TEST_ALL_PREFIXES(PrintPreviewTabControllerUnitTest,
                            TitleAfterReload);
 
   // Returns the Singleton instance of the PrintPreviewDataService.
   PrintPreviewDataService* print_preview_data_service();
 
   base::TimeTicks initial_preview_start_time_;
 
   // 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_;
 
   DISALLOW_COPY_AND_ASSIGN(PrintPreviewUI);
 };
 
 #endif  // CHROME_BROWSER_UI_WEBUI_PRINT_PREVIEW_UI_H_