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