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'' AND ANY
14 * EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
15 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
16 * DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS BE LIABLE FOR ANY
17 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
18 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
19 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
20 * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
21 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
22 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
23 */
24
25#include "config.h"
26#include "TestsController.h"
27
28#include "Test.h"
29#include <wtf/PassOwnPtr.h>
30
31using namespace std;
32
33namespace WebKitAPITest {
34
35static const LPCWSTR testsControllerWindowClassName = L"TestsControllerWindowClass";
36
37enum { runNextTestTimerID = 1 };
38
39inline TestsController::TestsController()
40    : m_testFailed(false)
41    , m_anyTestFailed(false)
42{
43    registerWindowClass();
44    m_window = CreateWindowExW(0, testsControllerWindowClassName, 0, WS_CHILD, 0, 0, 0, 0, HWND_MESSAGE, 0, GetModuleHandle(0), 0);
45}
46
47TestsController& TestsController::shared()
48{
49    static TestsController& shared = *new TestsController;
50    return shared;
51}
52
53bool TestsController::runAllTests()
54{
55    if (m_tests.isEmpty())
56        return true;
57
58    MSG msg;
59    BOOL result;
60    while ((result = GetMessage(&msg, 0, 0, 0))) {
61        if (result == -1)
62            break;
63        TranslateMessage(&msg);
64        DispatchMessage(&msg);
65    }
66
67    if (msg.message != WM_QUIT)
68        return false;
69
70    return !m_anyTestFailed;
71}
72
73void TestsController::addTest(PassOwnPtr<Test> test)
74{
75    m_tests.append(test.leakPtr());
76    runNextTestSoon();
77}
78
79void TestsController::testFailed(const char* file, int line, const char* message)
80{
81    ASSERT(!m_tests.isEmpty());
82
83    m_testFailed = true;
84    m_anyTestFailed = true;
85
86    printf("FAIL: %s\n\t%s (%s:%d)\n", m_tests.first()->name(), message, file, line);
87    fflush(stdout);
88}
89
90void TestsController::runNextTest()
91{
92    if (m_tests.isEmpty()) {
93        PostQuitMessage(0);
94        return;
95    }
96
97    Test* test = m_tests.first();
98
99    m_testFailed = false;
100    printf("RUN: %s\n", test->name());
101    fflush(stdout);
102    test->run();
103
104    if (!m_testFailed) {
105        printf("PASS: %s\n", test->name());
106        fflush(stdout);
107    }
108
109    m_tests.removeFirst();
110    delete test;
111
112    runNextTestSoon();
113}
114
115void TestsController::runNextTestSoon()
116{
117    SetTimer(m_window, runNextTestTimerID, 0, 0);
118}
119
120void TestsController::registerWindowClass()
121{
122    static bool initialized;
123    if (initialized)
124        return;
125    initialized = true;
126
127    WNDCLASSEXW wndClass = {0};
128    wndClass.cbSize = sizeof(wndClass);
129    wndClass.lpfnWndProc = wndProc;
130    wndClass.hCursor = LoadCursor(0, IDC_ARROW);
131    wndClass.hInstance = GetModuleHandle(0);
132    wndClass.lpszClassName = testsControllerWindowClassName;
133
134    RegisterClassExW(&wndClass);
135}
136
137LRESULT TestsController::wndProc(HWND hWnd, UINT uMsg, WPARAM wParam, LPARAM lParam)
138{
139    if (uMsg == WM_TIMER && wParam == runNextTestTimerID) {
140        KillTimer(hWnd, runNextTestTimerID);
141        TestsController::shared().runNextTest();
142        return 0;
143    }
144
145    return DefWindowProcW(hWnd, uMsg, wParam, lParam);
146}
147
148} // namespace WebKitAPITest
149