summaryrefslogtreecommitdiffstats
path: root/content/browser/in_process_webkit/dom_storage_browsertest.cc
blob: 9f335fc9a427baa4ec1fe28555b9f4e3d6eeef94 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
// 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/file_path.h"
#include "base/file_util.h"
#include "base/scoped_temp_dir.h"
#include "base/test/thread_test_helper.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "content/browser/in_process_webkit/dom_storage_context_impl.h"
#include "content/public/browser/browser_thread.h"
#include "content/public/common/url_constants.h"
#include "content/test/test_browser_context.h"
#include "googleurl/src/gurl.h"
#include "webkit/quota/special_storage_policy.h"

using content::BrowserContext;
using content::BrowserThread;

typedef InProcessBrowserTest DOMStorageBrowserTest;

namespace {

class TestSpecialStoragePolicy : public quota::SpecialStoragePolicy {
 public:
  TestSpecialStoragePolicy() {
  }

  virtual bool IsStorageProtected(const GURL& origin) OVERRIDE {
    return origin.SchemeIs(chrome::kChromeDevToolsScheme);
  }

  virtual bool IsStorageUnlimited(const GURL& origin) OVERRIDE {
    return false;
  }

  virtual bool IsFileHandler(const std::string& extension_id) OVERRIDE {
    return false;
  }

  virtual bool IsStorageSessionOnly(const GURL& origin) OVERRIDE {
    return false;
  }

  virtual bool HasSessionOnlyOrigins() OVERRIDE {
    return false;
  }
};

}  // namespace

#if defined(OS_MACOSX)
// http://crbug.com/115150. On Mac, failure rate is about 30% currently.
#define MAYBE_ClearLocalState DISABLED_ClearLocalState
#else
#define MAYBE_ClearLocalState ClearLocalState
#endif

// In proc browser test is needed here because ClearLocalState indirectly calls
// WebKit's isMainThread through WebSecurityOrigin->SecurityOrigin.
IN_PROC_BROWSER_TEST_F(DOMStorageBrowserTest, MAYBE_ClearLocalState) {
  // Create test files.
  ScopedTempDir temp_dir;
  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
  FilePath domstorage_dir = temp_dir.path().Append(
      DOMStorageContextImpl::kLocalStorageDirectory);
  ASSERT_TRUE(file_util::CreateDirectory(domstorage_dir));

  FilePath::StringType file_name_1(FILE_PATH_LITERAL("http_foo_0"));
  file_name_1.append(DOMStorageContextImpl::kLocalStorageExtension);
  FilePath::StringType file_name_2(FILE_PATH_LITERAL("chrome-devtools_foo_0"));
  file_name_2.append(DOMStorageContextImpl::kLocalStorageExtension);
  FilePath temp_file_path_1 = domstorage_dir.Append(file_name_1);
  FilePath temp_file_path_2 = domstorage_dir.Append(file_name_2);

  ASSERT_EQ(1, file_util::WriteFile(temp_file_path_1, ".", 1));
  ASSERT_EQ(1, file_util::WriteFile(temp_file_path_2, "o", 1));

  // Create the scope which will ensure we run the destructor of the webkit
  // context which should trigger the clean up.
  {
    TestBrowserContext browser_context;
    browser_context.SetSpecialStoragePolicy(new TestSpecialStoragePolicy());
    DOMStorageContextImpl* dom_storage_context =
        static_cast<DOMStorageContextImpl*>(
            BrowserContext::GetDOMStorageContext(&browser_context));
    dom_storage_context->set_data_path_for_testing(temp_dir.path());
    dom_storage_context->set_clear_local_state_on_exit(true);
  }
  // Make sure we wait until the destructor has run.
  scoped_refptr<base::ThreadTestHelper> helper(
      new base::ThreadTestHelper(
          BrowserThread::GetMessageLoopProxyForThread(
              BrowserThread::WEBKIT_DEPRECATED)));
  ASSERT_TRUE(helper->Run());

  // Because we specified https for scheme to be skipped the second file
  // should survive and the first go into vanity.
  ASSERT_FALSE(file_util::PathExists(temp_file_path_1));
  ASSERT_TRUE(file_util::PathExists(temp_file_path_2));
}