1 files changed, 12 insertions, 1 deletions
diff --git a/Tools/WebKitTestRunner/InjectedBundle/InjectedBundlePage.cpp b/Tools/WebKitTestRunner/InjectedBundle/InjectedBundlePage.cpp
index 57d2e1496..dc2ad3d39 100644
--- a/Tools/WebKitTestRunner/InjectedBundle/InjectedBundlePage.cpp
+++ b/Tools/WebKitTestRunner/InjectedBundle/InjectedBundlePage.cpp
@@ -330,7 +330,7 @@ void InjectedBundlePage::stopLoading()
     WKBundlePageStopLoading(m_page);
 }
 
-void InjectedBundlePage::reset()
+void InjectedBundlePage::prepare()
 {
     WKBundlePageClearMainFrameName(m_page);
 
@@ -347,6 +347,17 @@ void InjectedBundlePage::reset()
     WKBundlePageSetTracksRepaints(m_page, false);
 }
 
+void InjectedBundlePage::resetAfterTest()
+{
+    WKBundleFrameRef frame = WKBundlePageGetMainFrame(m_page);
+    JSGlobalContextRef context = WKBundleFrameGetJavaScriptContext(frame);
+#if PLATFORM(QT)
+    DumpRenderTreeSupportQt::injectInternalsObject(context);
+#else
+    WebCoreTestSupport::resetInternalsObject(context);
+#endif
+}
+
 // Loader Client Callbacks
 
 // String output must be identical to -[WebFrame _drt_descriptionSuitableForTestResult].