1 files changed, 9 insertions, 1 deletions
diff --git a/WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h b/WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h
index ba021b1..d094f42 100644
--- a/WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h
+++ b/WebKitTools/WebKitTestRunner/InjectedBundle/InjectedBundle.h
@@ -61,6 +61,8 @@ public:
     void done();
     std::ostringstream& os() { return m_outputStream; }
 
+    bool isTestRunning() { return m_state == Testing; }
+
 private:
     InjectedBundle();
     ~InjectedBundle();
@@ -73,7 +75,7 @@ private:
     void willDestroyPage(WKBundlePageRef page);
     void didReceiveMessage(WKStringRef messageName, WKTypeRef messageBody);
 
-    void reset();
+    void beginTesting();
 
     WKBundleRef m_bundle;
     HashMap<WKBundlePageRef, InjectedBundlePage*> m_otherPages;
@@ -84,6 +86,12 @@ private:
     RefPtr<EventSendingController> m_eventSendingController;
 
     std::ostringstream m_outputStream;
+    
+    enum State {
+        Idle,
+        Testing
+    };
+    State m_state;
 };
 
 } // namespace WTR