2 files changed, 16 insertions, 0 deletions
diff --git a/webkit/support/webkit_support.cc b/webkit/support/webkit_support.cc
index 5a22040..48eb89d 100644
--- a/webkit/support/webkit_support.cc
+++ b/webkit/support/webkit_support.cc
@@ -59,6 +59,11 @@ class TestEnvironment {
 
   WebKit::WebKitClient* webkit_client() { return webkit_client_.get(); }
 
+  void set_theme_engine(WebKit::WebThemeEngine* engine) {
+    DCHECK(webkit_client_ != 0);
+    webkit_client_->SetThemeEngine(engine);
+  }
+
  private:
   scoped_ptr<base::AtExitManager> at_exit_manager_;
   scoped_ptr<MessageLoopForUI> main_message_loop_;
@@ -292,4 +297,11 @@ void SetAcceptAllCookies(bool accept) {
   SimpleResourceLoaderBridge::SetAcceptAllCookies(accept);
 }
 
+// Theme engine
+
+void SetThemeEngine(WebKit::WebThemeEngine* engine) {
+  DCHECK(test_environment);
+  test_environment->set_theme_engine(engine);
+}
+
 }  // namespace webkit_support
diff --git a/webkit/support/webkit_support.h b/webkit/support/webkit_support.h
index bfc87f0..7375568 100644
--- a/webkit/support/webkit_support.h
+++ b/webkit/support/webkit_support.h
@@ -19,6 +19,7 @@ class WebMediaPlayer;
 class WebMediaPlayerClient;
 class WebPlugin;
 class WebString;
+class WebThemeEngine;
 class WebURL;
 struct WebPluginParams;
 }
@@ -97,6 +98,9 @@ void ClearAllDatabases();
 // - Resource loader
 void SetAcceptAllCookies(bool accept);
 
+// - Theme engine
+void SetThemeEngine(WebKit::WebThemeEngine* engine);
+
 }  // namespace webkit_support
 
 #endif  // WEBKIT_SUPPORT_WEBIT_CLIENT_IMPL_H_