diff -r 000000000000 -r 4f2f89ce4247 WebKitTools/WebKitTestRunner/TestInvocation.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/WebKitTools/WebKitTestRunner/TestInvocation.cpp Fri Sep 17 09:02:29 2010 +0300 @@ -0,0 +1,165 @@ +/* + * Copyright (C) 2010 Apple Inc. All rights reserved. + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions + * are met: + * 1. Redistributions of source code must retain the above copyright + * notice, this list of conditions and the following disclaimer. + * 2. Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * + * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, + * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR + * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS + * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF + * THE POSSIBILITY OF SUCH DAMAGE. + */ + +#include "TestInvocation.h" + +#include "PlatformWebView.h" +#include "TestController.h" +#include +#include +#include +#include +#include +#include +#include + +using namespace WebKit; + +namespace WTR { + +static WKURLRef createWKURL(const char* pathOrURL) +{ + RetainPtr pathOrURLCFString(AdoptCF, CFStringCreateWithCString(0, pathOrURL, kCFStringEncodingUTF8)); + RetainPtr cfURL; + if (CFStringHasPrefix(pathOrURLCFString.get(), CFSTR("http://")) || CFStringHasPrefix(pathOrURLCFString.get(), CFSTR("https://"))) + cfURL.adoptCF(CFURLCreateWithString(0, pathOrURLCFString.get(), 0)); + else +#if defined(WIN32) || defined(_WIN32) + cfURL.adoptCF(CFURLCreateWithFileSystemPath(0, pathOrURLCFString.get(), kCFURLWindowsPathStyle, false)); +#else + cfURL.adoptCF(CFURLCreateWithFileSystemPath(0, pathOrURLCFString.get(), kCFURLPOSIXPathStyle, false)); +#endif + return WKURLCreateWithCFURL(cfURL.get()); +} + +static PassOwnPtr > WKStringToUTF8(WKStringRef wkStringRef) +{ + RetainPtr cfString(AdoptCF, WKStringCopyCFString(0, wkStringRef)); + CFIndex bufferLength = CFStringGetMaximumSizeForEncoding(CFStringGetLength(cfString.get()), kCFStringEncodingUTF8) + 1; + OwnPtr > buffer(new Vector(bufferLength)); + if (!CFStringGetCString(cfString.get(), buffer->data(), bufferLength, kCFStringEncodingUTF8)) { + buffer->shrink(1); + (*buffer)[0] = 0; + } else + buffer->shrink(strlen(buffer->data()) + 1); + return buffer.release(); +} + +TestInvocation::TestInvocation(const char* pathOrURL) + : m_url(AdoptWK, createWKURL(pathOrURL)) + , m_pathOrURL(fastStrDup(pathOrURL)) + , m_gotInitialResponse(false) + , m_gotFinalMessage(false) + , m_error(false) +{ +} + +TestInvocation::~TestInvocation() +{ + fastFree(m_pathOrURL); +} + +static const unsigned w3cSVGWidth = 480; +static const unsigned w3cSVGHeight = 360; +static const unsigned normalWidth = 800; +static const unsigned normalHeight = 600; + +static void sizeWebViewForCurrentTest(char* pathOrURL) +{ + bool isSVGW3CTest = strstr(pathOrURL, "svg/W3C-SVG-1.1"); + + if (isSVGW3CTest) + TestController::shared().mainWebView()->resizeTo(w3cSVGWidth, w3cSVGHeight); + else + TestController::shared().mainWebView()->resizeTo(normalWidth, normalHeight); +} + +void TestInvocation::resetPreferencesToConsistentValues() +{ + WKPreferencesRef preferences = WKContextGetPreferences(TestController::shared().context()); + + WKPreferencesSetOfflineWebApplicationCacheEnabled(preferences, true); +} + +void TestInvocation::invoke() +{ + sizeWebViewForCurrentTest(m_pathOrURL); + resetPreferencesToConsistentValues(); + + WKRetainPtr message(AdoptWK, WKStringCreateWithCFString(CFSTR("BeginTest"))); + WKContextPostMessageToInjectedBundle(TestController::shared().context(), message.get()); + + runUntil(m_gotInitialResponse); + if (m_error) { + dump("FAIL\n"); + return; + } + + WKPageLoadURL(TestController::shared().mainWebView()->page(), m_url.get()); + + runUntil(m_gotFinalMessage); + if (m_error) { + dump("FAIL\n"); + return; + } +} + +void TestInvocation::dump(const char* stringToDump) +{ + printf("Content-Type: text/plain\n"); + printf("%s", stringToDump); + + fputs("#EOF\n", stdout); + fputs("#EOF\n", stdout); + fputs("#EOF\n", stderr); + + fflush(stdout); + fflush(stderr); +} + +void TestInvocation::didRecieveMessageFromInjectedBundle(WKStringRef message) +{ + RetainPtr cfMessage(AdoptCF, WKStringCopyCFString(0, message)); + + if (CFEqual(cfMessage.get(), CFSTR("Error"))) { + // Set all states to true to stop spinning the runloop. + m_gotInitialResponse = true; + m_gotFinalMessage = true; + m_error = true; + return; + } + + if (CFEqual(cfMessage.get(), CFSTR("BeginTestAck"))) { + m_gotInitialResponse = true; + return; + } + + OwnPtr > utf8Message = WKStringToUTF8(message); + + dump(utf8Message->data()); + m_gotFinalMessage = true; +} + +} // namespace WTR