Implement layoutTestController.numberOfPages.

webkit/tools/test_shell/layout_test_controller.cc

 // Copyright (c) 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.
 
 // This file contains the definition for LayoutTestController.
 
 #include <vector>
 
 #include "webkit/tools/test_shell/layout_test_controller.h"
 
@@ skipping 111 matching lines @@
   BindMethod("waitForPolicyDelegate", &LayoutTestController::waitForPolicyDelegate);
   BindMethod("setWillSendRequestReturnsNullOnRedirect", &LayoutTestController::setWillSendRequestReturnsNullOnRedirect);
   BindMethod("setWillSendRequestReturnsNull", &LayoutTestController::setWillSendRequestReturnsNull);
   BindMethod("whiteListAccessFromOrigin", &LayoutTestController::whiteListAccessFromOrigin);
   BindMethod("clearAllDatabases", &LayoutTestController::clearAllDatabases);
   BindMethod("setDatabaseQuota", &LayoutTestController::setDatabaseQuota);
   BindMethod("setPOSIXLocale", &LayoutTestController::setPOSIXLocale);
   BindMethod("counterValueForElementById", &LayoutTestController::counterValueForElementById);
   BindMethod("addUserScript", &LayoutTestController::addUserScript);
   BindMethod("pageNumberForElementById", &LayoutTestController::pageNumberForElementById);
+  BindMethod("numberOfPages", &LayoutTestController::numberOfPages);
 
   // The following are stubs.
   BindMethod("dumpAsWebArchive", &LayoutTestController::dumpAsWebArchive);
   BindMethod("setMainFrameIsFirstResponder", &LayoutTestController::setMainFrameIsFirstResponder);
   BindMethod("dumpSelectionRect", &LayoutTestController::dumpSelectionRect);
   BindMethod("display", &LayoutTestController::display);
   BindMethod("testRepaint", &LayoutTestController::testRepaint);
   BindMethod("repaintSweepHorizontally", &LayoutTestController::repaintSweepHorizontally);
   BindMethod("clearBackForwardList", &LayoutTestController::clearBackForwardList);
   BindMethod("keepWebHistory", &LayoutTestController::keepWebHistory);
@@ skipping 922 matching lines @@
   if (args.size() < 1 || !args[0].isString())
     return;
   std::wstring counterValue;
   if (!webkit_glue::CounterValueForElementById(shell_->webView()->mainFrame(),
                                                args[0].ToString(),
                                                &counterValue))
     return;
   result->Set(WideToUTF8(counterValue));
 }
 
+static bool ParsePageSizeParameters(const CppArgumentList& args,
+                                    int arg_offset,
+                                    float* page_width_in_pixels,
+                                    float* page_height_in_pixels) {
+  // WebKit is using the window width/height of DumpRenderTree as the
+  // default value of the page size.
+  // TODO(hamaji): Once chromium DumpRenderTree is implemented,
+  //               share these values with other ports.
+  *page_width_in_pixels = 800;
+  *page_height_in_pixels = 600;
+  switch (args.size() - arg_offset) {
+  case 2:
+    if (!args[arg_offset].isNumber() || !args[1 + arg_offset].isNumber())
+      return false;
+    *page_width_in_pixels = static_cast<float>(args[arg_offset].ToInt32());
+    *page_height_in_pixels = static_cast<float>(args[1 + arg_offset].ToInt32());
+  // fall through.
+  case 0:
+    break;
+  default:
+    return false;
+  }
+  return true;
+}
+
 void LayoutTestController::pageNumberForElementById(
     const CppArgumentList& args, CppVariant* result) {
   result->SetNull();
-  // WebKit is using the window width/height of DumpRenderTree as the
-  // default value of the page size.
-  // TODO(hamaji): Once chromium DumpRenderTree is implemented,
-  //               share these values with other ports.
-  float page_width_in_pixels = 800;
-  float page_height_in_pixels = 600;
-  switch (args.size()) {
-  case 3:
-    if (!args[1].isNumber() || !args[2].isNumber())
-      return;
-    page_width_in_pixels = static_cast<float>(args[1].ToInt32());
-    page_height_in_pixels = static_cast<float>(args[2].ToInt32());
-  case 1:  // fall through.
-    if (!args[0].isString())
-      return;
-    break;
-  default:
+  float page_width_in_pixels = 0;
+  float page_height_in_pixels = 0;
+  if (!ParsePageSizeParameters(args, 1,
+                               &page_width_in_pixels, &page_height_in_pixels))
     return;
-  }
+  if (!args[0].isString())
+    return;
+
   int page_number = webkit_glue::PageNumberForElementById(
       shell_->webView()->mainFrame(),
       args[0].ToString(),
       page_width_in_pixels,
       page_height_in_pixels);
   result->Set(page_number);
 }
 
+void LayoutTestController::numberOfPages(
+    const CppArgumentList& args, CppVariant* result) {
+  result->SetNull();
+  float page_width_in_pixels = 0;
+  float page_height_in_pixels = 0;
+  if (!ParsePageSizeParameters(args, 0,
+                               &page_width_in_pixels, &page_height_in_pixels))
+    return;
+
+  int page_number = webkit_glue::NumberOfPages(shell_->webView()->mainFrame(),
+                                               page_width_in_pixels,
+                                               page_height_in_pixels);
+  result->Set(page_number);
+}
+
 void LayoutTestController::LogErrorToConsole(const std::string& text) {
   shell_->delegate()->didAddMessageToConsole(
       WebConsoleMessage(WebConsoleMessage::LevelError,
                         WebString::fromUTF8(text)),
       WebString(), 0);
 }
 
 void LayoutTestController::setTimelineProfilingEnabled(
     const CppArgumentList& args, CppVariant* result) {
   result->SetNull();
@@ skipping 19 matching lines @@
 }
 
 void LayoutTestController::addUserScript(const CppArgumentList& args,
                                          CppVariant* result) {
   result->SetNull();
   if (args.size() < 1 || !args[0].isString() || !args[1].isBool())
     return;
   shell_->webView()->addUserScript(WebString::fromUTF8(args[0].ToString()),
                                    args[1].ToBoolean());
 }