1 files changed, 4 insertions, 13 deletions
diff --git a/Tools/TestWebKitAPI/Tests/WebKit2/RestoreSessionStateContainingFormData.cpp b/Tools/TestWebKitAPI/Tests/WebKit2/RestoreSessionStateContainingFormData.cpp
index b16537e70..079c51856 100644
--- a/Tools/TestWebKitAPI/Tests/WebKit2/RestoreSessionStateContainingFormData.cpp
+++ b/Tools/TestWebKitAPI/Tests/WebKit2/RestoreSessionStateContainingFormData.cpp
@@ -24,14 +24,10 @@
  */
 
 #include "config.h"
-
-#if WK_HAVE_C_SPI
-
 #include "JavaScriptTest.h"
 #include "PlatformUtilities.h"
 #include "PlatformWebView.h"
 #include "Test.h"
-#include <WebKit/WKSessionStateRef.h>
 
 namespace TestWebKitAPI {
 
@@ -53,7 +49,7 @@ static void setPageLoaderClient(WKPageRef page)
     WKPageSetPageLoaderClient(page, &loaderClient.base);
 }
 
-static WKRetainPtr<WKDataRef> createSessionStateDataContainingFormData(WKContextRef context)
+static WKRetainPtr<WKDataRef> createSessionStateContainingFormData(WKContextRef context)
 {
     PlatformWebView webView(context);
     setPageLoaderClient(webView.page());
@@ -66,8 +62,7 @@ static WKRetainPtr<WKDataRef> createSessionStateDataContainingFormData(WKContext
     Util::run(&didFinishLoad);
     didFinishLoad = false;
 
-    auto sessionState = adoptWK(static_cast<WKSessionStateRef>(WKPageCopySessionState(webView.page(), reinterpret_cast<void*>(1), nullptr)));
-    return adoptWK(WKSessionStateCopyData(sessionState.get()));
+    return adoptWK(WKPageCopySessionState(webView.page(), 0, 0));
 }
 
 TEST(WebKit2, RestoreSessionStateContainingFormData)
@@ -80,17 +75,13 @@ TEST(WebKit2, RestoreSessionStateContainingFormData)
     PlatformWebView webView(context.get());
     setPageLoaderClient(webView.page());
 
-    WKRetainPtr<WKDataRef> data = createSessionStateDataContainingFormData(context.get());
+    WKRetainPtr<WKDataRef> data = createSessionStateContainingFormData(context.get());
     EXPECT_NOT_NULL(data);
 
-    auto sessionState = adoptWK(WKSessionStateCreateFromData(data.get()));
-    WKPageRestoreFromSessionState(webView.page(), sessionState.get());
-
+    WKPageRestoreFromSessionState(webView.page(), data.get());
     Util::run(&didFinishLoad);
 
     EXPECT_TRUE(WKPageCanGoBack(webView.page()));
 }
 
 } // namespace TestWebKitAPI
-
-#endif