1 /* 2 * Copyright (C) 2011 Apple Inc. All rights reserved. 3 * 4 * Redistribution and use in source and binary forms, with or without 5 * modification, are permitted provided that the following conditions 6 * are met: 7 * 1. Redistributions of source code must retain the above copyright 8 * notice, this list of conditions and the following disclaimer. 9 * 2. Redistributions in binary form must reproduce the above copyright 10 * notice, this list of conditions and the following disclaimer in the 11 * documentation and/or other materials provided with the distribution. 12 * 13 * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS'' 14 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, 15 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR 16 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS 17 * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR 18 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF 19 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS 20 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN 21 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) 22 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF 23 * THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 #include "Test.h" 27 28 #include "JavaScriptTest.h" 29 #include "PlatformUtilities.h" 30 #include "PlatformWebView.h" 31 32 namespace TestWebKitAPI { 33 34 static bool didFinishLoad; 35 36 static void didFinishLoadForFrame(WKPageRef, WKFrameRef, WKTypeRef, const void*) 37 { 38 didFinishLoad = true; 39 } 40 41 static void setPageLoaderClient(WKPageRef page) 42 { 43 WKPageLoaderClient loaderClient; 44 memset(&loaderClient, 0, sizeof(loaderClient)); 45 loaderClient.version = 0; 46 loaderClient.didFinishLoadForFrame = didFinishLoadForFrame; 47 48 WKPageSetPageLoaderClient(page, &loaderClient); 49 } 50 51 static WKRetainPtr<WKDataRef> createSessionStateContainingFormData(WKContextRef context) 52 { 53 PlatformWebView webView(context); 54 setPageLoaderClient(webView.page()); 55 56 WKPageLoadURL(webView.page(), adoptWK(Util::createURLForResource("simple-form", "html")).get()); 57 Util::run(&didFinishLoad); 58 didFinishLoad = false; 59 60 TEST_ASSERT_RETURN(runJSTest(webView.page(), "submitForm()", "undefined"), 0); 61 Util::run(&didFinishLoad); 62 didFinishLoad = false; 63 64 return adoptWK(WKPageCopySessionState(webView.page(), 0, 0)); 65 } 66 67 TEST(WebKit2, RestoreSessionStateContainingFormData) 68 { 69 WKRetainPtr<WKContextRef> context(AdoptWK, WKContextCreate()); 70 71 // FIXME: Once <rdar://problem/8708435> is fixed, we can move the creation of this 72 // PlatformWebView after the call to createSessionStaetContainingFormData. Until then, it must 73 // remain here to avoid a race condition between the UI and web processes. 74 PlatformWebView webView(context.get()); 75 setPageLoaderClient(webView.page()); 76 77 WKRetainPtr<WKDataRef> data = createSessionStateContainingFormData(context.get()); 78 TEST_ASSERT(data); 79 80 WKPageRestoreFromSessionState(webView.page(), data.get()); 81 Util::run(&didFinishLoad); 82 83 TEST_ASSERT(WKPageCanGoBack(webView.page())); 84 } 85 86 } // namespace TestWebKitAPI 87