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
32namespace TestWebKitAPI {
33
34static bool didFinishLoad;
35
36static void didFinishLoadForFrame(WKPageRef, WKFrameRef, WKTypeRef, const void*)
37{
38    didFinishLoad = true;
39}
40
41static 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
51static 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
67TEST(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