REGRESSION(r244162) Clearing website data from ephemeral WKWebsiteDataStore should...
[WebKit-https.git] / Tools / TestWebKitAPI / Tests / WebKitCocoa / WKWebsiteDatastore.mm
1 /*
2  * Copyright (C) 2018 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 #import "config.h"
27
28 #import "PlatformUtilities.h"
29 #import "Test.h"
30 #import "TestWKWebView.h"
31 #import <WebKit/WKWebsiteDataStorePrivate.h>
32 #import <WebKit/WebKit.h>
33 #import <wtf/text/WTFString.h>
34
35 static bool readyToContinue;
36
37 TEST(WKWebsiteDataStore, RemoveAndFetchData)
38 {
39     readyToContinue = false;
40     [[WKWebsiteDataStore defaultDataStore] removeDataOfTypes:[WKWebsiteDataStore _allWebsiteDataTypesIncludingPrivate] modifiedSince:[NSDate distantPast] completionHandler:^() {
41         readyToContinue = true;
42     }];
43     TestWebKitAPI::Util::run(&readyToContinue);
44     
45     readyToContinue = false;
46     [[WKWebsiteDataStore defaultDataStore] fetchDataRecordsOfTypes:[WKWebsiteDataStore _allWebsiteDataTypesIncludingPrivate] completionHandler:^(NSArray<WKWebsiteDataRecord *> *dataRecords) {
47         ASSERT_EQ(0u, dataRecords.count);
48         readyToContinue = true;
49     }];
50     TestWebKitAPI::Util::run(&readyToContinue);
51 }
52
53 TEST(WKWebsiteDataStore, RemoveEphemeralData)
54 {
55     auto configuration = adoptNS([WKWebViewConfiguration new]);
56     [configuration setWebsiteDataStore:[WKWebsiteDataStore nonPersistentDataStore]];
57     auto webView = adoptNS([[TestWKWebView alloc] initWithFrame:NSMakeRect(0, 0, 800, 600) configuration:configuration.get()]);
58     [webView synchronouslyLoadTestPageNamed:@"simple"];
59     __block bool done = false;
60     [[configuration websiteDataStore] removeDataOfTypes:[WKWebsiteDataStore allWebsiteDataTypes] modifiedSince:[NSDate distantPast] completionHandler: ^{
61         done = true;
62     }];
63     TestWebKitAPI::Util::run(&done);
64 }