Fixed print_web_view_helper for CEF.

chrome/renderer/printing/print_web_view_helper_browsertest.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 "base/command_line.h"
 #include "base/run_loop.h"
 #include "chrome/common/chrome_switches.h"
 #include "chrome/common/print_messages.h"
 #include "chrome/renderer/printing/mock_printer.h"
 #include "chrome/renderer/printing/print_web_view_helper.h"
@@ skipping 218 matching lines @@
   virtual ~PrintWebViewHelperTest() {}
 
   virtual void SetUp() OVERRIDE {
     ChromeRenderViewTest::SetUp();
   }
 
  protected:
   DISALLOW_COPY_AND_ASSIGN(PrintWebViewHelperTest);
 };
 
+// This tests only for platforms without print preview.
+#if !defined(ENABLE_FULL_PRINTING)
+// Tests that the renderer blocks window.print() calls if they occur too
+// frequently.
+TEST_F(PrintWebViewHelperTest, BlockScriptInitiatedPrinting) {
+  // Pretend user will cancel printing.
+  chrome_render_thread_->set_print_dialog_user_response(false);
+  // Try to print with window.print() a few times.
+  PrintWithJavaScript();
+  PrintWithJavaScript();
+  PrintWithJavaScript();
+  VerifyPagesPrinted(false);
+
+  // Pretend user will print. (but printing is blocked.)
+  chrome_render_thread_->set_print_dialog_user_response(true);
+  PrintWithJavaScript();
+  VerifyPagesPrinted(false);
+
+  // Unblock script initiated printing and verify printing works.
+  PrintWebViewHelper::Get(view_)->scripting_throttler_.Reset();
+  chrome_render_thread_->printer()->ResetPrinter();
+  PrintWithJavaScript();
+  VerifyPageCount(1);
+  VerifyPagesPrinted(true);
+}
+
+// Tests that the renderer always allows window.print() calls if they are user
+// initiated.
+TEST_F(PrintWebViewHelperTest, AllowUserOriginatedPrinting) {
+  // Pretend user will cancel printing.
+  chrome_render_thread_->set_print_dialog_user_response(false);
+  // Try to print with window.print() a few times.
+  PrintWithJavaScript();
+  PrintWithJavaScript();
+  PrintWithJavaScript();
+  VerifyPagesPrinted(false);
+
+  // Pretend user will print. (but printing is blocked.)
+  chrome_render_thread_->set_print_dialog_user_response(true);
+  PrintWithJavaScript();
+  VerifyPagesPrinted(false);
+
+  // Try again as if user initiated, without resetting the print count.
+  chrome_render_thread_->printer()->ResetPrinter();
+  LoadHTML(kPrintOnUserAction);
+  gfx::Size new_size(200, 100);
+  Resize(new_size, gfx::Rect(), false);
+
+  gfx::Rect bounds = GetElementBounds("print");
+  EXPECT_FALSE(bounds.IsEmpty());
+  blink::WebMouseEvent mouse_event;
+  mouse_event.type = blink::WebInputEvent::MouseDown;
+  mouse_event.button = blink::WebMouseEvent::ButtonLeft;
+  mouse_event.x = bounds.CenterPoint().x();
+  mouse_event.y = bounds.CenterPoint().y();
+  mouse_event.clickCount = 1;
+  SendWebMouseEvent(mouse_event);
+  mouse_event.type = blink::WebInputEvent::MouseUp;
+  SendWebMouseEvent(mouse_event);
+  ProcessPendingMessages();
+
+  VerifyPageCount(1);
+  VerifyPagesPrinted(true);
+}
+
+// Duplicate of OnPrintPagesTest only using javascript to print.
+TEST_F(PrintWebViewHelperTest, PrintWithJavascript) {
+  PrintWithJavaScript();
+
+  VerifyPageCount(1);
+  VerifyPagesPrinted(true);
+}
+#endif  // !ENABLE_FULL_PRINTING
+
 #if !defined(DISABLE_BASIC_PRINTING)
 // Tests that printing pages work and sending and receiving messages through
 // that channel all works.
 TEST_F(PrintWebViewHelperTest, OnPrintPages) {
   LoadHTML(kHelloWorldHTML);
   OnPrintPages();
 
   VerifyPageCount(1);
   VerifyPagesPrinted(true);
 }
@@ skipping 231 matching lines @@
       EXPECT_EQ(margin_right, param.a.margin_right);
       EXPECT_EQ(margin_left, param.a.margin_left);
       EXPECT_EQ(margin_bottom, param.a.margin_bottom);
       EXPECT_EQ(page_has_print_css, param.c);
     }
   }
 
   DISALLOW_COPY_AND_ASSIGN(PrintWebViewHelperPreviewTest);
 };
 
+#if defined(ENABLE_FULL_PRINTING)
 TEST_F(PrintWebViewHelperPreviewTest, BlockScriptInitiatedPrinting) {
   PrintWebViewHelper* print_web_view_helper = PrintWebViewHelper::Get(view_);
   print_web_view_helper->SetScriptedPrintBlocked(true);
   PrintWithJavaScript();
   VerifyPreviewRequest(false);
 
   print_web_view_helper->SetScriptedPrintBlocked(false);
   PrintWithJavaScript();
   VerifyPreviewRequest(true);
 }
@@ skipping 10 matching lines @@
   mouse_event.button = blink::WebMouseEvent::ButtonLeft;
   mouse_event.x = bounds.CenterPoint().x();
   mouse_event.y = bounds.CenterPoint().y();
   mouse_event.clickCount = 1;
   SendWebMouseEvent(mouse_event);
   mouse_event.type = blink::WebInputEvent::MouseUp;
   SendWebMouseEvent(mouse_event);
 
   VerifyPreviewRequest(true);
 }
+#endif  // ENABLE_FULL_PRINTING
 
 // Tests that print preview work and sending and receiving messages through
 // that channel all works.
 TEST_F(PrintWebViewHelperPreviewTest, OnPrintPreview) {
   LoadHTML(kHelloWorldHTML);
 
   // Fill in some dummy values.
   base::DictionaryValue dict;
   CreatePrintSettingsDictionary(&dict);
   OnPrintPreview(dict);
@@ skipping 382 matching lines @@
   CreatePrintSettingsDictionary(&dict);
   OnPrintForPrintPreview(dict);
 
   VerifyPrintFailed(true);
   VerifyPagesPrinted(false);
 }
 
 #endif  // !defined(OS_CHROMEOS)
 
 }  // namespace printing