Tools/TestWebKitAPI/JavaScriptTest.cpp

   1 /*
   2  * Copyright (C) 2011 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 #include "config.h"
  27 #include "JavaScriptTest.h"
  28
  29 #include "PlatformUtilities.h"
  30 #include "Test.h"
  31 #include <JavaScriptCore/JSContextRef.h>
  32 #include <JavaScriptCore/JSRetainPtr.h>
  33 #include <WebKit2/WKRetainPtr.h>
  34 #include <WebKit2/WKSerializedScriptValue.h>
  35 #include <wtf/OwnArrayPtr.h>
  36
  37 namespace TestWebKitAPI {
  38
  39 struct JavaScriptCallbackContext {
  40     JavaScriptCallbackContext() : didFinish(false) { }
  41
  42     bool didFinish;
  43     JSRetainPtr<JSStringRef> actualString;
  44 };
  45
  46 static void javaScriptCallback(WKSerializedScriptValueRef resultSerializedScriptValue, WKErrorRef error, void* ctx)
  47 {
  48     ASSERT_NOT_NULL(resultSerializedScriptValue);
  49
  50     JavaScriptCallbackContext* context = static_cast<JavaScriptCallbackContext*>(ctx);
  51
  52     JSGlobalContextRef scriptContext = JSGlobalContextCreate(0);
  53     ASSERT_NOT_NULL(scriptContext);
  54
  55     JSValueRef scriptValue = WKSerializedScriptValueDeserialize(resultSerializedScriptValue, scriptContext, 0);
  56     ASSERT_NOT_NULL(scriptValue);
  57
  58     context->actualString.adopt(JSValueToStringCopy(scriptContext, scriptValue, 0));
  59     ASSERT_NOT_NULL(context->actualString.get());
  60
  61     context->didFinish = true;
  62
  63     JSGlobalContextRelease(scriptContext);
  64
  65     EXPECT_NULL(error);
  66 }
  67
  68 ::testing::AssertionResult runJSTest(const char*, const char*, const char*, WKPageRef page, const char* script, const char* expectedResult)
  69 {
  70     JavaScriptCallbackContext context;
  71     WKPageRunJavaScriptInMainFrame(page, Util::toWK(script).get(), &context, javaScriptCallback);
  72     Util::run(&context.didFinish);
  73
  74     size_t bufferSize = JSStringGetMaximumUTF8CStringSize(context.actualString.get());
  75     OwnArrayPtr<char> buffer = adoptArrayPtr(new char[bufferSize]);
  76     JSStringGetUTF8CString(context.actualString.get(), buffer.get(), bufferSize);
  77
  78     return compareJSResult(script, buffer.get(), expectedResult);
  79 }
  80
  81 ::testing::AssertionResult compareJSResult(const char* script, const char* actualResult, const char* expectedResult)
  82 {
  83     if (!strcmp(actualResult, expectedResult))
  84         return ::testing::AssertionSuccess();
  85
  86     return ::testing::AssertionFailure()
  87         << "JS expression: " << script << "\n"
  88         << "       Actual: " << actualResult << "\n"
  89         << "     Expected: " << expectedResult;
  90 }
  91
  92 } // namespace TestWebKitAPI