1 files changed, 9 insertions, 0 deletions
diff --git a/Tools/WebKitTestRunner/TestController.h b/Tools/WebKitTestRunner/TestController.h
index 0d33717b7..8125a3646 100644
--- a/Tools/WebKitTestRunner/TestController.h
+++ b/Tools/WebKitTestRunner/TestController.h
@@ -27,6 +27,7 @@
 #define TestController_h
 
 #include "WebNotificationProvider.h"
+#include <GeolocationProviderMock.h>
 #include <WebKit2/WKRetainPtr.h>
 #include <string>
 #include <vector>
@@ -65,6 +66,11 @@ public:
 
     void simulateWebNotificationClick(uint64_t notificationID);
 
+    // Geolocation.
+    void setGeolocationPermission(bool enabled) { m_isGeolocationPermissionAllowed = enabled; }
+    bool isGeolocationPermissionAllowed() const { return m_isGeolocationPermissionAllowed; }
+    void setMockGeolocationPosition(double latitude, double longitude, double accuracy);
+
     bool resetStateToConsistentValues();
 
 private:
@@ -123,6 +129,7 @@ private:
     OwnPtr<PlatformWebView> m_mainWebView;
     WKRetainPtr<WKContextRef> m_context;
     WKRetainPtr<WKPageGroupRef> m_pageGroup;
+    OwnPtr<GeolocationProviderMock> m_geolocationProvider;
 
     enum State {
         Initial,
@@ -143,6 +150,8 @@ private:
     
     bool m_beforeUnloadReturnValue;
 
+    bool m_isGeolocationPermissionAllowed;
+
     EventSenderProxy* m_eventSenderProxy;
 };