2010-10-06 Balazs Kelemen <kbalazs@webkit.org>
[WebKit-https.git] / WebKitTools / WebKitTestRunner / TestController.h
1 /*
2  * Copyright (C) 2010 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #ifndef TestController_h
27 #define TestController_h
28
29 #include <WebKit2/WKRetainPtr.h>
30 #include <string>
31 #include <vector>
32 #include <wtf/OwnPtr.h>
33
34 namespace WTR {
35
36 class TestInvocation;
37 class PlatformWebView;
38
39 // FIXME: Rename this TestRunner?
40 class TestController {
41 public:
42     static TestController& shared();
43
44     TestController(int argc, const char* argv[]);
45     ~TestController();
46
47     bool verbose() const { return m_verbose; }
48
49     WKStringRef injectedBundlePath() { return m_injectedBundlePath.get(); }
50     WKStringRef testPluginDirectory() { return m_testPluginDirectory.get(); }
51
52     PlatformWebView* mainWebView() { return m_mainWebView.get(); }
53     WKPageNamespaceRef pageNamespace() { return m_pageNamespace.get(); }
54     WKContextRef context() { return m_context.get(); }
55
56     // Helper
57     static void runUntil(bool& done);
58
59 private:
60     void initialize(int argc, const char* argv[]);
61     void run();
62
63     void runTestingServerLoop();
64     void runTest(const char* pathOrURL);
65
66     void platformInitialize();
67     void platformInitializeContext();
68     void initializeInjectedBundlePath();
69     void initializeTestPluginDirectory();
70
71     void resetStateToConsistentValues();
72
73     // WKContextInjectedBundleClient
74     static void didReceiveMessageFromInjectedBundle(WKContextRef context, WKStringRef messageName, WKTypeRef messageBody, const void*);
75     void didReceiveMessageFromInjectedBundle(WKStringRef messageName, WKTypeRef messageBody);
76
77     // WKPageLoaderClient
78     static void didFinishLoadForFrame(WKPageRef page, WKFrameRef frame, WKTypeRef userData, const void*);
79     void didFinishLoadForFrame(WKPageRef page, WKFrameRef frame);
80
81
82     OwnPtr<TestInvocation> m_currentInvocation;
83
84     bool m_dumpPixels;
85     bool m_verbose;
86     bool m_printSeparators;
87     bool m_usingServerMode;
88     std::vector<std::string> m_paths;
89     WKRetainPtr<WKStringRef> m_injectedBundlePath;
90     WKRetainPtr<WKStringRef> m_testPluginDirectory;
91
92     OwnPtr<PlatformWebView> m_mainWebView;
93     WKRetainPtr<WKContextRef> m_context;
94     WKRetainPtr<WKPageNamespaceRef> m_pageNamespace;
95     
96     enum State {
97         Initial,
98         Resetting,
99         RunningTest
100     };
101     State m_state;
102     bool m_doneResetting;
103 };
104
105 } // namespace WTR
106
107 #endif // TestController_h