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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
|
// Copyright (c) 2011 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/bind_helpers.h"
#include "base/file_util.h"
#include "base/utf_string_conversions.h"
#include "chrome/browser/browsing_data_database_helper.h"
#include "chrome/browser/browsing_data_helper_browsertest.h"
#include "chrome/browser/profiles/profile.h"
#include "chrome/browser/ui/browser.h"
#include "chrome/test/base/in_process_browser_test.h"
#include "chrome/test/base/ui_test_utils.h"
#include "content/test/test_browser_thread.h"
using content::BrowserThread;
namespace {
typedef BrowsingDataHelperCallback<BrowsingDataDatabaseHelper::DatabaseInfo>
TestCompletionCallback;
const char kTestIdentifier1[] = "http_www.google.com_0";
const char kTestIdentifierExtension[] =
"chrome-extension_behllobkkfkfnphdnhnkndlbkcpglgmj_0";
class BrowsingDataDatabaseHelperTest : public InProcessBrowserTest {
public:
virtual void CreateDatabases() {
webkit_database::DatabaseTracker* db_tracker =
browser()->profile()->GetDatabaseTracker();
string16 db_name = ASCIIToUTF16("db");
string16 description = ASCIIToUTF16("db_description");
int64 size;
string16 identifier1(UTF8ToUTF16(kTestIdentifier1));
db_tracker->DatabaseOpened(identifier1, db_name, description, 1, &size);
db_tracker->DatabaseClosed(identifier1, db_name);
FilePath db_path1 = db_tracker->GetFullDBFilePath(identifier1, db_name);
file_util::CreateDirectory(db_path1.DirName());
ASSERT_EQ(0, file_util::WriteFile(db_path1, NULL, 0));
string16 identifierExtension(UTF8ToUTF16(kTestIdentifierExtension));
db_tracker->DatabaseOpened(identifierExtension, db_name, description, 1,
&size);
db_tracker->DatabaseClosed(identifierExtension, db_name);
FilePath db_path2 =
db_tracker->GetFullDBFilePath(identifierExtension, db_name);
file_util::CreateDirectory(db_path2.DirName());
ASSERT_EQ(0, file_util::WriteFile(db_path2, NULL, 0));
std::vector<webkit_database::OriginInfo> origins;
db_tracker->GetAllOriginsInfo(&origins);
ASSERT_EQ(2U, origins.size());
}
};
// Called back by BrowsingDataDatabaseHelper on the UI thread once the database
// information has been retrieved.
class StopTestOnCallback {
public:
explicit StopTestOnCallback(
BrowsingDataDatabaseHelper* database_helper)
: database_helper_(database_helper) {
DCHECK(database_helper_);
}
void Callback(const std::list<BrowsingDataDatabaseHelper::DatabaseInfo>&
database_info_list) {
DCHECK(BrowserThread::CurrentlyOn(BrowserThread::UI));
ASSERT_EQ(1UL, database_info_list.size());
EXPECT_EQ(std::string(kTestIdentifier1),
database_info_list.begin()->origin_identifier);
MessageLoop::current()->Quit();
}
private:
BrowsingDataDatabaseHelper* database_helper_;
};
// Flaky on Win/Mac/Linux: http://crbug.com/92460
IN_PROC_BROWSER_TEST_F(BrowsingDataDatabaseHelperTest, DISABLED_FetchData) {
CreateDatabases();
scoped_refptr<BrowsingDataDatabaseHelper> database_helper(
new BrowsingDataDatabaseHelper(browser()->profile()));
StopTestOnCallback stop_test_on_callback(database_helper);
database_helper->StartFetching(
base::Bind(&StopTestOnCallback::Callback,
base::Unretained(&stop_test_on_callback)));
// Blocks until StopTestOnCallback::Callback is notified.
ui_test_utils::RunMessageLoop();
}
IN_PROC_BROWSER_TEST_F(BrowsingDataDatabaseHelperTest, CannedAddDatabase) {
const GURL origin1("http://host1:1/");
const GURL origin2("http://host2:1/");
const char origin_str1[] = "http_host1_1";
const char origin_str2[] = "http_host2_1";
const char db1[] = "db1";
const char db2[] = "db2";
const char db3[] = "db3";
scoped_refptr<CannedBrowsingDataDatabaseHelper> helper(
new CannedBrowsingDataDatabaseHelper(browser()->profile()));
helper->AddDatabase(origin1, db1, "");
helper->AddDatabase(origin1, db2, "");
helper->AddDatabase(origin2, db3, "");
TestCompletionCallback callback;
helper->StartFetching(
base::Bind(&TestCompletionCallback::callback,
base::Unretained(&callback)));
std::list<BrowsingDataDatabaseHelper::DatabaseInfo> result =
callback.result();
ASSERT_EQ(3u, result.size());
std::list<BrowsingDataDatabaseHelper::DatabaseInfo>::iterator info =
result.begin();
EXPECT_STREQ(origin_str1, info->origin_identifier.c_str());
EXPECT_STREQ(db1, info->database_name.c_str());
info++;
EXPECT_STREQ(origin_str1, info->origin_identifier.c_str());
EXPECT_STREQ(db2, info->database_name.c_str());
info++;
EXPECT_STREQ(origin_str2, info->origin_identifier.c_str());
EXPECT_STREQ(db3, info->database_name.c_str());
}
IN_PROC_BROWSER_TEST_F(BrowsingDataDatabaseHelperTest, CannedUnique) {
const GURL origin("http://host1:1/");
const char origin_str[] = "http_host1_1";
const char db[] = "db1";
scoped_refptr<CannedBrowsingDataDatabaseHelper> helper(
new CannedBrowsingDataDatabaseHelper(browser()->profile()));
helper->AddDatabase(origin, db, "");
helper->AddDatabase(origin, db, "");
TestCompletionCallback callback;
helper->StartFetching(
base::Bind(&TestCompletionCallback::callback,
base::Unretained(&callback)));
std::list<BrowsingDataDatabaseHelper::DatabaseInfo> result =
callback.result();
ASSERT_EQ(1u, result.size());
EXPECT_STREQ(origin_str, result.begin()->origin_identifier.c_str());
EXPECT_STREQ(db, result.begin()->database_name.c_str());
}
} // namespace
|