Tools/TestWebKitAPI/Tests/WebKitObjC/UserContentTest.mm - WebKit - Git at Google

 /*
  * Copyright (C) 2012 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.
  */

 #import "config.h"

 #if WK_HAVE_C_SPI

 #import "Test.h"

 #import "PlatformUtilities.h"
 #import "TestBrowsingContextLoadDelegate.h"
 #import <JavaScriptCore/JSRetainPtr.h>
 #import <JavaScriptCore/JavaScriptCore.h>
 #import <WebKit/WKSerializedScriptValue.h>
 #import <WebKit/WKViewPrivate.h>

 #if WK_API_ENABLED && PLATFORM(MAC)

 static bool testFinished = false;
 static NSString *htmlString = @"<body style='background-color: red'>";
 static NSString *userStyleSheet = @"body { background-color: green !important; }";
 static const char* backgroundColorScript = "window.getComputedStyle(document.body, null).getPropertyValue('background-color')";
 static const char* greenInRGB = "rgb(0, 128, 0)";
 static const char* redInRGB = "rgb(255, 0, 0)";
 static const char* userScriptTestProperty = "window._userScriptInstalled";

 namespace {
     class WebKit2UserContentTest : public ::testing::Test {
     public:
         WKProcessGroup *processGroup;
         WKBrowsingContextGroup *browsingContextGroup;

         WebKit2UserContentTest()
             : processGroup(nil)
             , browsingContextGroup(nil)
         {
         }

         virtual void SetUp()
         {
             processGroup = [[WKProcessGroup alloc] init];
             browsingContextGroup = [[WKBrowsingContextGroup alloc] initWithIdentifier:@"UserContentIdentifier"];
         }

         virtual void TearDown()
         {
             [browsingContextGroup release];
             [processGroup release];
         }
     };
 } // namespace

 static void expectScriptValueIsString(WKSerializedScriptValueRef serializedScriptValue, const char* expectedValue)
 {
     JSGlobalContextRef scriptContext = JSGlobalContextCreate(0);

     JSValueRef scriptValue = WKSerializedScriptValueDeserialize(serializedScriptValue, scriptContext, 0);
     EXPECT_TRUE(JSValueIsString(scriptContext, scriptValue));

     JSRetainPtr<JSStringRef> scriptString(Adopt, JSValueToStringCopy(scriptContext, scriptValue, 0));
     EXPECT_TRUE(JSStringIsEqualToUTF8CString(scriptString.get(), expectedValue));

     JSGlobalContextRelease(scriptContext);
 }

 static void expectScriptValueIsBoolean(WKSerializedScriptValueRef serializedScriptValue, bool expectedValue)
 {
     JSGlobalContextRef scriptContext = JSGlobalContextCreate(0);

     JSValueRef scriptValue = WKSerializedScriptValueDeserialize(serializedScriptValue, scriptContext, 0);
     EXPECT_TRUE(JSValueIsBoolean(scriptContext, scriptValue));
     EXPECT_EQ(JSValueToBoolean(scriptContext, scriptValue), expectedValue);

     JSGlobalContextRelease(scriptContext);
 }

 static void expectScriptValueIsUndefined(WKSerializedScriptValueRef serializedScriptValue)
 {
     JSGlobalContextRef scriptContext = JSGlobalContextCreate(0);

     JSValueRef scriptValue = WKSerializedScriptValueDeserialize(serializedScriptValue, scriptContext, 0);
     EXPECT_TRUE(JSValueIsUndefined(scriptContext, scriptValue));

     JSGlobalContextRelease(scriptContext);
 }

 typedef void (^RunJavaScriptBlock)(WKSerializedScriptValueRef, WKErrorRef);

 static void callRunJavaScriptBlockAndRelease(WKSerializedScriptValueRef resultValue, WKErrorRef error, void* context)
 {
     auto block = (RunJavaScriptBlock)context;
     block(resultValue, error);
     Block_release(block);
 }

 static void runJavaScriptInMainFrame(WKPageRef pageRef, WKStringRef scriptRef, RunJavaScriptBlock block)
 {
     WKPageRunJavaScriptInMainFrame(pageRef, scriptRef, Block_copy(block), callRunJavaScriptBlockAndRelease);
 }

 TEST_F(WebKit2UserContentTest, AddUserStyleSheetBeforeCreatingView)
 {
     testFinished = false;
     [browsingContextGroup addUserStyleSheet:userStyleSheet baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil mainFrameOnly:YES];

     WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
     WKStringRef backgroundColorQuery = WKStringCreateWithUTF8CString(backgroundColorScript);
     wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
         runJavaScriptInMainFrame(wkView.pageRef, backgroundColorQuery, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
             expectScriptValueIsString(serializedScriptValue, greenInRGB);
             testFinished = true;
             WKRelease(backgroundColorQuery);
         });
     }];

     [wkView.browsingContextController loadHTMLString:htmlString baseURL:nil];

     TestWebKitAPI::Util::run(&testFinished);
 }

 TEST_F(WebKit2UserContentTest, AddUserStyleSheetAfterCreatingView)
 {
     testFinished = false;

     WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
     WKStringRef backgroundColorQuery = WKStringCreateWithUTF8CString(backgroundColorScript);
     wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
         runJavaScriptInMainFrame(wkView.pageRef, backgroundColorQuery, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
             expectScriptValueIsString(serializedScriptValue, greenInRGB);
             testFinished = true;
             WKRelease(backgroundColorQuery);
         });
     }];

     [browsingContextGroup addUserStyleSheet:userStyleSheet baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil mainFrameOnly:YES];

     [wkView.browsingContextController loadHTMLString:htmlString baseURL:nil];

     TestWebKitAPI::Util::run(&testFinished);
 }

 TEST_F(WebKit2UserContentTest, RemoveAllUserStyleSheets)
 {
     testFinished = false;
     [browsingContextGroup addUserStyleSheet:userStyleSheet baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil mainFrameOnly:YES];

     WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
     WKStringRef backgroundColorQuery = WKStringCreateWithUTF8CString(backgroundColorScript);
     wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
         runJavaScriptInMainFrame(wkView.pageRef, backgroundColorQuery, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
             expectScriptValueIsString(serializedScriptValue, redInRGB);
             testFinished = true;
             WKRelease(backgroundColorQuery);
         });
     }];

     [browsingContextGroup removeAllUserStyleSheets];

     [wkView.browsingContextController loadHTMLString:htmlString baseURL:nil];

     TestWebKitAPI::Util::run(&testFinished);
 }

 TEST_F(WebKit2UserContentTest, AddUserScriptBeforeCreatingView)
 {
     testFinished = false;
     [browsingContextGroup addUserScript:[NSString stringWithFormat:@"%s = true;", userScriptTestProperty] baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil injectionTime:kWKInjectAtDocumentStart mainFrameOnly:YES];

     WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
     WKStringRef userScriptTestPropertyString = WKStringCreateWithUTF8CString(userScriptTestProperty);
     wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
         runJavaScriptInMainFrame(wkView.pageRef, userScriptTestPropertyString, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
             expectScriptValueIsBoolean(serializedScriptValue, true);
             testFinished = true;
             WKRelease(userScriptTestPropertyString);
         });
     }];

     [wkView.browsingContextController loadHTMLString:@"" baseURL:nil];

     TestWebKitAPI::Util::run(&testFinished);
 }

 TEST_F(WebKit2UserContentTest, AddUserScriptAfterCreatingView)
 {
     testFinished = false;

     WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
     WKStringRef userScriptTestPropertyString = WKStringCreateWithUTF8CString(userScriptTestProperty);
     wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
         runJavaScriptInMainFrame(wkView.pageRef, userScriptTestPropertyString, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
             expectScriptValueIsBoolean(serializedScriptValue, true);
             testFinished = true;
             WKRelease(userScriptTestPropertyString);
         });
     }];

     [browsingContextGroup addUserScript:[NSString stringWithFormat:@"%s = true;", userScriptTestProperty] baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil injectionTime:kWKInjectAtDocumentStart mainFrameOnly:YES];

     [wkView.browsingContextController loadHTMLString:@"" baseURL:nil];

     TestWebKitAPI::Util::run(&testFinished);
 }

 TEST_F(WebKit2UserContentTest, RemoveAllUserScripts)
 {
     testFinished = false;
     [browsingContextGroup addUserScript:[NSString stringWithFormat:@"%s = true;", userScriptTestProperty] baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil injectionTime:kWKInjectAtDocumentStart mainFrameOnly:YES];

     WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
     WKStringRef userScriptTestPropertyString = WKStringCreateWithUTF8CString(userScriptTestProperty);
     wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
         runJavaScriptInMainFrame(wkView.pageRef, userScriptTestPropertyString, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
             expectScriptValueIsUndefined(serializedScriptValue);
             testFinished = true;
             WKRelease(userScriptTestPropertyString);
         });
     }];

     [browsingContextGroup removeAllUserScripts];

     [wkView.browsingContextController loadHTMLString:htmlString baseURL:nil];

     TestWebKitAPI::Util::run(&testFinished);
 }

 #endif // WK_API_ENABLED

 #endif
	/*
	* Copyright (C) 2012 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.
	*/

	#import "config.h"

	#if WK_HAVE_C_SPI

	#import "Test.h"

	#import "PlatformUtilities.h"
	#import "TestBrowsingContextLoadDelegate.h"
	#import <JavaScriptCore/JSRetainPtr.h>
	#import <JavaScriptCore/JavaScriptCore.h>
	#import <WebKit/WKSerializedScriptValue.h>
	#import <WebKit/WKViewPrivate.h>

	#if WK_API_ENABLED && PLATFORM(MAC)

	static bool testFinished = false;
	static NSString *htmlString = @"<body style='background-color: red'>";
	static NSString *userStyleSheet = @"body { background-color: green !important; }";
	static const char* backgroundColorScript = "window.getComputedStyle(document.body, null).getPropertyValue('background-color')";
	static const char* greenInRGB = "rgb(0, 128, 0)";
	static const char* redInRGB = "rgb(255, 0, 0)";
	static const char* userScriptTestProperty = "window._userScriptInstalled";

	namespace {
	class WebKit2UserContentTest : public ::testing::Test {
	public:
	WKProcessGroup *processGroup;
	WKBrowsingContextGroup *browsingContextGroup;

	WebKit2UserContentTest()
	: processGroup(nil)
	, browsingContextGroup(nil)
	{
	}

	virtual void SetUp()
	{
	processGroup = [[WKProcessGroup alloc] init];
	browsingContextGroup = [[WKBrowsingContextGroup alloc] initWithIdentifier:@"UserContentIdentifier"];
	}

	virtual void TearDown()
	{
	[browsingContextGroup release];
	[processGroup release];
	}
	};
	} // namespace

	static void expectScriptValueIsString(WKSerializedScriptValueRef serializedScriptValue, const char* expectedValue)
	{
	JSGlobalContextRef scriptContext = JSGlobalContextCreate(0);

	JSValueRef scriptValue = WKSerializedScriptValueDeserialize(serializedScriptValue, scriptContext, 0);
	EXPECT_TRUE(JSValueIsString(scriptContext, scriptValue));

	JSRetainPtr<JSStringRef> scriptString(Adopt, JSValueToStringCopy(scriptContext, scriptValue, 0));
	EXPECT_TRUE(JSStringIsEqualToUTF8CString(scriptString.get(), expectedValue));

	JSGlobalContextRelease(scriptContext);
	}

	static void expectScriptValueIsBoolean(WKSerializedScriptValueRef serializedScriptValue, bool expectedValue)
	{
	JSGlobalContextRef scriptContext = JSGlobalContextCreate(0);

	JSValueRef scriptValue = WKSerializedScriptValueDeserialize(serializedScriptValue, scriptContext, 0);
	EXPECT_TRUE(JSValueIsBoolean(scriptContext, scriptValue));
	EXPECT_EQ(JSValueToBoolean(scriptContext, scriptValue), expectedValue);

	JSGlobalContextRelease(scriptContext);
	}

	static void expectScriptValueIsUndefined(WKSerializedScriptValueRef serializedScriptValue)
	{
	JSGlobalContextRef scriptContext = JSGlobalContextCreate(0);

	JSValueRef scriptValue = WKSerializedScriptValueDeserialize(serializedScriptValue, scriptContext, 0);
	EXPECT_TRUE(JSValueIsUndefined(scriptContext, scriptValue));

	JSGlobalContextRelease(scriptContext);
	}

	typedef void (^RunJavaScriptBlock)(WKSerializedScriptValueRef, WKErrorRef);

	static void callRunJavaScriptBlockAndRelease(WKSerializedScriptValueRef resultValue, WKErrorRef error, void* context)
	{
	auto block = (RunJavaScriptBlock)context;
	block(resultValue, error);
	Block_release(block);
	}

	static void runJavaScriptInMainFrame(WKPageRef pageRef, WKStringRef scriptRef, RunJavaScriptBlock block)
	{
	WKPageRunJavaScriptInMainFrame(pageRef, scriptRef, Block_copy(block), callRunJavaScriptBlockAndRelease);
	}

	TEST_F(WebKit2UserContentTest, AddUserStyleSheetBeforeCreatingView)
	{
	testFinished = false;
	[browsingContextGroup addUserStyleSheet:userStyleSheet baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil mainFrameOnly:YES];

	WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
	WKStringRef backgroundColorQuery = WKStringCreateWithUTF8CString(backgroundColorScript);
	wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
	runJavaScriptInMainFrame(wkView.pageRef, backgroundColorQuery, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
	expectScriptValueIsString(serializedScriptValue, greenInRGB);
	testFinished = true;
	WKRelease(backgroundColorQuery);
	});
	}];

	[wkView.browsingContextController loadHTMLString:htmlString baseURL:nil];

	TestWebKitAPI::Util::run(&testFinished);
	}

	TEST_F(WebKit2UserContentTest, AddUserStyleSheetAfterCreatingView)
	{
	testFinished = false;

	WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
	WKStringRef backgroundColorQuery = WKStringCreateWithUTF8CString(backgroundColorScript);
	wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
	runJavaScriptInMainFrame(wkView.pageRef, backgroundColorQuery, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
	expectScriptValueIsString(serializedScriptValue, greenInRGB);
	testFinished = true;
	WKRelease(backgroundColorQuery);
	});
	}];

	[browsingContextGroup addUserStyleSheet:userStyleSheet baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil mainFrameOnly:YES];

	[wkView.browsingContextController loadHTMLString:htmlString baseURL:nil];

	TestWebKitAPI::Util::run(&testFinished);
	}

	TEST_F(WebKit2UserContentTest, RemoveAllUserStyleSheets)
	{
	testFinished = false;
	[browsingContextGroup addUserStyleSheet:userStyleSheet baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil mainFrameOnly:YES];

	WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
	WKStringRef backgroundColorQuery = WKStringCreateWithUTF8CString(backgroundColorScript);
	wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
	runJavaScriptInMainFrame(wkView.pageRef, backgroundColorQuery, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
	expectScriptValueIsString(serializedScriptValue, redInRGB);
	testFinished = true;
	WKRelease(backgroundColorQuery);
	});
	}];

	[browsingContextGroup removeAllUserStyleSheets];

	[wkView.browsingContextController loadHTMLString:htmlString baseURL:nil];

	TestWebKitAPI::Util::run(&testFinished);
	}

	TEST_F(WebKit2UserContentTest, AddUserScriptBeforeCreatingView)
	{
	testFinished = false;
	[browsingContextGroup addUserScript:[NSString stringWithFormat:@"%s = true;", userScriptTestProperty] baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil injectionTime:kWKInjectAtDocumentStart mainFrameOnly:YES];

	WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
	WKStringRef userScriptTestPropertyString = WKStringCreateWithUTF8CString(userScriptTestProperty);
	wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
	runJavaScriptInMainFrame(wkView.pageRef, userScriptTestPropertyString, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
	expectScriptValueIsBoolean(serializedScriptValue, true);
	testFinished = true;
	WKRelease(userScriptTestPropertyString);
	});
	}];

	[wkView.browsingContextController loadHTMLString:@"" baseURL:nil];

	TestWebKitAPI::Util::run(&testFinished);
	}

	TEST_F(WebKit2UserContentTest, AddUserScriptAfterCreatingView)
	{
	testFinished = false;

	WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
	WKStringRef userScriptTestPropertyString = WKStringCreateWithUTF8CString(userScriptTestProperty);
	wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
	runJavaScriptInMainFrame(wkView.pageRef, userScriptTestPropertyString, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
	expectScriptValueIsBoolean(serializedScriptValue, true);
	testFinished = true;
	WKRelease(userScriptTestPropertyString);
	});
	}];

	[browsingContextGroup addUserScript:[NSString stringWithFormat:@"%s = true;", userScriptTestProperty] baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil injectionTime:kWKInjectAtDocumentStart mainFrameOnly:YES];

	[wkView.browsingContextController loadHTMLString:@"" baseURL:nil];

	TestWebKitAPI::Util::run(&testFinished);
	}

	TEST_F(WebKit2UserContentTest, RemoveAllUserScripts)
	{
	testFinished = false;
	[browsingContextGroup addUserScript:[NSString stringWithFormat:@"%s = true;", userScriptTestProperty] baseURL:nil whitelistedURLPatterns:nil blacklistedURLPatterns:nil injectionTime:kWKInjectAtDocumentStart mainFrameOnly:YES];

	WKView *wkView = [[WKView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) processGroup:processGroup browsingContextGroup:browsingContextGroup];
	WKStringRef userScriptTestPropertyString = WKStringCreateWithUTF8CString(userScriptTestProperty);
	wkView.browsingContextController.loadDelegate = [[TestBrowsingContextLoadDelegate alloc] initWithBlockToRunOnLoad:^(WKBrowsingContextController *sender) {
	runJavaScriptInMainFrame(wkView.pageRef, userScriptTestPropertyString, ^(WKSerializedScriptValueRef serializedScriptValue, WKErrorRef error) {
	expectScriptValueIsUndefined(serializedScriptValue);
	testFinished = true;
	WKRelease(userScriptTestPropertyString);
	});
	}];

	[browsingContextGroup removeAllUserScripts];

	[wkView.browsingContextController loadHTMLString:htmlString baseURL:nil];

	TestWebKitAPI::Util::run(&testFinished);
	}

	#endif // WK_API_ENABLED

	#endif