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
151
|
// Copyright (c) 2009 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 "app/sql/connection.h"
#include "app/sql/statement.h"
#include "base/string_util.h"
#include "base/utf_string_conversions.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "webkit/database/databases_table.h"
namespace {
class TestErrorDelegate : public sql::ErrorDelegate {
public:
virtual ~TestErrorDelegate() { }
virtual int OnError(
int error, sql::Connection* connection, sql::Statement* stmt) {
return error;
}
};
} // namespace
namespace webkit_database {
static void CheckDetailsAreEqual(const DatabaseDetails& d1,
const DatabaseDetails& d2) {
EXPECT_EQ(d1.origin_identifier, d2.origin_identifier);
EXPECT_EQ(d1.database_name, d2.database_name);
EXPECT_EQ(d1.description, d2.description);
EXPECT_EQ(d1.estimated_size, d2.estimated_size);
}
static bool DatabasesTableIsEmpty(sql::Connection* db) {
sql::Statement statement(db->GetCachedStatement(
SQL_FROM_HERE, "SELECT COUNT(*) FROM Databases"));
return (statement.is_valid() && statement.Step() && !statement.ColumnInt(0));
}
TEST(DatabasesTableTest, TestIt) {
// Initialize the 'Databases' table.
sql::Connection db;
// Set an error delegate that will make all operations return false on error.
scoped_refptr<TestErrorDelegate> error_delegate(new TestErrorDelegate());
db.set_error_delegate(error_delegate);
// Initialize the temp dir and the 'Databases' table.
EXPECT_TRUE(db.OpenInMemory());
DatabasesTable databases_table(&db);
EXPECT_TRUE(databases_table.Init());
// The 'Databases' table should be empty.
EXPECT_TRUE(DatabasesTableIsEmpty(&db));
// Create the details for a databases.
DatabaseDetails details_in1;
DatabaseDetails details_out1;
details_in1.origin_identifier = ASCIIToUTF16("origin1");
details_in1.database_name = ASCIIToUTF16("db1");
details_in1.description = ASCIIToUTF16("description_db1");
details_in1.estimated_size = 100;
// Updating details for this database should fail.
EXPECT_FALSE(databases_table.UpdateDatabaseDetails(details_in1));
EXPECT_FALSE(databases_table.GetDatabaseDetails(
details_in1.origin_identifier,
details_in1.database_name,
&details_out1));
// Inserting details for this database should pass.
EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in1));
EXPECT_TRUE(databases_table.GetDatabaseDetails(
details_in1.origin_identifier,
details_in1.database_name,
&details_out1));
EXPECT_EQ(1, databases_table.GetDatabaseID(details_in1.origin_identifier,
details_in1.database_name));
// Check that the details were correctly written to the database.
CheckDetailsAreEqual(details_in1, details_out1);
// Check that inserting a duplicate row fails.
EXPECT_FALSE(databases_table.InsertDatabaseDetails(details_in1));
// Insert details for another database with the same origin.
DatabaseDetails details_in2;
details_in2.origin_identifier = ASCIIToUTF16("origin1");
details_in2.database_name = ASCIIToUTF16("db2");
details_in2.description = ASCIIToUTF16("description_db2");
details_in2.estimated_size = 200;
EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in2));
EXPECT_EQ(2, databases_table.GetDatabaseID(details_in2.origin_identifier,
details_in2.database_name));
// Insert details for a third database with a different origin.
DatabaseDetails details_in3;
details_in3.origin_identifier = ASCIIToUTF16("origin2");
details_in3.database_name = ASCIIToUTF16("db3");
details_in3.description = ASCIIToUTF16("description_db3");
details_in3.estimated_size = 300;
EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in3));
EXPECT_EQ(3, databases_table.GetDatabaseID(details_in3.origin_identifier,
details_in3.database_name));
// There should be no database with origin "origin3".
std::vector<DatabaseDetails> details_out_origin3;
EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOrigin(
ASCIIToUTF16("origin3"), &details_out_origin3));
EXPECT_TRUE(details_out_origin3.empty());
// There should be only two databases with origin "origin1".
std::vector<DatabaseDetails> details_out_origin1;
EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOrigin(
details_in1.origin_identifier, &details_out_origin1));
EXPECT_EQ(size_t(2), details_out_origin1.size());
CheckDetailsAreEqual(details_in1, details_out_origin1[0]);
CheckDetailsAreEqual(details_in2, details_out_origin1[1]);
// Get the list of all origins: should be "origin1" and "origin2".
std::vector<string16> origins_out;
EXPECT_TRUE(databases_table.GetAllOrigins(&origins_out));
EXPECT_EQ(size_t(2), origins_out.size());
EXPECT_EQ(details_in1.origin_identifier, origins_out[0]);
EXPECT_EQ(details_in3.origin_identifier, origins_out[1]);
// Delete an origin and check that it's no longer in the table.
origins_out.clear();
EXPECT_TRUE(databases_table.DeleteOrigin(details_in3.origin_identifier));
EXPECT_TRUE(databases_table.GetAllOrigins(&origins_out));
EXPECT_EQ(size_t(1), origins_out.size());
EXPECT_EQ(details_in1.origin_identifier, origins_out[0]);
// Deleting an origin that doesn't have any record in this table should fail.
EXPECT_FALSE(databases_table.DeleteOrigin(ASCIIToUTF16("unknown_origin")));
// Delete the details for 'db1' and check that they're no longer there.
EXPECT_TRUE(databases_table.DeleteDatabaseDetails(
details_in1.origin_identifier, details_in1.database_name));
EXPECT_FALSE(databases_table.GetDatabaseDetails(
details_in1.origin_identifier,
details_in1.database_name,
&details_out1));
// Check that trying to delete a record that doesn't exist fails.
EXPECT_FALSE(databases_table.DeleteDatabaseDetails(
ASCIIToUTF16("unknown_origin"), ASCIIToUTF16("unknown_database")));
}
} // namespace webkit_database
|
