0970bd2f9e34b938889b298fb1f668586ca09206
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WebKit2 / RestoreSessionStateContainingFormData.cpp
1 /*
2  * Copyright (C) 2011 Apple Inc. All rights reserved.
3  *
4  * Redistribution and use in source and binary forms, with or without
5  * modification, are permitted provided that the following conditions
6  * are met:
7  * 1. Redistributions of source code must retain the above copyright
8  *    notice, this list of conditions and the following disclaimer.
9  * 2. Redistributions in binary form must reproduce the above copyright
10  *    notice, this list of conditions and the following disclaimer in the
11  *    documentation and/or other materials provided with the distribution.
12  *
13  * THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
14  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
15  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
16  * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
17  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
18  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
19  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
20  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
21  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
22  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
23  * THE POSSIBILITY OF SUCH DAMAGE.
24  */
25
26 #include "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