| // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "base/bind.h" |
| #include "base/command_line.h" |
| #include "base/files/file_util.h" |
| #include "base/run_loop.h" |
| #include "base/test/bind_test_util.h" |
| #include "base/threading/thread_restrictions.h" |
| #include "build/build_config.h" |
| #include "components/services/storage/public/cpp/constants.h" |
| #include "content/browser/dom_storage/dom_storage_context_wrapper.h" |
| #include "content/browser/dom_storage/dom_storage_database.h" |
| #include "content/browser/dom_storage/dom_storage_types.h" |
| #include "content/browser/dom_storage/local_storage_context_mojo.h" |
| #include "content/browser/dom_storage/session_storage_context_mojo.h" |
| #include "content/browser/dom_storage/session_storage_namespace_impl.h" |
| #include "content/browser/web_contents/web_contents_impl.h" |
| #include "content/public/browser/browser_context.h" |
| #include "content/public/browser/browser_thread.h" |
| #include "content/public/browser/storage_partition.h" |
| #include "content/public/browser/storage_usage_info.h" |
| #include "content/public/common/content_paths.h" |
| #include "content/public/common/content_switches.h" |
| #include "content/public/test/browser_test_utils.h" |
| #include "content/public/test/content_browser_test.h" |
| #include "content/public/test/content_browser_test_utils.h" |
| #include "content/public/test/test_launcher.h" |
| #include "content/shell/browser/shell.h" |
| #include "content/shell/browser/shell_browser_context.h" |
| #include "content/shell/browser/shell_content_browser_client.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "third_party/blink/public/common/features.h" |
| |
| namespace content { |
| |
| // This browser test is aimed towards exercising the DOMStorage system |
| // from end-to-end. |
| class DOMStorageBrowserTest : public ContentBrowserTest { |
| public: |
| DOMStorageBrowserTest() {} |
| |
| void SimpleTest(const GURL& test_url, bool incognito) { |
| // The test page will perform tests then navigate to either |
| // a #pass or #fail ref. |
| Shell* the_browser = incognito ? CreateOffTheRecordBrowser() : shell(); |
| NavigateToURLBlockUntilNavigationsComplete(the_browser, test_url, 2); |
| std::string result = |
| the_browser->web_contents()->GetLastCommittedURL().ref(); |
| if (result != "pass") { |
| std::string js_result; |
| ASSERT_TRUE(ExecuteScriptAndExtractString( |
| the_browser, "window.domAutomationController.send(getLog())", |
| &js_result)); |
| FAIL() << "Failed: " << js_result; |
| } |
| } |
| |
| StoragePartition* partition() { |
| return BrowserContext::GetDefaultStoragePartition( |
| shell()->web_contents()->GetBrowserContext()); |
| } |
| |
| std::vector<StorageUsageInfo> GetUsage() { |
| base::RunLoop loop; |
| std::vector<StorageUsageInfo> usage; |
| partition()->GetDOMStorageContext()->GetLocalStorageUsage( |
| base::BindLambdaForTesting([&](const std::vector<StorageUsageInfo>& u) { |
| usage = u; |
| loop.Quit(); |
| })); |
| loop.Run(); |
| return usage; |
| } |
| |
| void DeletePhysicalOrigin(url::Origin origin) { |
| base::RunLoop loop; |
| partition()->GetDOMStorageContext()->DeleteLocalStorage(origin, |
| loop.QuitClosure()); |
| loop.Run(); |
| } |
| |
| DOMStorageContextWrapper* context_wrapper() { |
| return static_cast<DOMStorageContextWrapper*>( |
| partition()->GetDOMStorageContext()); |
| } |
| |
| base::SequencedTaskRunner* mojo_task_runner() { |
| return context_wrapper()->mojo_task_runner(); |
| } |
| |
| SessionStorageContextMojo* session_storage_context() { |
| return context_wrapper()->mojo_session_state_; |
| } |
| }; |
| |
| static const bool kIncognito = true; |
| static const bool kNotIncognito = false; |
| |
| IN_PROC_BROWSER_TEST_F(DOMStorageBrowserTest, SanityCheck) { |
| SimpleTest(GetTestUrl("dom_storage", "sanity_check.html"), kNotIncognito); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(DOMStorageBrowserTest, SanityCheckIncognito) { |
| SimpleTest(GetTestUrl("dom_storage", "sanity_check.html"), kIncognito); |
| } |
| |
| // http://crbug.com/654704 PRE_ tests aren't supported on Android. |
| // Also crbug/1027940: Fails on linux-chromeos-rel and other platforms. |
| IN_PROC_BROWSER_TEST_F(DOMStorageBrowserTest, PRE_DataPersists) { |
| SimpleTest(GetTestUrl("dom_storage", "store_data.html"), kNotIncognito); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(DOMStorageBrowserTest, DISABLED_DataPersists) { |
| SimpleTest(GetTestUrl("dom_storage", "verify_data.html"), kNotIncognito); |
| } |
| |
| IN_PROC_BROWSER_TEST_F(DOMStorageBrowserTest, DeletePhysicalOrigin) { |
| EXPECT_EQ(0U, GetUsage().size()); |
| SimpleTest(GetTestUrl("dom_storage", "store_data.html"), kNotIncognito); |
| std::vector<StorageUsageInfo> usage = GetUsage(); |
| ASSERT_EQ(1U, usage.size()); |
| DeletePhysicalOrigin(usage[0].origin); |
| EXPECT_EQ(0U, GetUsage().size()); |
| } |
| |
| // On Windows file://localhost/C:/src/chromium/src/content/test/data/title1.html |
| // doesn't work. |
| #if !defined(OS_WIN) |
| // Regression test for https://crbug.com/776160. The test verifies that there |
| // is no disagreement between 1) site URL used for browser-side isolation |
| // enforcement and 2) the origin requested by Blink. Before this bug was fixed, |
| // (1) was file://localhost/ and (2) was file:// - this led to renderer kills. |
| IN_PROC_BROWSER_TEST_F(DOMStorageBrowserTest, FileUrlWithHost) { |
| // Navigate to file://localhost/.../title1.html |
| GURL regular_file_url = GetTestUrl(nullptr, "title1.html"); |
| GURL::Replacements host_replacement; |
| host_replacement.SetHostStr("localhost"); |
| GURL file_with_host_url = |
| regular_file_url.ReplaceComponents(host_replacement); |
| EXPECT_TRUE(NavigateToURL(shell(), file_with_host_url)); |
| EXPECT_THAT(shell()->web_contents()->GetLastCommittedURL().spec(), |
| testing::StartsWith("file://localhost/")); |
| EXPECT_THAT(shell()->web_contents()->GetLastCommittedURL().spec(), |
| testing::EndsWith("/title1.html")); |
| |
| // Verify that window.localStorage works fine. |
| std::string result; |
| std::string script = R"( |
| localStorage["foo"] = "bar"; |
| domAutomationController.send(localStorage["foo"]); |
| )"; |
| EXPECT_TRUE(ExecuteScriptAndExtractString(shell(), script, &result)); |
| EXPECT_EQ("bar", result); |
| } |
| #endif |
| |
| IN_PROC_BROWSER_TEST_F(DOMStorageBrowserTest, DataMigrates) { |
| const base::FilePath legacy_local_storage_path = |
| partition()->GetPath().Append(storage::kLocalStoragePath); |
| base::FilePath db_path = legacy_local_storage_path.Append( |
| LocalStorageContextMojo::LegacyDatabaseFileNameFromOrigin( |
| url::Origin::Create(GetTestUrl("dom_storage", "store_data.html")))); |
| { |
| base::ScopedAllowBlockingForTesting allow_blocking; |
| EXPECT_TRUE(base::CreateDirectory(legacy_local_storage_path)); |
| DOMStorageDatabase db(db_path); |
| DOMStorageValuesMap data; |
| data[base::ASCIIToUTF16("foo")] = |
| base::NullableString16(base::ASCIIToUTF16("bar"), false); |
| db.CommitChanges(false, data); |
| EXPECT_TRUE(base::PathExists(db_path)); |
| } |
| std::vector<StorageUsageInfo> usage = GetUsage(); |
| ASSERT_EQ(1U, usage.size()); |
| EXPECT_GT(usage[0].total_size_bytes, 6u); |
| |
| SimpleTest(GetTestUrl("dom_storage", "verify_data.html"), kNotIncognito); |
| usage = GetUsage(); |
| ASSERT_EQ(1U, usage.size()); |
| EXPECT_GT(usage[0].total_size_bytes, 6u); |
| { |
| base::ScopedAllowBlockingForTesting allow_blocking; |
| EXPECT_FALSE(base::PathExists(db_path)); |
| } |
| } |
| |
| } // namespace content |