| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | |
| 2 // Use of this source code is governed by a BSD-style license that can be | |
| 3 // found in the LICENSE file. | |
| 4 | |
| 5 #include "base/bind.h" | |
| 6 #include "base/strings/string_util.h" | |
| 7 #include "base/strings/utf_string_conversions.h" | |
| 8 #include "sql/connection.h" | |
| 9 #include "sql/statement.h" | |
| 10 #include "sql/test/scoped_error_ignorer.h" | |
| 11 #include "testing/gtest/include/gtest/gtest.h" | |
| 12 #include "third_party/sqlite/sqlite3.h" | |
| 13 #include "webkit/browser/database/databases_table.h" | |
| 14 | |
| 15 using base::ASCIIToUTF16; | |
| 16 | |
| 17 namespace webkit_database { | |
| 18 | |
| 19 static void CheckDetailsAreEqual(const DatabaseDetails& d1, | |
| 20 const DatabaseDetails& d2) { | |
| 21 EXPECT_EQ(d1.origin_identifier, d2.origin_identifier); | |
| 22 EXPECT_EQ(d1.database_name, d2.database_name); | |
| 23 EXPECT_EQ(d1.description, d2.description); | |
| 24 EXPECT_EQ(d1.estimated_size, d2.estimated_size); | |
| 25 } | |
| 26 | |
| 27 static bool DatabasesTableIsEmpty(sql::Connection* db) { | |
| 28 sql::Statement statement(db->GetCachedStatement( | |
| 29 SQL_FROM_HERE, "SELECT COUNT(*) FROM Databases")); | |
| 30 return (statement.is_valid() && statement.Step() && !statement.ColumnInt(0)); | |
| 31 } | |
| 32 | |
| 33 TEST(DatabasesTableTest, TestIt) { | |
| 34 // Initialize the 'Databases' table. | |
| 35 sql::Connection db; | |
| 36 | |
| 37 sql::ScopedErrorIgnorer ignore_errors; | |
| 38 // TODO(shess): Suppressing SQLITE_CONSTRAINT because the code | |
| 39 // expects that and handles the resulting error. Consider revising | |
| 40 // the code to use INSERT OR IGNORE (which would not throw | |
| 41 // SQLITE_CONSTRAINT) and then check ChangeCount() to see if any | |
| 42 // changes were made. | |
| 43 ignore_errors.IgnoreError(SQLITE_CONSTRAINT); | |
| 44 | |
| 45 // Initialize the temp dir and the 'Databases' table. | |
| 46 EXPECT_TRUE(db.OpenInMemory()); | |
| 47 DatabasesTable databases_table(&db); | |
| 48 EXPECT_TRUE(databases_table.Init()); | |
| 49 | |
| 50 // The 'Databases' table should be empty. | |
| 51 EXPECT_TRUE(DatabasesTableIsEmpty(&db)); | |
| 52 | |
| 53 // Create the details for a databases. | |
| 54 DatabaseDetails details_in1; | |
| 55 DatabaseDetails details_out1; | |
| 56 details_in1.origin_identifier = "origin1"; | |
| 57 details_in1.database_name = ASCIIToUTF16("db1"); | |
| 58 details_in1.description = ASCIIToUTF16("description_db1"); | |
| 59 details_in1.estimated_size = 100; | |
| 60 | |
| 61 // Updating details for this database should fail. | |
| 62 EXPECT_FALSE(databases_table.UpdateDatabaseDetails(details_in1)); | |
| 63 EXPECT_FALSE(databases_table.GetDatabaseDetails( | |
| 64 details_in1.origin_identifier, | |
| 65 details_in1.database_name, | |
| 66 &details_out1)); | |
| 67 | |
| 68 // Inserting details for this database should pass. | |
| 69 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in1)); | |
| 70 EXPECT_TRUE(databases_table.GetDatabaseDetails( | |
| 71 details_in1.origin_identifier, | |
| 72 details_in1.database_name, | |
| 73 &details_out1)); | |
| 74 EXPECT_EQ(1, databases_table.GetDatabaseID(details_in1.origin_identifier, | |
| 75 details_in1.database_name)); | |
| 76 | |
| 77 // Check that the details were correctly written to the database. | |
| 78 CheckDetailsAreEqual(details_in1, details_out1); | |
| 79 | |
| 80 // Check that inserting a duplicate row fails. | |
| 81 EXPECT_FALSE(databases_table.InsertDatabaseDetails(details_in1)); | |
| 82 | |
| 83 // Insert details for another database with the same origin. | |
| 84 DatabaseDetails details_in2; | |
| 85 details_in2.origin_identifier = "origin1"; | |
| 86 details_in2.database_name = ASCIIToUTF16("db2"); | |
| 87 details_in2.description = ASCIIToUTF16("description_db2"); | |
| 88 details_in2.estimated_size = 200; | |
| 89 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in2)); | |
| 90 EXPECT_EQ(2, databases_table.GetDatabaseID(details_in2.origin_identifier, | |
| 91 details_in2.database_name)); | |
| 92 | |
| 93 // Insert details for a third database with a different origin. | |
| 94 DatabaseDetails details_in3; | |
| 95 details_in3.origin_identifier = "origin2"; | |
| 96 details_in3.database_name = ASCIIToUTF16("db3"); | |
| 97 details_in3.description = ASCIIToUTF16("description_db3"); | |
| 98 details_in3.estimated_size = 300; | |
| 99 EXPECT_TRUE(databases_table.InsertDatabaseDetails(details_in3)); | |
| 100 EXPECT_EQ(3, databases_table.GetDatabaseID(details_in3.origin_identifier, | |
| 101 details_in3.database_name)); | |
| 102 | |
| 103 // There should be no database with origin "origin3". | |
| 104 std::vector<DatabaseDetails> details_out_origin3; | |
| 105 EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOriginIdentifier( | |
| 106 "origin3", &details_out_origin3)); | |
| 107 EXPECT_TRUE(details_out_origin3.empty()); | |
| 108 | |
| 109 // There should be only two databases with origin "origin1". | |
| 110 std::vector<DatabaseDetails> details_out_origin1; | |
| 111 EXPECT_TRUE(databases_table.GetAllDatabaseDetailsForOriginIdentifier( | |
| 112 details_in1.origin_identifier, &details_out_origin1)); | |
| 113 EXPECT_EQ(size_t(2), details_out_origin1.size()); | |
| 114 CheckDetailsAreEqual(details_in1, details_out_origin1[0]); | |
| 115 CheckDetailsAreEqual(details_in2, details_out_origin1[1]); | |
| 116 | |
| 117 // Get the list of all origins: should be "origin1" and "origin2". | |
| 118 std::vector<std::string> origins_out; | |
| 119 EXPECT_TRUE(databases_table.GetAllOriginIdentifiers(&origins_out)); | |
| 120 EXPECT_EQ(size_t(2), origins_out.size()); | |
| 121 EXPECT_EQ(details_in1.origin_identifier, origins_out[0]); | |
| 122 EXPECT_EQ(details_in3.origin_identifier, origins_out[1]); | |
| 123 | |
| 124 // Delete an origin and check that it's no longer in the table. | |
| 125 origins_out.clear(); | |
| 126 EXPECT_TRUE(databases_table.DeleteOriginIdentifier( | |
| 127 details_in3.origin_identifier)); | |
| 128 EXPECT_TRUE(databases_table.GetAllOriginIdentifiers(&origins_out)); | |
| 129 EXPECT_EQ(size_t(1), origins_out.size()); | |
| 130 EXPECT_EQ(details_in1.origin_identifier, origins_out[0]); | |
| 131 | |
| 132 // Deleting an origin that doesn't have any record in this table should fail. | |
| 133 EXPECT_FALSE(databases_table.DeleteOriginIdentifier("unknown_origin")); | |
| 134 | |
| 135 // Delete the details for 'db1' and check that they're no longer there. | |
| 136 EXPECT_TRUE(databases_table.DeleteDatabaseDetails( | |
| 137 details_in1.origin_identifier, details_in1.database_name)); | |
| 138 EXPECT_FALSE(databases_table.GetDatabaseDetails( | |
| 139 details_in1.origin_identifier, | |
| 140 details_in1.database_name, | |
| 141 &details_out1)); | |
| 142 | |
| 143 // Check that trying to delete a record that doesn't exist fails. | |
| 144 EXPECT_FALSE(databases_table.DeleteDatabaseDetails( | |
| 145 "unknown_origin", ASCIIToUTF16("unknown_database"))); | |
| 146 | |
| 147 ASSERT_TRUE(ignore_errors.CheckIgnoredErrors()); | |
| 148 } | |
| 149 | |
| 150 } // namespace webkit_database | |
| OLD | NEW |