blob: 2ea3cbd72b42b2a30ca2a76c865c1dab0229ec5a [file] [log] [blame]
/*
* Copyright (C) 2010-2017 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 "config.h"
#include "TestRunner.h"
#include "InjectedBundle.h"
#include "InjectedBundlePage.h"
#include "JSTestRunner.h"
#include "PlatformWebView.h"
#include "StringFunctions.h"
#include "TestController.h"
#include <JavaScriptCore/JSCTestRunnerUtils.h>
#include <WebCore/NetworkStorageSession.h>
#include <WebCore/ResourceLoadObserver.h>
#include <WebKit/WKBundle.h>
#include <WebKit/WKBundleBackForwardList.h>
#include <WebKit/WKBundleFrame.h>
#include <WebKit/WKBundleFramePrivate.h>
#include <WebKit/WKBundleInspector.h>
#include <WebKit/WKBundleNodeHandlePrivate.h>
#include <WebKit/WKBundlePage.h>
#include <WebKit/WKBundlePagePrivate.h>
#include <WebKit/WKBundlePrivate.h>
#include <WebKit/WKBundleScriptWorld.h>
#include <WebKit/WKData.h>
#include <WebKit/WKNumber.h>
#include <WebKit/WKPagePrivate.h>
#include <WebKit/WKRetainPtr.h>
#include <WebKit/WKSerializedScriptValue.h>
#include <WebKit/WebKit2_C.h>
#include <wtf/HashMap.h>
#include <wtf/Optional.h>
#include <wtf/StdLibExtras.h>
#include <wtf/UniqueArray.h>
#include <wtf/text/CString.h>
#include <wtf/text/StringBuilder.h>
#include <wtf/text/StringConcatenateNumbers.h>
namespace WTR {
Ref<TestRunner> TestRunner::create()
{
return adoptRef(*new TestRunner);
}
TestRunner::TestRunner()
: m_userStyleSheetLocation(adoptWK(WKStringCreateWithUTF8CString("")))
{
platformInitialize();
}
TestRunner::~TestRunner()
{
}
JSClassRef TestRunner::wrapperClass()
{
return JSTestRunner::testRunnerClass();
}
void TestRunner::display()
{
WKBundlePageRef page = InjectedBundle::singleton().page()->page();
WKBundlePageForceRepaint(page);
}
void TestRunner::displayAndTrackRepaints()
{
WKBundlePageRef page = InjectedBundle::singleton().page()->page();
WKBundlePageForceRepaint(page);
WKBundlePageSetTracksRepaints(page, true);
WKBundlePageResetTrackedRepaints(page);
}
bool TestRunner::shouldDumpPixels() const
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("GetDumpPixels"));
WKTypeRef returnData = nullptr;
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::setDumpPixels(bool dumpPixels)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetDumpPixels"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(dumpPixels));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::dumpAsText(bool dumpPixels)
{
if (whatToDump() < WhatToDump::MainFrameText)
setWhatToDump(WhatToDump::MainFrameText);
setDumpPixels(dumpPixels);
}
WhatToDump TestRunner::whatToDump() const
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("GetWhatToDump"));
WKTypeRef returnData = nullptr;
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKUInt64GetTypeID());
return static_cast<WhatToDump>(WKUInt64GetValue(adoptWK(static_cast<WKUInt64Ref>(returnData)).get()));
}
void TestRunner::setWhatToDump(WhatToDump whatToDump)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetWhatToDump"));
WKRetainPtr<WKUInt64Ref> messageBody = adoptWK(WKUInt64Create(static_cast<uint64_t>(whatToDump)));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setCustomPolicyDelegate(bool enabled, bool permissive)
{
m_policyDelegateEnabled = enabled;
m_policyDelegatePermissive = permissive;
InjectedBundle::singleton().setCustomPolicyDelegate(enabled, permissive);
}
void TestRunner::waitForPolicyDelegate()
{
setCustomPolicyDelegate(true);
waitUntilDone();
}
void TestRunner::waitUntilDownloadFinished()
{
m_shouldFinishAfterDownload = true;
waitUntilDone();
}
void TestRunner::waitUntilDone()
{
auto& injectedBundle = InjectedBundle::singleton();
RELEASE_ASSERT(injectedBundle.isTestRunning());
setWaitUntilDone(true);
}
void TestRunner::setWaitUntilDone(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetWaitUntilDone"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
bool TestRunner::shouldWaitUntilDone() const
{
WKRetainPtr<WKStringRef> messsageName = adoptWK(WKStringCreateWithUTF8CString("GetWaitUntilDone"));
WKTypeRef returnData = nullptr;
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messsageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::notifyDone()
{
auto& injectedBundle = InjectedBundle::singleton();
if (!injectedBundle.isTestRunning())
return;
if (shouldWaitUntilDone() && !injectedBundle.topLoadingFrame())
injectedBundle.page()->dump();
// We don't call invalidateWaitToDumpWatchdogTimer() here, even if we continue to wait for a load to finish.
// The test is still subject to timeout checking - it is better to detect an async timeout inside WebKitTestRunner
// than to let webkitpy do that, because WebKitTestRunner will dump partial results.
setWaitUntilDone(false);
}
void TestRunner::forceImmediateCompletion()
{
auto& injectedBundle = InjectedBundle::singleton();
if (!injectedBundle.isTestRunning())
return;
if (shouldWaitUntilDone() && injectedBundle.page())
injectedBundle.page()->dump();
setWaitUntilDone(false);
}
void TestRunner::setShouldDumpFrameLoadCallbacks(bool value)
{
WKRetainPtr<WKStringRef> messsageName = adoptWK(WKStringCreateWithUTF8CString("SetDumpFrameLoadCallbacks"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messsageName.get(), messageBody.get(), nullptr);
}
bool TestRunner::shouldDumpFrameLoadCallbacks()
{
WKRetainPtr<WKStringRef> messsageName = adoptWK(WKStringCreateWithUTF8CString("GetDumpFrameLoadCallbacks"));
WKTypeRef returnData = nullptr;
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messsageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
unsigned TestRunner::imageCountInGeneralPasteboard() const
{
return InjectedBundle::singleton().imageCountInGeneralPasteboard();
}
void TestRunner::addUserScript(JSStringRef source, bool runAtStart, bool allFrames)
{
WKRetainPtr<WKStringRef> sourceWK = toWK(source);
WKBundlePageAddUserScript(InjectedBundle::singleton().page()->page(), sourceWK.get(),
(runAtStart ? kWKInjectAtDocumentStart : kWKInjectAtDocumentEnd),
(allFrames ? kWKInjectInAllFrames : kWKInjectInTopFrameOnly));
}
void TestRunner::addUserStyleSheet(JSStringRef source, bool allFrames)
{
WKRetainPtr<WKStringRef> sourceWK = toWK(source);
WKBundlePageAddUserStyleSheet(InjectedBundle::singleton().page()->page(), sourceWK.get(),
(allFrames ? kWKInjectInAllFrames : kWKInjectInTopFrameOnly));
}
void TestRunner::keepWebHistory()
{
InjectedBundle::singleton().postSetAddsVisitedLinks(true);
}
void TestRunner::execCommand(JSStringRef name, JSStringRef showUI, JSStringRef value)
{
WKBundlePageExecuteEditingCommand(InjectedBundle::singleton().page()->page(), toWK(name).get(), toWK(value).get());
}
static Optional<WKFindOptions> findOptionsFromArray(JSValueRef optionsArrayAsValue)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(injectedBundle.page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
auto lengthPropertyName = adopt(JSStringCreateWithUTF8CString("length"));
JSObjectRef optionsArray = JSValueToObject(context, optionsArrayAsValue, 0);
JSValueRef lengthValue = JSObjectGetProperty(context, optionsArray, lengthPropertyName.get(), 0);
if (!JSValueIsNumber(context, lengthValue))
return WTF::nullopt;
WKFindOptions options = 0;
size_t length = static_cast<size_t>(JSValueToNumber(context, lengthValue, 0));
for (size_t i = 0; i < length; ++i) {
JSValueRef value = JSObjectGetPropertyAtIndex(context, optionsArray, i, 0);
if (!JSValueIsString(context, value))
continue;
auto optionName = adopt(JSValueToStringCopy(context, value, 0));
if (JSStringIsEqualToUTF8CString(optionName.get(), "CaseInsensitive"))
options |= kWKFindOptionsCaseInsensitive;
else if (JSStringIsEqualToUTF8CString(optionName.get(), "AtWordStarts"))
options |= kWKFindOptionsAtWordStarts;
else if (JSStringIsEqualToUTF8CString(optionName.get(), "TreatMedialCapitalAsWordStart"))
options |= kWKFindOptionsTreatMedialCapitalAsWordStart;
else if (JSStringIsEqualToUTF8CString(optionName.get(), "Backwards"))
options |= kWKFindOptionsBackwards;
else if (JSStringIsEqualToUTF8CString(optionName.get(), "WrapAround"))
options |= kWKFindOptionsWrapAround;
else if (JSStringIsEqualToUTF8CString(optionName.get(), "StartInSelection")) {
// FIXME: No kWKFindOptionsStartInSelection.
}
}
return options;
}
bool TestRunner::findString(JSStringRef target, JSValueRef optionsArrayAsValue)
{
if (auto options = findOptionsFromArray(optionsArrayAsValue))
return WKBundlePageFindString(InjectedBundle::singleton().page()->page(), toWK(target).get(), *options);
return false;
}
void TestRunner::findStringMatchesInPage(JSStringRef target, JSValueRef optionsArrayAsValue)
{
if (auto options = findOptionsFromArray(optionsArrayAsValue))
return WKBundlePageFindStringMatches(InjectedBundle::singleton().page()->page(), toWK(target).get(), *options);
}
void TestRunner::replaceFindMatchesAtIndices(JSValueRef matchIndicesAsValue, JSStringRef replacementText, bool selectionOnly)
{
auto& bundle = InjectedBundle::singleton();
auto mainFrame = WKBundlePageGetMainFrame(bundle.page()->page());
auto context = WKBundleFrameGetJavaScriptContext(mainFrame);
auto lengthPropertyName = adopt(JSStringCreateWithUTF8CString("length"));
auto matchIndicesObject = JSValueToObject(context, matchIndicesAsValue, 0);
auto lengthValue = JSObjectGetProperty(context, matchIndicesObject, lengthPropertyName.get(), 0);
if (!JSValueIsNumber(context, lengthValue))
return;
auto indices = adoptWK(WKMutableArrayCreate());
auto length = static_cast<size_t>(JSValueToNumber(context, lengthValue, 0));
for (size_t i = 0; i < length; ++i) {
auto value = JSObjectGetPropertyAtIndex(context, matchIndicesObject, i, 0);
if (!JSValueIsNumber(context, value))
continue;
auto index = adoptWK(WKUInt64Create(std::round(JSValueToNumber(context, value, nullptr))));
WKArrayAppendItem(indices.get(), index.get());
}
WKBundlePageReplaceStringMatches(bundle.page()->page(), indices.get(), toWK(replacementText).get(), selectionOnly);
}
void TestRunner::clearAllDatabases()
{
WKBundleClearAllDatabases(InjectedBundle::singleton().bundle());
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("DeleteAllIndexedDatabases"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(true));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setDatabaseQuota(uint64_t quota)
{
return WKBundleSetDatabaseQuota(InjectedBundle::singleton().bundle(), quota);
}
void TestRunner::syncLocalStorage()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SyncLocalStorage"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(true));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::clearAllApplicationCaches()
{
WKBundlePageClearApplicationCache(InjectedBundle::singleton().page()->page());
}
void TestRunner::clearApplicationCacheForOrigin(JSStringRef origin)
{
WKBundlePageClearApplicationCacheForOrigin(InjectedBundle::singleton().page()->page(), toWK(origin).get());
}
void TestRunner::setAppCacheMaximumSize(uint64_t size)
{
WKBundlePageSetAppCacheMaximumSize(InjectedBundle::singleton().page()->page(), size);
}
long long TestRunner::applicationCacheDiskUsageForOrigin(JSStringRef origin)
{
return WKBundlePageGetAppCacheUsageForOrigin(InjectedBundle::singleton().page()->page(), toWK(origin).get());
}
void TestRunner::disallowIncreaseForApplicationCacheQuota()
{
m_disallowIncreaseForApplicationCacheQuota = true;
}
static inline JSValueRef stringArrayToJS(JSContextRef context, WKArrayRef strings)
{
const size_t count = WKArrayGetSize(strings);
JSValueRef arrayResult = JSObjectMakeArray(context, 0, 0, 0);
JSObjectRef arrayObj = JSValueToObject(context, arrayResult, 0);
for (size_t i = 0; i < count; ++i) {
WKStringRef stringRef = static_cast<WKStringRef>(WKArrayGetItemAtIndex(strings, i));
JSRetainPtr<JSStringRef> stringJS = toJS(stringRef);
JSObjectSetPropertyAtIndex(context, arrayObj, i, JSValueMakeString(context, stringJS.get()), 0);
}
return arrayResult;
}
JSValueRef TestRunner::originsWithApplicationCache()
{
WKBundlePageRef page = InjectedBundle::singleton().page()->page();
WKRetainPtr<WKArrayRef> origins = adoptWK(WKBundlePageCopyOriginsWithApplicationCache(page));
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(page);
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
return stringArrayToJS(context, origins.get());
}
bool TestRunner::isCommandEnabled(JSStringRef name)
{
return WKBundlePageIsEditingCommandEnabled(InjectedBundle::singleton().page()->page(), toWK(name).get());
}
void TestRunner::setCanOpenWindows()
{
WKRetainPtr<WKStringRef> messsageName = adoptWK(WKStringCreateWithUTF8CString("SetCanOpenWindows"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(true));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messsageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setXSSAuditorEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitXSSAuditorEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setMediaDevicesEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitMediaDevicesEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setWebRTCMDNSICECandidatesEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitWebRTCMDNSICECandidatesEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setCustomUserAgent(JSStringRef userAgent)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetCustomUserAgent"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), toWK(userAgent).get(), nullptr);
}
void TestRunner::setWebAPIStatisticsEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitWebAPIStatisticsEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setModernMediaControlsEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitModernMediaControlsEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setWebGL2Enabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitWebGL2Enabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setWritableStreamAPIEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitWritableStreamAPIEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setReadableByteStreamAPIEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitReadableByteStreamAPIEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setEncryptedMediaAPIEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitEncryptedMediaAPIEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setPictureInPictureAPIEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitPictureInPictureAPIEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setGenericCueAPIEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitGenericCueAPIEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setAllowsAnySSLCertificate(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
injectedBundle.setAllowsAnySSLCertificate(enabled);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetAllowsAnySSLCertificate"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(enabled));
WKBundlePagePostSynchronousMessageForTesting(injectedBundle.page()->page(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setShouldSwapToEphemeralSessionOnNextNavigation(bool shouldSwap)
{
auto& injectedBundle = InjectedBundle::singleton();
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetShouldSwapToEphemeralSessionOnNextNavigation"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(shouldSwap));
WKBundlePagePostSynchronousMessageForTesting(injectedBundle.page()->page(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setShouldSwapToDefaultSessionOnNextNavigation(bool shouldSwap)
{
auto& injectedBundle = InjectedBundle::singleton();
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetShouldSwapToDefaultSessionOnNextNavigation"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(shouldSwap));
WKBundlePagePostSynchronousMessageForTesting(injectedBundle.page()->page(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setAllowUniversalAccessFromFileURLs(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetAllowUniversalAccessFromFileURLs(injectedBundle.bundle(), injectedBundle.pageGroup(), enabled);
}
void TestRunner::setAllowFileAccessFromFileURLs(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetAllowFileAccessFromFileURLs(injectedBundle.bundle(), injectedBundle.pageGroup(), enabled);
}
void TestRunner::setNeedsStorageAccessFromFileURLsQuirk(bool needsQuirk)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetAllowStorageAccessFromFileURLS(injectedBundle.bundle(), injectedBundle.pageGroup(), needsQuirk);
}
void TestRunner::setPluginsEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("WebKitPluginsEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
void TestRunner::setJavaScriptCanAccessClipboard(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetJavaScriptCanAccessClipboard(injectedBundle.bundle(), injectedBundle.pageGroup(), enabled);
}
void TestRunner::setPopupBlockingEnabled(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetPopupBlockingEnabled(injectedBundle.bundle(), injectedBundle.pageGroup(), enabled);
}
void TestRunner::setAuthorAndUserStylesEnabled(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetAuthorAndUserStylesEnabled(injectedBundle.bundle(), injectedBundle.pageGroup(), enabled);
}
void TestRunner::addOriginAccessWhitelistEntry(JSStringRef sourceOrigin, JSStringRef destinationProtocol, JSStringRef destinationHost, bool allowDestinationSubdomains)
{
WKBundleAddOriginAccessWhitelistEntry(InjectedBundle::singleton().bundle(), toWK(sourceOrigin).get(), toWK(destinationProtocol).get(), toWK(destinationHost).get(), allowDestinationSubdomains);
}
void TestRunner::removeOriginAccessWhitelistEntry(JSStringRef sourceOrigin, JSStringRef destinationProtocol, JSStringRef destinationHost, bool allowDestinationSubdomains)
{
WKBundleRemoveOriginAccessWhitelistEntry(InjectedBundle::singleton().bundle(), toWK(sourceOrigin).get(), toWK(destinationProtocol).get(), toWK(destinationHost).get(), allowDestinationSubdomains);
}
bool TestRunner::isPageBoxVisible(int pageIndex)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(injectedBundle.page()->page());
return WKBundleIsPageBoxVisible(injectedBundle.bundle(), mainFrame, pageIndex);
}
void TestRunner::setValueForUser(JSContextRef context, JSValueRef element, JSStringRef value)
{
if (!element || !JSValueIsObject(context, element))
return;
WKRetainPtr<WKBundleNodeHandleRef> nodeHandle = adoptWK(WKBundleNodeHandleCreate(context, const_cast<JSObjectRef>(element)));
WKBundleNodeHandleSetHTMLInputElementValueForUser(nodeHandle.get(), toWK(value).get());
}
void TestRunner::setAudioResult(JSContextRef context, JSValueRef data)
{
auto& injectedBundle = InjectedBundle::singleton();
// FIXME (123058): Use a JSC API to get buffer contents once such is exposed.
WKRetainPtr<WKDataRef> audioData = adoptWK(WKBundleCreateWKDataFromUInt8Array(injectedBundle.bundle(), context, data));
injectedBundle.setAudioResult(audioData.get());
setWhatToDump(WhatToDump::Audio);
setDumpPixels(false);
}
unsigned TestRunner::windowCount()
{
return InjectedBundle::singleton().pageCount();
}
void TestRunner::clearBackForwardList()
{
WKBundleClearHistoryForTesting(InjectedBundle::singleton().page()->page());
}
// Object Creation
void TestRunner::makeWindowObject(JSContextRef context, JSObjectRef windowObject, JSValueRef* exception)
{
setProperty(context, windowObject, "testRunner", this, kJSPropertyAttributeReadOnly | kJSPropertyAttributeDontDelete, exception);
}
void TestRunner::showWebInspector()
{
WKBundleInspectorShow(WKBundlePageGetInspector(InjectedBundle::singleton().page()->page()));
}
void TestRunner::closeWebInspector()
{
WKBundleInspectorClose(WKBundlePageGetInspector(InjectedBundle::singleton().page()->page()));
}
void TestRunner::evaluateInWebInspector(JSStringRef script)
{
WKRetainPtr<WKStringRef> scriptWK = toWK(script);
WKBundleInspectorEvaluateScriptForTest(WKBundlePageGetInspector(InjectedBundle::singleton().page()->page()), scriptWK.get());
}
typedef WTF::HashMap<unsigned, WKRetainPtr<WKBundleScriptWorldRef> > WorldMap;
static WorldMap& worldMap()
{
static WorldMap& map = *new WorldMap;
return map;
}
unsigned TestRunner::worldIDForWorld(WKBundleScriptWorldRef world)
{
WorldMap::const_iterator end = worldMap().end();
for (WorldMap::const_iterator it = worldMap().begin(); it != end; ++it) {
if (it->value == world)
return it->key;
}
return 0;
}
void TestRunner::evaluateScriptInIsolatedWorld(JSContextRef context, unsigned worldID, JSStringRef script)
{
// A worldID of 0 always corresponds to a new world. Any other worldID corresponds to a world
// that is created once and cached forever.
WKRetainPtr<WKBundleScriptWorldRef> world;
if (!worldID)
world.adopt(WKBundleScriptWorldCreateWorld());
else {
WKRetainPtr<WKBundleScriptWorldRef>& worldSlot = worldMap().add(worldID, nullptr).iterator->value;
if (!worldSlot)
worldSlot.adopt(WKBundleScriptWorldCreateWorld());
world = worldSlot;
}
WKBundleFrameRef frame = WKBundleFrameForJavaScriptContext(context);
if (!frame)
frame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSGlobalContextRef jsContext = WKBundleFrameGetJavaScriptContextForWorld(frame, world.get());
JSEvaluateScript(jsContext, script, 0, 0, 0, 0);
}
void TestRunner::setPOSIXLocale(JSStringRef locale)
{
char localeBuf[32];
JSStringGetUTF8CString(locale, localeBuf, sizeof(localeBuf));
setlocale(LC_ALL, localeBuf);
}
void TestRunner::setTextDirection(JSStringRef direction)
{
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
return WKBundleFrameSetTextDirection(mainFrame, toWK(direction).get());
}
void TestRunner::setShouldStayOnPageAfterHandlingBeforeUnload(bool shouldStayOnPage)
{
InjectedBundle::singleton().postNewBeforeUnloadReturnValue(!shouldStayOnPage);
}
bool TestRunner::didReceiveServerRedirectForProvisionalNavigation() const
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("DidReceiveServerRedirectForProvisionalNavigation"));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::clearDidReceiveServerRedirectForProvisionalNavigation()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("ClearDidReceiveServerRedirectForProvisionalNavigation"));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, nullptr);
}
void TestRunner::setPageVisibility(JSStringRef state)
{
InjectedBundle::singleton().setHidden(JSStringIsEqualToUTF8CString(state, "hidden") || JSStringIsEqualToUTF8CString(state, "prerender"));
}
void TestRunner::resetPageVisibility()
{
InjectedBundle::singleton().setHidden(false);
}
typedef WTF::HashMap<unsigned, JSValueRef> CallbackMap;
static CallbackMap& callbackMap()
{
static CallbackMap& map = *new CallbackMap;
return map;
}
enum {
AddChromeInputFieldCallbackID = 1,
RemoveChromeInputFieldCallbackID,
FocusWebViewCallbackID,
SetBackingScaleFactorCallbackID,
DidBeginSwipeCallbackID,
WillEndSwipeCallbackID,
DidEndSwipeCallbackID,
DidRemoveSwipeSnapshotCallbackID,
SetStatisticsDebugModeCallbackID,
SetStatisticsPrevalentResourceForDebugModeCallbackID,
SetStatisticsLastSeenCallbackID,
SetStatisticsMergeStatisticCallbackID,
SetStatisticsPrevalentResourceCallbackID,
SetStatisticsVeryPrevalentResourceCallbackID,
SetStatisticsHasHadUserInteractionCallbackID,
StatisticsDidModifyDataRecordsCallbackID,
StatisticsDidScanDataRecordsCallbackID,
StatisticsDidRunTelemetryCallbackID,
StatisticsDidClearThroughWebsiteDataRemovalCallbackID,
StatisticsDidResetToConsistentStateCallbackID,
StatisticsDidSetBlockCookiesForHostCallbackID,
StatisticsDidSetShouldDowngradeReferrerCallbackID,
StatisticsDidSetShouldBlockThirdPartyCookiesCallbackID,
StatisticsDidSetFirstPartyWebsiteDataRemovalModeCallbackID,
AllStorageAccessEntriesCallbackID,
DidRemoveAllSessionCredentialsCallbackID,
GetApplicationManifestCallbackID,
TextDidChangeInTextFieldCallbackID,
TextFieldDidBeginEditingCallbackID,
TextFieldDidEndEditingCallbackID,
CustomMenuActionCallbackID,
FirstUIScriptCallbackID = 100
};
static void cacheTestRunnerCallback(unsigned index, JSValueRef callback)
{
if (!callback)
return;
if (callbackMap().contains(index)) {
InjectedBundle::singleton().outputText(makeString("FAIL: Tried to install a second TestRunner callback for the same event (id ", index, ")\n\n"));
return;
}
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
JSValueProtect(context, callback);
callbackMap().add(index, callback);
}
static void callTestRunnerCallback(unsigned index, size_t argumentCount = 0, const JSValueRef arguments[] = nullptr)
{
if (!callbackMap().contains(index))
return;
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
JSObjectRef callback = JSValueToObject(context, callbackMap().take(index), 0);
JSObjectCallAsFunction(context, callback, JSContextGetGlobalObject(context), argumentCount, arguments, 0);
JSValueUnprotect(context, callback);
}
void TestRunner::clearTestRunnerCallbacks()
{
for (auto& iter : callbackMap()) {
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
JSObjectRef callback = JSValueToObject(context, iter.value, 0);
JSValueUnprotect(context, callback);
}
callbackMap().clear();
}
void TestRunner::accummulateLogsForChannel(JSStringRef)
{
// FIXME: Implement getting the call to all processes.
}
void TestRunner::addChromeInputField(JSValueRef callback)
{
cacheTestRunnerCallback(AddChromeInputFieldCallbackID, callback);
InjectedBundle::singleton().postAddChromeInputField();
}
void TestRunner::removeChromeInputField(JSValueRef callback)
{
cacheTestRunnerCallback(RemoveChromeInputFieldCallbackID, callback);
InjectedBundle::singleton().postRemoveChromeInputField();
}
void TestRunner::focusWebView(JSValueRef callback)
{
cacheTestRunnerCallback(FocusWebViewCallbackID, callback);
InjectedBundle::singleton().postFocusWebView();
}
void TestRunner::setBackingScaleFactor(double backingScaleFactor, JSValueRef callback)
{
cacheTestRunnerCallback(SetBackingScaleFactorCallbackID, callback);
InjectedBundle::singleton().postSetBackingScaleFactor(backingScaleFactor);
}
void TestRunner::setWindowIsKey(bool isKey)
{
InjectedBundle::singleton().postSetWindowIsKey(isKey);
}
void TestRunner::setViewSize(double width, double height)
{
InjectedBundle::singleton().postSetViewSize(width, height);
}
void TestRunner::callAddChromeInputFieldCallback()
{
callTestRunnerCallback(AddChromeInputFieldCallbackID);
}
void TestRunner::callRemoveChromeInputFieldCallback()
{
callTestRunnerCallback(RemoveChromeInputFieldCallbackID);
}
void TestRunner::callFocusWebViewCallback()
{
callTestRunnerCallback(FocusWebViewCallbackID);
}
void TestRunner::callSetBackingScaleFactorCallback()
{
callTestRunnerCallback(SetBackingScaleFactorCallbackID);
}
static inline bool toBool(JSStringRef value)
{
return JSStringIsEqualToUTF8CString(value, "true") || JSStringIsEqualToUTF8CString(value, "1");
}
void TestRunner::overridePreference(JSStringRef preference, JSStringRef value)
{
auto& injectedBundle = InjectedBundle::singleton();
// Should use `<!-- webkit-test-runner [ enableBackForwardCache=true ] -->` instead.
RELEASE_ASSERT(!JSStringIsEqualToUTF8CString(preference, "WebKitUsesBackForwardCachePreferenceKey"));
// FIXME: handle non-boolean preferences.
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), toWK(preference).get(), toBool(value));
}
void TestRunner::setAlwaysAcceptCookies(bool accept)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetAlwaysAcceptCookies"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(accept));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setOnlyAcceptFirstPartyCookies(bool accept)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetOnlyAcceptFirstPartyCookies"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(accept));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
double TestRunner::preciseTime()
{
return WallTime::now().secondsSinceEpoch().seconds();
}
void TestRunner::setRenderTreeDumpOptions(unsigned short options)
{
m_renderTreeDumpOptions = options;
}
void TestRunner::setUserStyleSheetEnabled(bool enabled)
{
m_userStyleSheetEnabled = enabled;
WKRetainPtr<WKStringRef> emptyUrl = adoptWK(WKStringCreateWithUTF8CString(""));
WKStringRef location = enabled ? m_userStyleSheetLocation.get() : emptyUrl.get();
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetUserStyleSheetLocation(injectedBundle.bundle(), injectedBundle.pageGroup(), location);
}
void TestRunner::setUserStyleSheetLocation(JSStringRef location)
{
m_userStyleSheetLocation = adoptWK(WKStringCreateWithJSString(location));
if (m_userStyleSheetEnabled)
setUserStyleSheetEnabled(true);
}
void TestRunner::setSpatialNavigationEnabled(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetSpatialNavigationEnabled(injectedBundle.bundle(), injectedBundle.pageGroup(), enabled);
}
void TestRunner::setTabKeyCyclesThroughElements(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetTabKeyCyclesThroughElements(injectedBundle.bundle(), injectedBundle.page()->page(), enabled);
}
void TestRunner::setSerializeHTTPLoads()
{
// WK2 doesn't reorder loads.
}
void TestRunner::dispatchPendingLoadRequests()
{
// WK2 doesn't keep pending requests.
}
void TestRunner::setCacheModel(int model)
{
InjectedBundle::singleton().setCacheModel(model);
}
void TestRunner::setAsynchronousSpellCheckingEnabled(bool enabled)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetAsynchronousSpellCheckingEnabled(injectedBundle.bundle(), injectedBundle.pageGroup(), enabled);
}
void TestRunner::grantWebNotificationPermission(JSStringRef origin)
{
WKRetainPtr<WKStringRef> originWK = toWK(origin);
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetWebNotificationPermission(injectedBundle.bundle(), injectedBundle.page()->page(), originWK.get(), true);
}
void TestRunner::denyWebNotificationPermission(JSStringRef origin)
{
WKRetainPtr<WKStringRef> originWK = toWK(origin);
auto& injectedBundle = InjectedBundle::singleton();
WKBundleSetWebNotificationPermission(injectedBundle.bundle(), injectedBundle.page()->page(), originWK.get(), false);
}
void TestRunner::removeAllWebNotificationPermissions()
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleRemoveAllWebNotificationPermissions(injectedBundle.bundle(), injectedBundle.page()->page());
}
void TestRunner::simulateWebNotificationClick(JSValueRef notification)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(injectedBundle.page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
uint64_t notificationID = WKBundleGetWebNotificationID(injectedBundle.bundle(), context, notification);
injectedBundle.postSimulateWebNotificationClick(notificationID);
}
void TestRunner::setGeolocationPermission(bool enabled)
{
// FIXME: this should be done by frame.
InjectedBundle::singleton().setGeolocationPermission(enabled);
}
bool TestRunner::isGeolocationProviderActive()
{
return InjectedBundle::singleton().isGeolocationProviderActive();
}
void TestRunner::setMockGeolocationPosition(double latitude, double longitude, double accuracy, JSValueRef jsAltitude, JSValueRef jsAltitudeAccuracy, JSValueRef jsHeading, JSValueRef jsSpeed, JSValueRef jsFloorLevel)
{
auto& injectedBundle = InjectedBundle::singleton();
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(injectedBundle.page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
bool providesAltitude = false;
double altitude = 0.;
if (!JSValueIsUndefined(context, jsAltitude)) {
providesAltitude = true;
altitude = JSValueToNumber(context, jsAltitude, 0);
}
bool providesAltitudeAccuracy = false;
double altitudeAccuracy = 0.;
if (!JSValueIsUndefined(context, jsAltitudeAccuracy)) {
providesAltitudeAccuracy = true;
altitudeAccuracy = JSValueToNumber(context, jsAltitudeAccuracy, 0);
}
bool providesHeading = false;
double heading = 0.;
if (!JSValueIsUndefined(context, jsHeading)) {
providesHeading = true;
heading = JSValueToNumber(context, jsHeading, 0);
}
bool providesSpeed = false;
double speed = 0.;
if (!JSValueIsUndefined(context, jsSpeed)) {
providesSpeed = true;
speed = JSValueToNumber(context, jsSpeed, 0);
}
bool providesFloorLevel = false;
double floorLevel = 0.;
if (!JSValueIsUndefined(context, jsFloorLevel)) {
providesFloorLevel = true;
floorLevel = JSValueToNumber(context, jsFloorLevel, 0);
}
injectedBundle.setMockGeolocationPosition(latitude, longitude, accuracy, providesAltitude, altitude, providesAltitudeAccuracy, altitudeAccuracy, providesHeading, heading, providesSpeed, speed, providesFloorLevel, floorLevel);
}
void TestRunner::setMockGeolocationPositionUnavailableError(JSStringRef message)
{
WKRetainPtr<WKStringRef> messageWK = toWK(message);
InjectedBundle::singleton().setMockGeolocationPositionUnavailableError(messageWK.get());
}
void TestRunner::setUserMediaPermission(bool enabled)
{
// FIXME: this should be done by frame.
InjectedBundle::singleton().setUserMediaPermission(enabled);
}
void TestRunner::resetUserMediaPermission()
{
// FIXME: this should be done by frame.
InjectedBundle::singleton().resetUserMediaPermission();
}
bool TestRunner::isDoingMediaCapture() const
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("IsDoingMediaCapture"));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::setUserMediaPersistentPermissionForOrigin(bool permission, JSStringRef origin, JSStringRef parentOrigin)
{
WKRetainPtr<WKStringRef> originWK = toWK(origin);
WKRetainPtr<WKStringRef> parentOriginWK = toWK(parentOrigin);
InjectedBundle::singleton().setUserMediaPersistentPermissionForOrigin(permission, originWK.get(), parentOriginWK.get());
}
unsigned TestRunner::userMediaPermissionRequestCountForOrigin(JSStringRef origin, JSStringRef parentOrigin) const
{
WKRetainPtr<WKStringRef> originWK = toWK(origin);
WKRetainPtr<WKStringRef> parentOriginWK = toWK(parentOrigin);
return InjectedBundle::singleton().userMediaPermissionRequestCountForOrigin(originWK.get(), parentOriginWK.get());
}
void TestRunner::resetUserMediaPermissionRequestCountForOrigin(JSStringRef origin, JSStringRef parentOrigin)
{
WKRetainPtr<WKStringRef> originWK = toWK(origin);
WKRetainPtr<WKStringRef> parentOriginWK = toWK(parentOrigin);
InjectedBundle::singleton().resetUserMediaPermissionRequestCountForOrigin(originWK.get(), parentOriginWK.get());
}
bool TestRunner::callShouldCloseOnWebView()
{
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
return WKBundleFrameCallShouldCloseOnWebView(mainFrame);
}
void TestRunner::queueBackNavigation(unsigned howFarBackward)
{
InjectedBundle::singleton().queueBackNavigation(howFarBackward);
}
void TestRunner::queueForwardNavigation(unsigned howFarForward)
{
InjectedBundle::singleton().queueForwardNavigation(howFarForward);
}
void TestRunner::queueLoad(JSStringRef url, JSStringRef target, bool shouldOpenExternalURLs)
{
auto& injectedBundle = InjectedBundle::singleton();
WKRetainPtr<WKURLRef> baseURLWK = adoptWK(WKBundleFrameCopyURL(WKBundlePageGetMainFrame(injectedBundle.page()->page())));
WKRetainPtr<WKURLRef> urlWK = adoptWK(WKURLCreateWithBaseURL(baseURLWK.get(), toWTFString(toWK(url)).utf8().data()));
WKRetainPtr<WKStringRef> urlStringWK = adoptWK(WKURLCopyString(urlWK.get()));
injectedBundle.queueLoad(urlStringWK.get(), toWK(target).get(), shouldOpenExternalURLs);
}
void TestRunner::queueLoadHTMLString(JSStringRef content, JSStringRef baseURL, JSStringRef unreachableURL)
{
WKRetainPtr<WKStringRef> contentWK = toWK(content);
WKRetainPtr<WKStringRef> baseURLWK = baseURL ? toWK(baseURL) : WKRetainPtr<WKStringRef>();
WKRetainPtr<WKStringRef> unreachableURLWK = unreachableURL ? toWK(unreachableURL) : WKRetainPtr<WKStringRef>();
InjectedBundle::singleton().queueLoadHTMLString(contentWK.get(), baseURLWK.get(), unreachableURLWK.get());
}
void TestRunner::queueReload()
{
InjectedBundle::singleton().queueReload();
}
void TestRunner::queueLoadingScript(JSStringRef script)
{
WKRetainPtr<WKStringRef> scriptWK = toWK(script);
InjectedBundle::singleton().queueLoadingScript(scriptWK.get());
}
void TestRunner::queueNonLoadingScript(JSStringRef script)
{
WKRetainPtr<WKStringRef> scriptWK = toWK(script);
InjectedBundle::singleton().queueNonLoadingScript(scriptWK.get());
}
void TestRunner::setRejectsProtectionSpaceAndContinueForAuthenticationChallenges(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetRejectsProtectionSpaceAndContinueForAuthenticationChallenges"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setHandlesAuthenticationChallenges(bool handlesAuthenticationChallenges)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetHandlesAuthenticationChallenges"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(handlesAuthenticationChallenges));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setShouldLogCanAuthenticateAgainstProtectionSpace(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetShouldLogCanAuthenticateAgainstProtectionSpace"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setShouldLogDownloadCallbacks(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetShouldLogDownloadCallbacks"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setAuthenticationUsername(JSStringRef username)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetAuthenticationUsername"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(username));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setAuthenticationPassword(JSStringRef password)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetAuthenticationPassword"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(password));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
bool TestRunner::secureEventInputIsEnabled() const
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SecureEventInputIsEnabled"));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::setBlockAllPlugins(bool shouldBlock)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetBlockAllPlugins"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(shouldBlock));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setPluginSupportedMode(JSStringRef mode)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetPluginSupportedMode"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(mode));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
JSValueRef TestRunner::failNextNewCodeBlock()
{
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
return JSC::failNextNewCodeBlock(context);
}
JSValueRef TestRunner::numberOfDFGCompiles(JSValueRef theFunction)
{
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
return JSC::numberOfDFGCompiles(context, theFunction);
}
JSValueRef TestRunner::neverInlineFunction(JSValueRef theFunction)
{
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
return JSC::setNeverInline(context, theFunction);
}
void TestRunner::setShouldDecideNavigationPolicyAfterDelay(bool value)
{
m_shouldDecideNavigationPolicyAfterDelay = value;
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetShouldDecideNavigationPolicyAfterDelay"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setShouldDecideResponsePolicyAfterDelay(bool value)
{
m_shouldDecideResponsePolicyAfterDelay = value;
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetShouldDecideResponsePolicyAfterDelay"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setNavigationGesturesEnabled(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetNavigationGesturesEnabled"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setIgnoresViewportScaleLimits(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetIgnoresViewportScaleLimits"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setShouldDownloadUndisplayableMIMETypes(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetShouldDownloadUndisplayableMIMETypes"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::setShouldAllowDeviceOrientationAndMotionAccess(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetShouldAllowDeviceOrientationAndMotionAccess"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::terminateNetworkProcess()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("TerminateNetworkProcess"));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, nullptr);
}
void TestRunner::terminateServiceWorkers()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("TerminateServiceWorkers"));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, nullptr);
}
void TestRunner::setUseSeparateServiceWorkerProcess(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetUseSeparateServiceWorkerProcess"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), nullptr);
}
static unsigned nextUIScriptCallbackID()
{
static unsigned callbackID = FirstUIScriptCallbackID;
return callbackID++;
}
void TestRunner::runUIScript(JSStringRef script, JSValueRef callback)
{
unsigned callbackID = nextUIScriptCallbackID();
cacheTestRunnerCallback(callbackID, callback);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("RunUIProcessScript"));
WKRetainPtr<WKMutableDictionaryRef> testDictionary = adoptWK(WKMutableDictionaryCreate());
WKRetainPtr<WKStringRef> scriptKey = adoptWK(WKStringCreateWithUTF8CString("Script"));
WKRetainPtr<WKStringRef> scriptValue = adoptWK(WKStringCreateWithJSString(script));
WKRetainPtr<WKStringRef> callbackIDKey = adoptWK(WKStringCreateWithUTF8CString("CallbackID"));
WKRetainPtr<WKUInt64Ref> callbackIDValue = adoptWK(WKUInt64Create(callbackID));
WKDictionarySetItem(testDictionary.get(), scriptKey.get(), scriptValue.get());
WKDictionarySetItem(testDictionary.get(), callbackIDKey.get(), callbackIDValue.get());
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), testDictionary.get());
}
void TestRunner::runUIScriptCallback(unsigned callbackID, JSStringRef result)
{
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
JSValueRef resultValue = JSValueMakeString(context, result);
callTestRunnerCallback(callbackID, 1, &resultValue);
}
void TestRunner::setAllowedMenuActions(JSValueRef actions)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetAllowedMenuActions"));
WKRetainPtr<WKMutableArrayRef> messageBody = adoptWK(WKMutableArrayCreate());
auto page = InjectedBundle::singleton().page()->page();
auto mainFrame = WKBundlePageGetMainFrame(page);
auto context = WKBundleFrameGetJavaScriptContext(mainFrame);
auto lengthPropertyName = adopt(JSStringCreateWithUTF8CString("length"));
auto actionsArray = JSValueToObject(context, actions, nullptr);
auto lengthValue = JSObjectGetProperty(context, actionsArray, lengthPropertyName.get(), nullptr);
if (!JSValueIsNumber(context, lengthValue))
return;
auto length = static_cast<size_t>(JSValueToNumber(context, lengthValue, 0));
for (size_t i = 0; i < length; ++i) {
auto value = JSObjectGetPropertyAtIndex(context, actionsArray, i, 0);
if (!JSValueIsString(context, value))
continue;
auto actionName = adopt(JSValueToStringCopy(context, value, 0));
WKRetainPtr<WKStringRef> action = adoptWK(WKStringCreateWithJSString(actionName.get()));
WKArrayAppendItem(messageBody.get(), action.get());
}
WKBundlePagePostMessage(page, messageName.get(), messageBody.get());
}
void TestRunner::installCustomMenuAction(JSStringRef name, bool dismissesAutomatically, JSValueRef callback)
{
cacheTestRunnerCallback(CustomMenuActionCallbackID, callback);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("InstallCustomMenuAction"));
WKRetainPtr<WKMutableDictionaryRef> messageBody = adoptWK(WKMutableDictionaryCreate());
WKRetainPtr<WKStringRef> nameKey = adoptWK(WKStringCreateWithUTF8CString("name"));
WKRetainPtr<WKStringRef> nameValue = adoptWK(WKStringCreateWithJSString(name));
WKDictionarySetItem(messageBody.get(), nameKey.get(), nameValue.get());
WKRetainPtr<WKStringRef> dismissesAutomaticallyKey = adoptWK(WKStringCreateWithUTF8CString("dismissesAutomatically"));
WKRetainPtr<WKBooleanRef> dismissesAutomaticallyValue = adoptWK(WKBooleanCreate(dismissesAutomatically));
WKDictionarySetItem(messageBody.get(), dismissesAutomaticallyKey.get(), dismissesAutomaticallyValue.get());
WKBundlePagePostMessage(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get());
}
void TestRunner::installDidBeginSwipeCallback(JSValueRef callback)
{
cacheTestRunnerCallback(DidBeginSwipeCallbackID, callback);
}
void TestRunner::installWillEndSwipeCallback(JSValueRef callback)
{
cacheTestRunnerCallback(WillEndSwipeCallbackID, callback);
}
void TestRunner::installDidEndSwipeCallback(JSValueRef callback)
{
cacheTestRunnerCallback(DidEndSwipeCallbackID, callback);
}
void TestRunner::installDidRemoveSwipeSnapshotCallback(JSValueRef callback)
{
cacheTestRunnerCallback(DidRemoveSwipeSnapshotCallbackID, callback);
}
void TestRunner::callDidBeginSwipeCallback()
{
callTestRunnerCallback(DidBeginSwipeCallbackID);
}
void TestRunner::callWillEndSwipeCallback()
{
callTestRunnerCallback(WillEndSwipeCallbackID);
}
void TestRunner::callDidEndSwipeCallback()
{
callTestRunnerCallback(DidEndSwipeCallbackID);
}
void TestRunner::callDidRemoveSwipeSnapshotCallback()
{
callTestRunnerCallback(DidRemoveSwipeSnapshotCallbackID);
}
void TestRunner::setStatisticsEnabled(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsEnabled"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsDebugMode(bool value, JSValueRef completionHandler)
{
cacheTestRunnerCallback(SetStatisticsDebugModeCallbackID, completionHandler);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsDebugMode"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsCallDidSetDebugModeCallback()
{
callTestRunnerCallback(SetStatisticsDebugModeCallbackID);
}
void TestRunner::setStatisticsPrevalentResourceForDebugMode(JSStringRef hostName, JSValueRef completionHandler)
{
cacheTestRunnerCallback(SetStatisticsPrevalentResourceForDebugModeCallbackID, completionHandler);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsPrevalentResourceForDebugMode"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(hostName));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsCallDidSetPrevalentResourceForDebugModeCallback()
{
callTestRunnerCallback(SetStatisticsPrevalentResourceForDebugModeCallbackID);
}
void TestRunner::setStatisticsLastSeen(JSStringRef hostName, double seconds, JSValueRef completionHandler)
{
cacheTestRunnerCallback(SetStatisticsLastSeenCallbackID, completionHandler);
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Value")));
values.append(adoptWK(WKDoubleCreate(seconds)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsLastSeen"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsCallDidSetLastSeenCallback()
{
callTestRunnerCallback(SetStatisticsLastSeenCallbackID);
}
void TestRunner::setStatisticsMergeStatistic(JSStringRef hostName, JSStringRef topFrameDomain1, JSStringRef topFrameDomain2, double lastSeen, bool hadUserInteraction, double mostRecentUserInteraction, bool isGrandfathered, bool isPrevalent, bool isVeryPrevalent, unsigned dataRecordsRemoved, JSValueRef completionHandler)
{
cacheTestRunnerCallback(SetStatisticsMergeStatisticCallbackID, completionHandler);
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("TopFrameDomain1")));
values.append(adoptWK(WKStringCreateWithJSString(topFrameDomain1)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("TopFrameDomain2")));
values.append(adoptWK(WKStringCreateWithJSString(topFrameDomain2)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("LastSeen")));
values.append(adoptWK(WKDoubleCreate(lastSeen)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("HadUserInteraction")));
values.append(adoptWK(WKBooleanCreate(hadUserInteraction)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("MostRecentUserInteraction")));
values.append(adoptWK(WKDoubleCreate(mostRecentUserInteraction)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("IsGrandfathered")));
values.append(adoptWK(WKBooleanCreate(isGrandfathered)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("IsPrevalent")));
values.append(adoptWK(WKBooleanCreate(isPrevalent)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("IsVeryPrevalent")));
values.append(adoptWK(WKBooleanCreate(isVeryPrevalent)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("DataRecordsRemoved")));
values.append(adoptWK(WKUInt64Create(dataRecordsRemoved)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsMergeStatistic"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsCallDidSetMergeStatisticCallback()
{
callTestRunnerCallback(SetStatisticsMergeStatisticCallbackID);
}
void TestRunner::setStatisticsPrevalentResource(JSStringRef hostName, bool value, JSValueRef completionHandler)
{
cacheTestRunnerCallback(SetStatisticsPrevalentResourceCallbackID, completionHandler);
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Value")));
values.append(adoptWK(WKBooleanCreate(value)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsPrevalentResource"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsCallDidSetPrevalentResourceCallback()
{
callTestRunnerCallback(SetStatisticsPrevalentResourceCallbackID);
}
void TestRunner::setStatisticsVeryPrevalentResource(JSStringRef hostName, bool value, JSValueRef completionHandler)
{
cacheTestRunnerCallback(SetStatisticsVeryPrevalentResourceCallbackID, completionHandler);
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Value")));
values.append(adoptWK(WKBooleanCreate(value)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsVeryPrevalentResource"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsCallDidSetVeryPrevalentResourceCallback()
{
callTestRunnerCallback(SetStatisticsVeryPrevalentResourceCallbackID);
}
void TestRunner::dumpResourceLoadStatistics()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("dumpResourceLoadStatistics"));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, nullptr);
}
bool TestRunner::isStatisticsPrevalentResource(JSStringRef hostName)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsPrevalentResource"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(hostName));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
bool TestRunner::isStatisticsVeryPrevalentResource(JSStringRef hostName)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsVeryPrevalentResource"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(hostName));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
bool TestRunner::isStatisticsRegisteredAsSubresourceUnder(JSStringRef subresourceHost, JSStringRef topFrameHost)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("SubresourceHost")));
values.append(adoptWK(WKStringCreateWithJSString(subresourceHost)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("TopFrameHost")));
values.append(adoptWK(WKStringCreateWithJSString(topFrameHost)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsRegisteredAsSubresourceUnder"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
bool TestRunner::isStatisticsRegisteredAsSubFrameUnder(JSStringRef subFrameHost, JSStringRef topFrameHost)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("SubFrameHost")));
values.append(adoptWK(WKStringCreateWithJSString(subFrameHost)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("TopFrameHost")));
values.append(adoptWK(WKStringCreateWithJSString(topFrameHost)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsRegisteredAsSubFrameUnder"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
bool TestRunner::isStatisticsRegisteredAsRedirectingTo(JSStringRef hostRedirectedFrom, JSStringRef hostRedirectedTo)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostRedirectedFrom")));
values.append(adoptWK(WKStringCreateWithJSString(hostRedirectedFrom)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostRedirectedTo")));
values.append(adoptWK(WKStringCreateWithJSString(hostRedirectedTo)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsRegisteredAsRedirectingTo"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::setStatisticsHasHadUserInteraction(JSStringRef hostName, bool value, JSValueRef completionHandler)
{
cacheTestRunnerCallback(SetStatisticsHasHadUserInteractionCallbackID, completionHandler);
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Value")));
values.append(adoptWK(WKBooleanCreate(value)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsHasHadUserInteraction"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsCallDidSetHasHadUserInteractionCallback()
{
callTestRunnerCallback(SetStatisticsHasHadUserInteractionCallbackID);
}
bool TestRunner::isStatisticsHasHadUserInteraction(JSStringRef hostName)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsHasHadUserInteraction"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(hostName));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
bool TestRunner::isStatisticsOnlyInDatabaseOnce(JSStringRef subHost, JSStringRef topHost)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("SubHost")));
values.append(adoptWK(WKStringCreateWithJSString(subHost)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("TopHost")));
values.append(adoptWK(WKStringCreateWithJSString(topHost)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
auto messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsOnlyInDatabaseOnce"));
auto messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::setStatisticsGrandfathered(JSStringRef hostName, bool value)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Value")));
values.append(adoptWK(WKBooleanCreate(value)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsGrandfathered"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setUseITPDatabase(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetUseITPDatabase"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
bool TestRunner::isStatisticsGrandfathered(JSStringRef hostName)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsGrandfathered"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(hostName));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::setStatisticsSubframeUnderTopFrameOrigin(JSStringRef hostName, JSStringRef topFrameHostName)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("TopFrameHostName")));
values.append(adoptWK(WKStringCreateWithJSString(topFrameHostName)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsSubframeUnderTopFrameOrigin"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsSubresourceUnderTopFrameOrigin(JSStringRef hostName, JSStringRef topFrameHostName)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("TopFrameHostName")));
values.append(adoptWK(WKStringCreateWithJSString(topFrameHostName)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsSubresourceUnderTopFrameOrigin"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsSubresourceUniqueRedirectTo(JSStringRef hostName, JSStringRef hostNameRedirectedTo)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostNameRedirectedTo")));
values.append(adoptWK(WKStringCreateWithJSString(hostNameRedirectedTo)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsSubresourceUniqueRedirectTo"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsSubresourceUniqueRedirectFrom(JSStringRef hostName, JSStringRef hostNameRedirectedFrom)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostNameRedirectedFrom")));
values.append(adoptWK(WKStringCreateWithJSString(hostNameRedirectedFrom)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsSubresourceUniqueRedirectFrom"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsTopFrameUniqueRedirectTo(JSStringRef hostName, JSStringRef hostNameRedirectedTo)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostNameRedirectedTo")));
values.append(adoptWK(WKStringCreateWithJSString(hostNameRedirectedTo)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsTopFrameUniqueRedirectTo"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsTopFrameUniqueRedirectFrom(JSStringRef hostName, JSStringRef hostNameRedirectedFrom)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostNameRedirectedFrom")));
values.append(adoptWK(WKStringCreateWithJSString(hostNameRedirectedFrom)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsTopFrameUniqueRedirectFrom"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsCrossSiteLoadWithLinkDecoration(JSStringRef fromHost, JSStringRef toHost)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("FromHost")));
values.append(adoptWK(WKStringCreateWithJSString(fromHost)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("ToHost")));
values.append(adoptWK(WKStringCreateWithJSString(toHost)));
Vector<WKStringRef> rawKeys(keys.size());
Vector<WKTypeRef> rawValues(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsCrossSiteLoadWithLinkDecoration"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsTimeToLiveUserInteraction(double seconds)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsTimeToLiveUserInteraction"));
WKRetainPtr<WKDoubleRef> messageBody = adoptWK(WKDoubleCreate(seconds));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::installStatisticsDidModifyDataRecordsCallback(JSValueRef callback)
{
cacheTestRunnerCallback(StatisticsDidModifyDataRecordsCallbackID, callback);
}
void TestRunner::statisticsDidModifyDataRecordsCallback()
{
callTestRunnerCallback(StatisticsDidModifyDataRecordsCallbackID);
}
void TestRunner::installStatisticsDidScanDataRecordsCallback(JSValueRef callback)
{
cacheTestRunnerCallback(StatisticsDidScanDataRecordsCallbackID, callback);
bool notifyPagesWhenDataRecordsWereScanned = !!callback;
// Setting a callback implies we expect to receive callbacks. So register for them.
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsNotifyPagesWhenDataRecordsWereScanned"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(notifyPagesWhenDataRecordsWereScanned));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsDidScanDataRecordsCallback()
{
callTestRunnerCallback(StatisticsDidScanDataRecordsCallbackID);
}
void TestRunner::installStatisticsDidRunTelemetryCallback(JSValueRef callback)
{
cacheTestRunnerCallback(StatisticsDidRunTelemetryCallbackID, callback);
}
void TestRunner::statisticsDidRunTelemetryCallback(unsigned numberOfPrevalentResources, unsigned numberOfPrevalentResourcesWithUserInteraction, unsigned numberOfPrevalentResourcesWithoutUserInteraction, unsigned topPrevalentResourceWithUserInteractionDaysSinceUserInteraction, unsigned medianDaysSinceUserInteractionPrevalentResourceWithUserInteraction, unsigned top3NumberOfPrevalentResourcesWithUI, unsigned top3MedianSubFrameWithoutUI, unsigned top3MedianSubResourceWithoutUI, unsigned top3MedianUniqueRedirectsWithoutUI, unsigned top3MedianDataRecordsRemovedWithoutUI)
{
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
String string = makeString("{ \"numberOfPrevalentResources\" : ", numberOfPrevalentResources, ", \"numberOfPrevalentResourcesWithUserInteraction\" : ", numberOfPrevalentResourcesWithUserInteraction, ", \"numberOfPrevalentResourcesWithoutUserInteraction\" : ", numberOfPrevalentResourcesWithoutUserInteraction, ", \"topPrevalentResourceWithUserInteractionDaysSinceUserInteraction\" : ", topPrevalentResourceWithUserInteractionDaysSinceUserInteraction, ", \"medianDaysSinceUserInteractionPrevalentResourceWithUserInteraction\" : ", medianDaysSinceUserInteractionPrevalentResourceWithUserInteraction, ", \"top3NumberOfPrevalentResourcesWithUI\" : ", top3NumberOfPrevalentResourcesWithUI, ", \"top3MedianSubFrameWithoutUI\" : ", top3MedianSubFrameWithoutUI, ", \"top3MedianSubResourceWithoutUI\" : ", top3MedianSubResourceWithoutUI, ", \"top3MedianUniqueRedirectsWithoutUI\" : ", top3MedianUniqueRedirectsWithoutUI, ", \"top3MedianDataRecordsRemovedWithoutUI\" : ", top3MedianDataRecordsRemovedWithoutUI, " }");
JSValueRef result = JSValueMakeFromJSONString(context, adopt(JSStringCreateWithUTF8CString(string.utf8().data())).get());
callTestRunnerCallback(StatisticsDidRunTelemetryCallbackID, 1, &result);
}
bool TestRunner::statisticsNotifyObserver()
{
return InjectedBundle::singleton().statisticsNotifyObserver();
}
void TestRunner::statisticsProcessStatisticsAndDataRecords()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsProcessStatisticsAndDataRecords"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::statisticsUpdateCookieBlocking(JSValueRef completionHandler)
{
cacheTestRunnerCallback(StatisticsDidSetBlockCookiesForHostCallbackID, completionHandler);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsUpdateCookieBlocking"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::statisticsCallDidSetBlockCookiesForHostCallback()
{
callTestRunnerCallback(StatisticsDidSetBlockCookiesForHostCallbackID);
}
void TestRunner::statisticsSubmitTelemetry()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsSubmitTelemetry"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::setStatisticsNotifyPagesWhenDataRecordsWereScanned(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsNotifyPagesWhenDataRecordsWereScanned"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsIsRunningTest(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsSetIsRunningTest"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsShouldClassifyResourcesBeforeDataRecordsRemoval(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsShouldClassifyResourcesBeforeDataRecordsRemoval"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsNotifyPagesWhenTelemetryWasCaptured(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsNotifyPagesWhenTelemetryWasCaptured"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsMinimumTimeBetweenDataRecordsRemoval(double seconds)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsMinimumTimeBetweenDataRecordsRemoval"));
WKRetainPtr<WKDoubleRef> messageBody = adoptWK(WKDoubleCreate(seconds));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsGrandfatheringTime(double seconds)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsGrandfatheringTime"));
WKRetainPtr<WKDoubleRef> messageBody = adoptWK(WKDoubleCreate(seconds));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsMaxStatisticsEntries(unsigned entries)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetMaxStatisticsEntries"));
WKRetainPtr<WKTypeRef> messageBody = adoptWK(WKUInt64Create(entries));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setStatisticsPruneEntriesDownTo(unsigned entries)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetPruneEntriesDownTo"));
WKRetainPtr<WKTypeRef> messageBody = adoptWK(WKUInt64Create(entries));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsClearInMemoryAndPersistentStore(JSValueRef callback)
{
cacheTestRunnerCallback(StatisticsDidClearThroughWebsiteDataRemovalCallbackID, callback);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsClearInMemoryAndPersistentStore"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::statisticsClearInMemoryAndPersistentStoreModifiedSinceHours(unsigned hours, JSValueRef callback)
{
cacheTestRunnerCallback(StatisticsDidClearThroughWebsiteDataRemovalCallbackID, callback);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsClearInMemoryAndPersistentStoreModifiedSinceHours"));
WKRetainPtr<WKTypeRef> messageBody = adoptWK(WKUInt64Create(hours));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::statisticsClearThroughWebsiteDataRemoval(JSValueRef callback)
{
cacheTestRunnerCallback(StatisticsDidClearThroughWebsiteDataRemovalCallbackID, callback);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsClearThroughWebsiteDataRemoval"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::statisticsDeleteCookiesForHost(JSStringRef hostName, bool includeHttpOnlyCookies)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("HostName")));
values.append(adoptWK(WKStringCreateWithJSString(hostName)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("IncludeHttpOnlyCookies")));
values.append(adoptWK(WKBooleanCreate(includeHttpOnlyCookies)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsDeleteCookiesForHost"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
bool TestRunner::isStatisticsHasLocalStorage(JSStringRef hostName)
{
auto messageName = adoptWK(WKStringCreateWithUTF8CString("IsStatisticsHasLocalStorage"));
auto messageBody = adoptWK(WKStringCreateWithJSString(hostName));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::setStatisticsCacheMaxAgeCap(double seconds)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsCacheMaxAgeCap"));
WKRetainPtr<WKDoubleRef> messageBody = adoptWK(WKDoubleCreate(seconds));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
bool TestRunner::hasStatisticsIsolatedSession(JSStringRef hostName)
{
auto messageName = adoptWK(WKStringCreateWithUTF8CString("HasStatisticsIsolatedSession"));
auto messageBody = adoptWK(WKStringCreateWithJSString(hostName));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
void TestRunner::setStatisticsShouldDowngradeReferrer(bool value, JSValueRef completionHandler)
{
if (m_hasSetDowngradeReferrerCallback)
return;
cacheTestRunnerCallback(StatisticsDidSetShouldDowngradeReferrerCallbackID, completionHandler);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsShouldDowngradeReferrer"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
m_hasSetDowngradeReferrerCallback = true;
}
void TestRunner::statisticsCallDidSetShouldDowngradeReferrerCallback()
{
callTestRunnerCallback(StatisticsDidSetShouldDowngradeReferrerCallbackID);
m_hasSetDowngradeReferrerCallback = false;
}
void TestRunner::setStatisticsShouldBlockThirdPartyCookies(bool value, JSValueRef completionHandler, bool onlyOnSitesWithoutUserInteraction)
{
if (m_hasSetBlockThirdPartyCookiesCallback)
return;
cacheTestRunnerCallback(StatisticsDidSetShouldBlockThirdPartyCookiesCallbackID, completionHandler);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsShouldBlockThirdPartyCookies"));
if (onlyOnSitesWithoutUserInteraction)
messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsShouldBlockThirdPartyCookiesOnSitesWithoutUserInteraction"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get());
m_hasSetBlockThirdPartyCookiesCallback = true;
}
void TestRunner::statisticsCallDidSetShouldBlockThirdPartyCookiesCallback()
{
callTestRunnerCallback(StatisticsDidSetShouldBlockThirdPartyCookiesCallbackID);
m_hasSetBlockThirdPartyCookiesCallback = false;
}
void TestRunner::setStatisticsFirstPartyWebsiteDataRemovalMode(bool value, JSValueRef completionHandler)
{
if (m_hasSetFirstPartyWebsiteDataRemovalModeCallback)
return;
cacheTestRunnerCallback(StatisticsDidSetFirstPartyWebsiteDataRemovalModeCallbackID, completionHandler);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStatisticsFirstPartyWebsiteDataRemovalMode"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
m_hasSetFirstPartyWebsiteDataRemovalModeCallback = true;
}
void TestRunner::statisticsCallDidSetFirstPartyWebsiteDataRemovalModeCallback()
{
callTestRunnerCallback(StatisticsDidSetFirstPartyWebsiteDataRemovalModeCallbackID);
m_hasSetFirstPartyWebsiteDataRemovalModeCallback = false;
}
void TestRunner::statisticsCallClearThroughWebsiteDataRemovalCallback()
{
callTestRunnerCallback(StatisticsDidClearThroughWebsiteDataRemovalCallbackID);
}
void TestRunner::statisticsResetToConsistentState(JSValueRef completionHandler)
{
cacheTestRunnerCallback(StatisticsDidResetToConsistentStateCallbackID, completionHandler);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("StatisticsResetToConsistentState"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::statisticsCallDidResetToConsistentStateCallback()
{
callTestRunnerCallback(StatisticsDidResetToConsistentStateCallbackID);
}
void TestRunner::installTextDidChangeInTextFieldCallback(JSValueRef callback)
{
cacheTestRunnerCallback(TextDidChangeInTextFieldCallbackID, callback);
}
void TestRunner::textDidChangeInTextFieldCallback()
{
callTestRunnerCallback(TextDidChangeInTextFieldCallbackID);
}
void TestRunner::installTextFieldDidBeginEditingCallback(JSValueRef callback)
{
cacheTestRunnerCallback(TextFieldDidBeginEditingCallbackID, callback);
}
void TestRunner::textFieldDidBeginEditingCallback()
{
callTestRunnerCallback(TextFieldDidBeginEditingCallbackID);
}
void TestRunner::installTextFieldDidEndEditingCallback(JSValueRef callback)
{
cacheTestRunnerCallback(TextFieldDidEndEditingCallbackID, callback);
}
void TestRunner::textFieldDidEndEditingCallback()
{
callTestRunnerCallback(TextFieldDidEndEditingCallbackID);
}
void TestRunner::setStorageAccessAPIEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetStorageAccessAPIEnabled"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(enabled));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::getAllStorageAccessEntries(JSValueRef callback)
{
cacheTestRunnerCallback(AllStorageAccessEntriesCallbackID, callback);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("GetAllStorageAccessEntries"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::callDidReceiveAllStorageAccessEntriesCallback(Vector<String>& domains)
{
WKBundleFrameRef mainFrame = WKBundlePageGetMainFrame(InjectedBundle::singleton().page()->page());
JSContextRef context = WKBundleFrameGetJavaScriptContext(mainFrame);
StringBuilder stringBuilder;
stringBuilder.appendLiteral("[");
bool firstDomain = true;
for (auto& domain : domains) {
if (firstDomain)
firstDomain = false;
else
stringBuilder.appendLiteral(", ");
stringBuilder.appendLiteral("\"");
stringBuilder.append(domain);
stringBuilder.appendLiteral("\"");
}
stringBuilder.appendLiteral("]");
JSValueRef result = JSValueMakeFromJSONString(context, adopt(JSStringCreateWithUTF8CString(stringBuilder.toString().utf8().data())).get());
callTestRunnerCallback(AllStorageAccessEntriesCallbackID, 1, &result);
}
void TestRunner::addMockMediaDevice(JSStringRef persistentId, JSStringRef label, const char* type)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("PersistentID")));
values.append(toWK(persistentId));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Label")));
values.append(toWK(label));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Type")));
values.append(adoptWK(WKStringCreateWithUTF8CString(type)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("AddMockMediaDevice"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::addMockCameraDevice(JSStringRef persistentId, JSStringRef label)
{
addMockMediaDevice(persistentId, label, "camera");
}
void TestRunner::addMockMicrophoneDevice(JSStringRef persistentId, JSStringRef label)
{
addMockMediaDevice(persistentId, label, "microphone");
}
void TestRunner::addMockScreenDevice(JSStringRef persistentId, JSStringRef label)
{
addMockMediaDevice(persistentId, label, "screen");
}
void TestRunner::clearMockMediaDevices()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("ClearMockMediaDevices"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::removeMockMediaDevice(JSStringRef persistentId)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("RemoveMockMediaDevice"));
WKRetainPtr<WKTypeRef> messageBody(toWK(persistentId));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::resetMockMediaDevices()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("ResetMockMediaDevices"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::setMockCameraOrientation(unsigned orientation)
{
auto messageName = adoptWK(WKStringCreateWithUTF8CString("SetMockCameraOrientation"));
auto messageBody = adoptWK(WKUInt64Create(orientation));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
bool TestRunner::isMockRealtimeMediaSourceCenterEnabled()
{
auto messageName = adoptWK(WKStringCreateWithUTF8CString("IsMockRealtimeMediaSourceCenterEnabled"));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
#if PLATFORM(MAC)
void TestRunner::connectMockGamepad(unsigned index)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("ConnectMockGamepad"));
WKRetainPtr<WKTypeRef> messageBody = adoptWK(WKUInt64Create(index));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::disconnectMockGamepad(unsigned index)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("DisconnectMockGamepad"));
WKRetainPtr<WKTypeRef> messageBody = adoptWK(WKUInt64Create(index));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setMockGamepadDetails(unsigned index, JSStringRef gamepadID, unsigned axisCount, unsigned buttonCount)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("GamepadID")));
values.append(toWK(gamepadID));
keys.append(adoptWK(WKStringCreateWithUTF8CString("GamepadIndex")));
values.append(adoptWK(WKUInt64Create(index)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("AxisCount")));
values.append(adoptWK(WKUInt64Create(axisCount)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("ButtonCount")));
values.append(adoptWK(WKUInt64Create(buttonCount)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetMockGamepadDetails"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setMockGamepadAxisValue(unsigned index, unsigned axisIndex, double value)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("GamepadIndex")));
values.append(adoptWK(WKUInt64Create(index)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("AxisIndex")));
values.append(adoptWK(WKUInt64Create(axisIndex)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Value")));
values.append(adoptWK(WKDoubleCreate(value)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetMockGamepadAxisValue"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setMockGamepadButtonValue(unsigned index, unsigned buttonIndex, double value)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("GamepadIndex")));
values.append(adoptWK(WKUInt64Create(index)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("ButtonIndex")));
values.append(adoptWK(WKUInt64Create(buttonIndex)));
keys.append(adoptWK(WKStringCreateWithUTF8CString("Value")));
values.append(adoptWK(WKDoubleCreate(value)));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetMockGamepadButtonValue"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
#else
void TestRunner::connectMockGamepad(unsigned)
{
}
void TestRunner::disconnectMockGamepad(unsigned)
{
}
void TestRunner::setMockGamepadDetails(unsigned, JSStringRef, unsigned, unsigned)
{
}
void TestRunner::setMockGamepadAxisValue(unsigned, unsigned, double)
{
}
void TestRunner::setMockGamepadButtonValue(unsigned, unsigned, double)
{
}
#endif // PLATFORM(MAC)
void TestRunner::setOpenPanelFiles(JSValueRef filesValue)
{
WKBundlePageRef page = InjectedBundle::singleton().page()->page();
JSContextRef context = WKBundleFrameGetJavaScriptContext(WKBundlePageGetMainFrame(page));
if (!JSValueIsArray(context, filesValue))
return;
JSObjectRef files = JSValueToObject(context, filesValue, nullptr);
static auto lengthProperty = adopt(JSStringCreateWithUTF8CString("length"));
JSValueRef filesLengthValue = JSObjectGetProperty(context, files, lengthProperty.get(), nullptr);
if (!JSValueIsNumber(context, filesLengthValue))
return;
auto fileURLs = adoptWK(WKMutableArrayCreate());
auto filesLength = static_cast<size_t>(JSValueToNumber(context, filesLengthValue, nullptr));
for (size_t i = 0; i < filesLength; ++i) {
JSValueRef fileValue = JSObjectGetPropertyAtIndex(context, files, i, nullptr);
if (!JSValueIsString(context, fileValue))
continue;
auto file = adopt(JSValueToStringCopy(context, fileValue, nullptr));
size_t fileBufferSize = JSStringGetMaximumUTF8CStringSize(file.get()) + 1;
auto fileBuffer = makeUniqueArray<char>(fileBufferSize);
JSStringGetUTF8CString(file.get(), fileBuffer.get(), fileBufferSize);
WKArrayAppendItem(fileURLs.get(), adoptWK(WKURLCreateWithBaseURL(m_testURL.get(), fileBuffer.get())).get());
}
static auto messageName = adoptWK(WKStringCreateWithUTF8CString("SetOpenPanelFileURLs"));
WKBundlePagePostMessage(page, messageName.get(), fileURLs.get());
}
#if PLATFORM(IOS_FAMILY)
void TestRunner::setOpenPanelFilesMediaIcon(JSValueRef data)
{
WKBundlePageRef page = InjectedBundle::singleton().page()->page();
JSContextRef context = WKBundleFrameGetJavaScriptContext(WKBundlePageGetMainFrame(page));
auto& injectedBundle = InjectedBundle::singleton();
// FIXME (123058): Use a JSC API to get buffer contents once such is exposed.
WKRetainPtr<WKDataRef> iconData = adoptWK(WKBundleCreateWKDataFromUInt8Array(injectedBundle.bundle(), context, data));
static auto messageName = adoptWK(WKStringCreateWithUTF8CString("SetOpenPanelFileURLsMediaIcon"));
WKBundlePagePostMessage(page, messageName.get(), iconData.get());
}
#else
void TestRunner::setOpenPanelFilesMediaIcon(JSValueRef)
{
}
#endif // PLATFORM(IOS_FAMILY)
void TestRunner::removeAllSessionCredentials(JSValueRef callback)
{
cacheTestRunnerCallback(DidRemoveAllSessionCredentialsCallbackID, callback);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("RemoveAllSessionCredentials"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(true));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::callDidRemoveAllSessionCredentialsCallback()
{
callTestRunnerCallback(DidRemoveAllSessionCredentialsCallbackID);
}
void TestRunner::clearDOMCache(JSStringRef origin)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("ClearDOMCache"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(origin));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::clearDOMCaches()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("ClearDOMCaches"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
bool TestRunner::hasDOMCache(JSStringRef origin)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("HasDOMCache"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(origin));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
uint64_t TestRunner::domCacheSize(JSStringRef origin)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("DOMCacheSize"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(origin));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), &returnData);
return WKUInt64GetValue(static_cast<WKUInt64Ref>(returnData));
}
void TestRunner::setAllowStorageQuotaIncrease(bool willIncrease)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetAllowStorageQuotaIncrease"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(willIncrease));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::getApplicationManifestThen(JSValueRef callback)
{
cacheTestRunnerCallback(GetApplicationManifestCallbackID, callback);
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("GetApplicationManifest"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::didGetApplicationManifest()
{
callTestRunnerCallback(GetApplicationManifestCallbackID);
}
void TestRunner::performCustomMenuAction()
{
callTestRunnerCallback(CustomMenuActionCallbackID);
}
size_t TestRunner::userScriptInjectedCount() const
{
return InjectedBundle::singleton().userScriptInjectedCount();
}
void TestRunner::injectUserScript(JSStringRef script)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("InjectUserScript"));
WKRetainPtr<WKStringRef> messageBody = adoptWK(WKStringCreateWithJSString(script));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::sendDisplayConfigurationChangedMessageForTesting()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SendDisplayConfigurationChangedMessageForTesting"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::setServiceWorkerFetchTimeout(double seconds)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetServiceWorkerFetchTimeout"));
WKRetainPtr<WKDoubleRef> messageBody = adoptWK(WKDoubleCreate(seconds));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
// WebAuthn
void TestRunner::addTestKeyToKeychain(JSStringRef privateKeyBase64, JSStringRef attrLabel, JSStringRef applicationTagBase64)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("PrivateKey")));
values.append(toWK(privateKeyBase64));
keys.append(adoptWK(WKStringCreateWithUTF8CString("AttrLabel")));
values.append(toWK(attrLabel));
keys.append(adoptWK(WKStringCreateWithUTF8CString("ApplicationTag")));
values.append(toWK(applicationTagBase64));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("AddTestKeyToKeychain"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::cleanUpKeychain(JSStringRef attrLabel, JSStringRef applicationTagBase64)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("AttrLabel")));
values.append(toWK(attrLabel));
if (applicationTagBase64) {
keys.append(adoptWK(WKStringCreateWithUTF8CString("ApplicationTag")));
values.append(toWK(applicationTagBase64));
}
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("CleanUpKeychain"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
bool TestRunner::keyExistsInKeychain(JSStringRef attrLabel, JSStringRef applicationTagBase64)
{
Vector<WKRetainPtr<WKStringRef>> keys;
Vector<WKRetainPtr<WKTypeRef>> values;
keys.append(adoptWK(WKStringCreateWithUTF8CString("AttrLabel")));
values.append(toWK(attrLabel));
keys.append(adoptWK(WKStringCreateWithUTF8CString("ApplicationTag")));
values.append(toWK(applicationTagBase64));
Vector<WKStringRef> rawKeys;
Vector<WKTypeRef> rawValues;
rawKeys.resize(keys.size());
rawValues.resize(values.size());
for (size_t i = 0; i < keys.size(); ++i) {
rawKeys[i] = keys[i].get();
rawValues[i] = values[i].get();
}
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("KeyExistsInKeychain"));
WKRetainPtr<WKDictionaryRef> messageBody = adoptWK(WKDictionaryCreate(rawKeys.data(), rawValues.data(), rawKeys.size()));
WKTypeRef returnData = nullptr;
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), &returnData);
ASSERT(WKGetTypeID(returnData) == WKBooleanGetTypeID());
return WKBooleanGetValue(adoptWK(static_cast<WKBooleanRef>(returnData)).get());
}
unsigned long TestRunner::serverTrustEvaluationCallbackCallsCount()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("ServerTrustEvaluationCallbackCallsCount"));
WKTypeRef returnData = nullptr;
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, &returnData);
ASSERT(WKGetTypeID(returnData) == WKUInt64GetTypeID());
return WKUInt64GetValue(adoptWK(static_cast<WKUInt64Ref>(returnData)).get());
}
void TestRunner::setShouldDismissJavaScriptAlertsAsynchronously(bool shouldDismissAsynchronously)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("ShouldDismissJavaScriptAlertsAsynchronously"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(shouldDismissAsynchronously));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::abortModal()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("AbortModal"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::dumpAdClickAttribution()
{
auto messageName = adoptWK(WKStringCreateWithUTF8CString("DumpAdClickAttribution"));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, nullptr);
}
void TestRunner::clearAdClickAttribution()
{
auto messageName = adoptWK(WKStringCreateWithUTF8CString("ClearAdClickAttribution"));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, nullptr);
}
void TestRunner::clearAdClickAttributionsThroughWebsiteDataRemoval()
{
auto messageName = adoptWK(WKStringCreateWithUTF8CString("ClearAdClickAttributionsThroughWebsiteDataRemoval"));
WKBundlePostSynchronousMessage(InjectedBundle::singleton().bundle(), messageName.get(), nullptr, nullptr);
}
void TestRunner::setAdClickAttributionOverrideTimerForTesting(bool value)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetAdClickAttributionOverrideTimerForTesting"));
WKRetainPtr<WKBooleanRef> messageBody = adoptWK(WKBooleanCreate(value));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::setAdClickAttributionConversionURLForTesting(JSStringRef urlString)
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("SetAdClickAttributionConversionURLForTesting"));
auto wtfURLString = toWTFString(WKStringCreateWithJSString(urlString));
WKRetainPtr<WKURLRef> messageBody = adoptWK(WKURLCreateWithUTF8CString(wtfURLString.utf8().data()));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), messageBody.get(), nullptr);
}
void TestRunner::markAdClickAttributionsAsExpiredForTesting()
{
WKRetainPtr<WKStringRef> messageName = adoptWK(WKStringCreateWithUTF8CString("MarkAdClickAttributionsAsExpiredForTesting"));
WKBundlePagePostSynchronousMessageForTesting(InjectedBundle::singleton().page()->page(), messageName.get(), nullptr, nullptr);
}
void TestRunner::setOffscreenCanvasEnabled(bool enabled)
{
WKRetainPtr<WKStringRef> key = adoptWK(WKStringCreateWithUTF8CString("OffscreenCanvasEnabled"));
auto& injectedBundle = InjectedBundle::singleton();
WKBundleOverrideBoolPreferenceForTestRunner(injectedBundle.bundle(), injectedBundle.pageGroup(), key.get(), enabled);
}
} // namespace WTR