Client-based geolocation support.

webkit/tools/test_shell/layout_test_controller.cc

 // Copyright (c) 2010 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"
 
 #include "base/base64.h"
 #include "base/basictypes.h"
 #include "base/file_path.h"
 #include "base/file_util.h"
 #include "base/logging.h"
 #include "base/message_loop.h"
 #include "base/path_service.h"
 #include "base/string_number_conversions.h"
 #include "base/utf_string_conversions.h"
 #include "net/base/net_util.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebAnimationController.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebBindings.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebConsoleMessage.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebDeviceOrientation.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebDeviceOrientationClientMock.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebDocument.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebElement.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebFrame.h"
+#if defined(ENABLE_CLIENT_BASED_GEOLOCATION)
+#include "third_party/WebKit/WebKit/chromium/public/WebGeolocationClientMock.h"
+#else
 #include "third_party/WebKit/WebKit/chromium/public/WebGeolocationServiceMock.h"
+#endif
 #include "third_party/WebKit/WebKit/chromium/public/WebKit.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebScriptSource.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebSecurityPolicy.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebSettings.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebSize.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebSpeechInputControllerMock.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebURL.h"
 #include "third_party/WebKit/WebKit/chromium/public/WebView.h"
 #include "webkit/glue/dom_operations.h"
 #include "webkit/glue/webkit_glue.h"
 #include "webkit/glue/webpreferences.h"
 #include "webkit/tools/test_shell/notification_presenter.h"
 #include "webkit/tools/test_shell/simple_database_system.h"
 #include "webkit/tools/test_shell/simple_resource_loader_bridge.h"
 #include "webkit/tools/test_shell/test_navigation_controller.h"
 #include "webkit/tools/test_shell/test_shell.h"
 #include "webkit/tools/test_shell/test_shell_devtools_agent.h"
 #include "webkit/tools/test_shell/test_webview_delegate.h"
 
 using std::string;
 using std::wstring;
 
 using WebKit::WebBindings;
 using WebKit::WebConsoleMessage;
 using WebKit::WebElement;
-using WebKit::WebGeolocationServiceMock;
 using WebKit::WebScriptSource;
 using WebKit::WebSecurityPolicy;
 using WebKit::WebSize;
 using WebKit::WebString;
 using WebKit::WebURL;
 
+#if !defined(ENABLE_CLIENT_BASED_GEOLOCATION)
+using WebKit::WebGeolocationServiceMock;
+#endif
+
 TestShell* LayoutTestController::shell_ = NULL;
 // Most of these flags need to be cleared in Reset() so that they get turned
 // off between each test run.
 bool LayoutTestController::generate_pixel_results_ = true;
 bool LayoutTestController::dump_as_text_ = false;
 bool LayoutTestController::dump_editing_callbacks_ = false;
 bool LayoutTestController::dump_frame_load_callbacks_ = false;
 bool LayoutTestController::dump_resource_load_callbacks_ = false;
 bool LayoutTestController::dump_resource_response_mime_types_ = false;
 bool LayoutTestController::dump_back_forward_list_ = false;
@@ skipping 1414 matching lines @@
                       "Should be 'mac', 'win', or 'unix'.");
     return;
   }
   shell_->webView()->settings()->setEditingBehavior(behavior);
 }
 
 void LayoutTestController::setGeolocationPermission(const CppArgumentList& args,
                                                     CppVariant* result) {
   if (args.size() < 1 || !args[0].isBool())
     return;
+#if defined(ENABLE_CLIENT_BASED_GEOLOCATION)
+  shell_->geolocation_client_mock()->setPermission(
+      args[0].ToBoolean());
+#else
   shell_->delegate()->SetGeolocationPermission(args[0].ToBoolean());
+#endif
 }
 
 void LayoutTestController::setMockGeolocationPosition(
     const CppArgumentList& args, CppVariant* result) {
   if (args.size() < 3 ||
       !args[0].isNumber() || !args[1].isNumber() || !args[2].isNumber())
     return;
+#if defined(ENABLE_CLIENT_BASED_GEOLOCATION)
+  shell_->geolocation_client_mock()->setPosition(
+      args[0].ToDouble(), args[1].ToDouble(), args[2].ToDouble());
+#else
   WebGeolocationServiceMock::setMockGeolocationPosition(
       args[0].ToDouble(), args[1].ToDouble(), args[2].ToDouble());
+#endif
 }
 
 void LayoutTestController::setMockGeolocationError(const CppArgumentList& args,
                                                    CppVariant* result) {
   if (args.size() < 2 || !args[0].isNumber() || !args[1].isString())
     return;
+#if defined(ENABLE_CLIENT_BASED_GEOLOCATION)
+  shell_->geolocation_client_mock()->setError(
+      args[0].ToInt32(), WebString::fromUTF8(args[1].ToString()));
+#else
   WebGeolocationServiceMock::setMockGeolocationError(
       args[0].ToInt32(), WebString::fromUTF8(args[1].ToString()));
+#endif
 }
 
 void LayoutTestController::markerTextForListItem(const CppArgumentList& args,
                                                  CppVariant* result) {
   WebElement element;
   if (!WebBindings::getElement(args[0].value.objectValue, &element))
     result->SetNull();
   else
     result->Set(
         element.document().frame()->markerTextForListItem(element).utf8());
@@ skipping 16 matching lines @@
   shell_->device_orientation_client_mock()->setOrientation(orientation);
 }
 
 void LayoutTestController::hasSpellingMarker(const CppArgumentList& arguments,
                                              CppVariant* result) {
   if (arguments.size() < 2 || !arguments[0].isNumber() || !arguments[1].isNumber())
     return;
   result->Set(shell_->webView()->mainFrame()->selectionStartHasSpellingMarkerFor(
       arguments[0].ToInt32(), arguments[1].ToInt32()));
 }