Implement Printer Preference Persistence (Step 1/3)

chrome/browser/printing/page_overlays_unittest.cc

-// Copyright (c) 2006-2008 The Chromium Authors. All rights reserved.
+// Copyright (c) 2006-2009 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 "base/message_loop.h"
 #include "base/string_util.h"
 #include "chrome/browser/printing/page_overlays.h"
 #include "chrome/browser/printing/print_settings.h"
 #include "chrome/browser/printing/printed_document.h"
 #include "chrome/browser/printing/printed_page.h"
 #include "chrome/browser/printing/printed_pages_source.h"
@@ skipping 54 matching lines @@
   for (int i = 0; i < arraysize(kOverlayKeys); ++i) {
     input = StringPrintf(L"foo%lsbar", kOverlayKeys[i].key);
     out = printing::PageOverlays::ReplaceVariables(input, *doc.get(),
                                                    *page.get());
     EXPECT_FALSE(out.empty());
     if (wcslen(kOverlayKeys[i].expected) == 0)
       continue;
     EXPECT_EQ(StringPrintf(L"foo%lsbar", kOverlayKeys[i].expected), out) <<
         kOverlayKeys[i].key;
   }
+
+  // Check if SetOverlay really sets the page overlay.
+  overlays.SetOverlay(printing::PageOverlays::LEFT,
+                      printing::PageOverlays::TOP,
+                      UTF16ToWide(L"Page {page}"));
+  input = overlays.GetOverlay(printing::PageOverlays::LEFT,
+                              printing::PageOverlays::TOP);
+  EXPECT_EQ(input, L"Page {page}");
+
+  // Replace the variables to see if the page number is correct.
+  out = printing::PageOverlays::ReplaceVariables(input, *doc.get(),
+                                                 *page.get());
+  EXPECT_EQ(out, L"Page 1");
 }