| // 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/file_path.h" |
| #include "base/file_util.h" |
| #include "base/memory/scoped_ptr.h" |
| #include "base/message_loop_proxy.h" |
| #include "base/platform_file.h" |
| #include "base/scoped_temp_dir.h" |
| #include "base/time.h" |
| #include "base/utf_string_conversions.h" |
| #include "net/base/net_errors.h" |
| #include "net/base/test_completion_callback.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| #include "webkit/database/database_tracker.h" |
| #include "webkit/database/database_util.h" |
| #include "webkit/quota/mock_special_storage_policy.h" |
| #include "webkit/quota/quota_manager.h" |
| |
| namespace { |
| |
| const char kOrigin1Url[] = "http://origin1"; |
| const char kOrigin2Url[] = "http://protected_origin2"; |
| |
| class TestObserver : public webkit_database::DatabaseTracker::Observer { |
| public: |
| TestObserver() : new_notification_received_(false) {} |
| virtual ~TestObserver() {} |
| virtual void OnDatabaseSizeChanged(const string16& origin_identifier, |
| const string16& database_name, |
| int64 database_size) { |
| new_notification_received_ = true; |
| origin_identifier_ = origin_identifier; |
| database_name_ = database_name; |
| database_size_ = database_size; |
| } |
| virtual void OnDatabaseScheduledForDeletion(const string16& origin_identifier, |
| const string16& database_name) { |
| new_notification_received_ = true; |
| origin_identifier_ = origin_identifier; |
| database_name_ = database_name; |
| } |
| bool DidReceiveNewNotification() { |
| bool temp_new_notification_received = new_notification_received_; |
| new_notification_received_ = false; |
| return temp_new_notification_received; |
| } |
| string16 GetNotificationOriginIdentifier() { return origin_identifier_; } |
| string16 GetNotificationDatabaseName() { return database_name_; } |
| int64 GetNotificationDatabaseSize() { return database_size_; } |
| |
| private: |
| bool new_notification_received_; |
| string16 origin_identifier_; |
| string16 database_name_; |
| int64 database_size_; |
| }; |
| |
| void CheckNotificationReceived(TestObserver* observer, |
| const string16& expected_origin_identifier, |
| const string16& expected_database_name, |
| int64 expected_database_size) { |
| EXPECT_TRUE(observer->DidReceiveNewNotification()); |
| EXPECT_EQ(expected_origin_identifier, |
| observer->GetNotificationOriginIdentifier()); |
| EXPECT_EQ(expected_database_name, |
| observer->GetNotificationDatabaseName()); |
| EXPECT_EQ(expected_database_size, |
| observer->GetNotificationDatabaseSize()); |
| } |
| |
| class TestQuotaManagerProxy : public quota::QuotaManagerProxy { |
| public: |
| TestQuotaManagerProxy() |
| : QuotaManagerProxy(NULL, NULL), |
| registered_client_(NULL) { |
| } |
| |
| virtual ~TestQuotaManagerProxy() { |
| EXPECT_FALSE(registered_client_); |
| } |
| |
| virtual void RegisterClient(quota::QuotaClient* client) { |
| EXPECT_FALSE(registered_client_); |
| registered_client_ = client; |
| } |
| |
| virtual void NotifyStorageAccessed(quota::QuotaClient::ID client_id, |
| const GURL& origin, |
| quota::StorageType type) { |
| EXPECT_EQ(quota::QuotaClient::kDatabase, client_id); |
| EXPECT_EQ(quota::kStorageTypeTemporary, type); |
| accesses_[origin] += 1; |
| } |
| |
| virtual void NotifyStorageModified(quota::QuotaClient::ID client_id, |
| const GURL& origin, |
| quota::StorageType type, |
| int64 delta) { |
| EXPECT_EQ(quota::QuotaClient::kDatabase, client_id); |
| EXPECT_EQ(quota::kStorageTypeTemporary, type); |
| modifications_[origin].first += 1; |
| modifications_[origin].second += delta; |
| } |
| |
| // Not needed for our tests. |
| virtual void NotifyOriginInUse(const GURL& origin) {} |
| virtual void NotifyOriginNoLongerInUse(const GURL& origin) {} |
| |
| void SimulateQuotaManagerDestroyed() { |
| if (registered_client_) { |
| registered_client_->OnQuotaManagerDestroyed(); |
| registered_client_ = NULL; |
| } |
| } |
| |
| bool WasAccessNotified(const GURL& origin) { |
| return accesses_[origin] != 0; |
| } |
| |
| bool WasModificationNotified(const GURL& origin, int64 amount) { |
| return modifications_[origin].first != 0 && |
| modifications_[origin].second == amount; |
| } |
| |
| void reset() { |
| accesses_.clear(); |
| modifications_.clear(); |
| } |
| |
| quota::QuotaClient* registered_client_; |
| |
| // Map from origin to count of access notifications. |
| std::map<GURL, int> accesses_; |
| |
| // Map from origin to <count, sum of deltas> |
| std::map<GURL, std::pair<int, int64> > modifications_; |
| }; |
| |
| |
| bool EnsureFileOfSize(const FilePath& file_path, int64 length) { |
| base::PlatformFileError error_code(base::PLATFORM_FILE_ERROR_FAILED); |
| base::PlatformFile file = |
| base::CreatePlatformFile( |
| file_path, |
| base::PLATFORM_FILE_OPEN_ALWAYS | base::PLATFORM_FILE_WRITE, |
| NULL, |
| &error_code); |
| if (error_code != base::PLATFORM_FILE_OK) |
| return false; |
| if (!base::TruncatePlatformFile(file, length)) |
| error_code = base::PLATFORM_FILE_ERROR_FAILED; |
| base::ClosePlatformFile(file); |
| return error_code == base::PLATFORM_FILE_OK; |
| } |
| |
| } // namespace |
| |
| namespace webkit_database { |
| |
| // We declare a helper class, and make it a friend of DatabaseTracker using |
| // the FRIEND_TEST() macro, and we implement all tests we want to run as |
| // static methods of this class. Then we make our TEST() targets call these |
| // static functions. This allows us to run each test in normal mode and |
| // incognito mode without writing the same code twice. |
| class DatabaseTracker_TestHelper_Test { |
| public: |
| static void TestDeleteOpenDatabase(bool incognito_mode) { |
| // Initialize the tracker database. |
| ScopedTempDir temp_dir; |
| ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| scoped_refptr<quota::MockSpecialStoragePolicy> special_storage_policy = |
| new quota::MockSpecialStoragePolicy; |
| special_storage_policy->AddProtected(GURL(kOrigin2Url)); |
| scoped_refptr<DatabaseTracker> tracker( |
| new DatabaseTracker(temp_dir.path(), incognito_mode, false, |
| special_storage_policy, NULL, NULL)); |
| |
| // Create and open three databases. |
| int64 database_size = 0; |
| const string16 kOrigin1 = |
| DatabaseUtil::GetOriginIdentifier(GURL(kOrigin1Url)); |
| const string16 kOrigin2 = |
| DatabaseUtil::GetOriginIdentifier(GURL(kOrigin2Url)); |
| const string16 kDB1 = ASCIIToUTF16("db1"); |
| const string16 kDB2 = ASCIIToUTF16("db2"); |
| const string16 kDB3 = ASCIIToUTF16("db3"); |
| const string16 kDescription = ASCIIToUTF16("database_description"); |
| |
| tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, |
| &database_size); |
| tracker->DatabaseOpened(kOrigin2, kDB2, kDescription, 0, |
| &database_size); |
| tracker->DatabaseOpened(kOrigin2, kDB3, kDescription, 0, |
| &database_size); |
| |
| EXPECT_TRUE(file_util::CreateDirectory(tracker->DatabaseDirectory().Append( |
| FilePath::FromWStringHack(UTF16ToWide( |
| tracker->GetOriginDirectory(kOrigin1)))))); |
| EXPECT_TRUE(file_util::CreateDirectory(tracker->DatabaseDirectory().Append( |
| FilePath::FromWStringHack(UTF16ToWide( |
| tracker->GetOriginDirectory(kOrigin2)))))); |
| EXPECT_EQ(1, file_util::WriteFile( |
| tracker->GetFullDBFilePath(kOrigin1, kDB1), "a", 1)); |
| EXPECT_EQ(2, file_util::WriteFile( |
| tracker->GetFullDBFilePath(kOrigin2, kDB2), "aa", 2)); |
| EXPECT_EQ(3, file_util::WriteFile( |
| tracker->GetFullDBFilePath(kOrigin2, kDB3), "aaa", 3)); |
| tracker->DatabaseModified(kOrigin1, kDB1); |
| tracker->DatabaseModified(kOrigin2, kDB2); |
| tracker->DatabaseModified(kOrigin2, kDB3); |
| |
| // Delete db1. Should also delete origin1. |
| TestObserver observer; |
| tracker->AddObserver(&observer); |
| TestCompletionCallback callback; |
| int result = tracker->DeleteDatabase(kOrigin1, kDB1, &callback); |
| EXPECT_EQ(net::ERR_IO_PENDING, result); |
| ASSERT_FALSE(callback.have_result()); |
| EXPECT_TRUE(observer.DidReceiveNewNotification()); |
| EXPECT_EQ(kOrigin1, observer.GetNotificationOriginIdentifier()); |
| EXPECT_EQ(kDB1, observer.GetNotificationDatabaseName()); |
| tracker->DatabaseClosed(kOrigin1, kDB1); |
| result = callback.GetResult(result); |
| EXPECT_EQ(net::OK, result); |
| EXPECT_FALSE(file_util::PathExists(tracker->DatabaseDirectory().Append( |
| FilePath::FromWStringHack(UTF16ToWide(kOrigin1))))); |
| |
| // Recreate db1. |
| tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, |
| &database_size); |
| EXPECT_TRUE(file_util::CreateDirectory(tracker->DatabaseDirectory().Append( |
| FilePath::FromWStringHack(UTF16ToWide( |
| tracker->GetOriginDirectory(kOrigin1)))))); |
| EXPECT_EQ(1, file_util::WriteFile( |
| tracker->GetFullDBFilePath(kOrigin1, kDB1), "a", 1)); |
| tracker->DatabaseModified(kOrigin1, kDB1); |
| |
| // Setup file modification times. db1 and db2 are modified now, db3 three |
| // days ago. |
| EXPECT_TRUE(file_util::SetLastModifiedTime( |
| tracker->GetFullDBFilePath(kOrigin1, kDB1), base::Time::Now())); |
| EXPECT_TRUE(file_util::SetLastModifiedTime( |
| tracker->GetFullDBFilePath(kOrigin2, kDB2), base::Time::Now())); |
| base::Time three_days_ago = base::Time::Now(); |
| three_days_ago -= base::TimeDelta::FromDays(3); |
| EXPECT_TRUE(file_util::SetLastModifiedTime( |
| tracker->GetFullDBFilePath(kOrigin2, kDB3), three_days_ago)); |
| |
| // Delete databases modified since yesterday. db2 is whitelisted. |
| base::Time yesterday = base::Time::Now(); |
| yesterday -= base::TimeDelta::FromDays(1); |
| result = tracker->DeleteDataModifiedSince( |
| yesterday, &callback); |
| EXPECT_EQ(net::ERR_IO_PENDING, result); |
| ASSERT_FALSE(callback.have_result()); |
| EXPECT_TRUE(observer.DidReceiveNewNotification()); |
| tracker->DatabaseClosed(kOrigin1, kDB1); |
| tracker->DatabaseClosed(kOrigin2, kDB2); |
| result = callback.GetResult(result); |
| EXPECT_EQ(net::OK, result); |
| EXPECT_FALSE(file_util::PathExists(tracker->DatabaseDirectory().Append( |
| FilePath::FromWStringHack(UTF16ToWide(kOrigin1))))); |
| EXPECT_TRUE( |
| file_util::PathExists(tracker->GetFullDBFilePath(kOrigin2, kDB2))); |
| EXPECT_TRUE( |
| file_util::PathExists(tracker->GetFullDBFilePath(kOrigin2, kDB3))); |
| |
| tracker->DatabaseClosed(kOrigin2, kDB3); |
| tracker->RemoveObserver(&observer); |
| } |
| |
| static void TestDatabaseTracker(bool incognito_mode) { |
| // Initialize the tracker database. |
| ScopedTempDir temp_dir; |
| ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| scoped_refptr<quota::MockSpecialStoragePolicy> special_storage_policy = |
| new quota::MockSpecialStoragePolicy; |
| special_storage_policy->AddProtected(GURL(kOrigin2Url)); |
| scoped_refptr<DatabaseTracker> tracker( |
| new DatabaseTracker(temp_dir.path(), incognito_mode, false, |
| special_storage_policy, NULL, NULL)); |
| |
| // Add two observers. |
| TestObserver observer1; |
| TestObserver observer2; |
| tracker->AddObserver(&observer1); |
| tracker->AddObserver(&observer2); |
| |
| // Open three new databases. |
| int64 database_size = 0; |
| const string16 kOrigin1 = |
| DatabaseUtil::GetOriginIdentifier(GURL(kOrigin1Url)); |
| const string16 kOrigin2 = |
| DatabaseUtil::GetOriginIdentifier(GURL(kOrigin2Url)); |
| const string16 kDB1 = ASCIIToUTF16("db1"); |
| const string16 kDB2 = ASCIIToUTF16("db2"); |
| const string16 kDB3 = ASCIIToUTF16("db3"); |
| const string16 kDescription = ASCIIToUTF16("database_description"); |
| |
| // Get the info for kOrigin1 and kOrigin2 |
| DatabaseTracker::CachedOriginInfo* origin1_info = |
| tracker->GetCachedOriginInfo(kOrigin1); |
| DatabaseTracker::CachedOriginInfo* origin2_info = |
| tracker->GetCachedOriginInfo(kOrigin1); |
| EXPECT_TRUE(origin1_info); |
| EXPECT_TRUE(origin2_info); |
| |
| |
| tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, |
| &database_size); |
| EXPECT_EQ(0, database_size); |
| tracker->DatabaseOpened(kOrigin2, kDB2, kDescription, 0, |
| &database_size); |
| EXPECT_EQ(0, database_size); |
| tracker->DatabaseOpened(kOrigin1, kDB3, kDescription, 0, |
| &database_size); |
| EXPECT_EQ(0, database_size); |
| |
| // Write some data to each file and check that the listeners are |
| // called with the appropriate values. |
| EXPECT_TRUE(file_util::CreateDirectory(tracker->DatabaseDirectory().Append( |
| FilePath::FromWStringHack(UTF16ToWide( |
| tracker->GetOriginDirectory(kOrigin1)))))); |
| EXPECT_TRUE(file_util::CreateDirectory(tracker->DatabaseDirectory().Append( |
| FilePath::FromWStringHack(UTF16ToWide( |
| tracker->GetOriginDirectory(kOrigin2)))))); |
| EXPECT_EQ(1, file_util::WriteFile( |
| tracker->GetFullDBFilePath(kOrigin1, kDB1), "a", 1)); |
| EXPECT_EQ(2, file_util::WriteFile( |
| tracker->GetFullDBFilePath(kOrigin2, kDB2), "aa", 2)); |
| EXPECT_EQ(4, file_util::WriteFile( |
| tracker->GetFullDBFilePath(kOrigin1, kDB3), "aaaa", 4)); |
| tracker->DatabaseModified(kOrigin1, kDB1); |
| CheckNotificationReceived(&observer1, kOrigin1, kDB1, 1); |
| CheckNotificationReceived(&observer2, kOrigin1, kDB1, 1); |
| tracker->DatabaseModified(kOrigin2, kDB2); |
| CheckNotificationReceived(&observer1, kOrigin2, kDB2, 2); |
| CheckNotificationReceived(&observer2, kOrigin2, kDB2, 2); |
| tracker->DatabaseModified(kOrigin1, kDB3); |
| CheckNotificationReceived(&observer1, kOrigin1, kDB3, 4); |
| CheckNotificationReceived(&observer2, kOrigin1, kDB3, 4); |
| |
| // Close all databases |
| tracker->DatabaseClosed(kOrigin1, kDB1); |
| tracker->DatabaseClosed(kOrigin2, kDB2); |
| tracker->DatabaseClosed(kOrigin1, kDB3); |
| |
| // Open an existing database and check the reported size |
| tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, |
| &database_size); |
| EXPECT_EQ(1, database_size); |
| tracker->DatabaseClosed(kOrigin1, kDB1); |
| |
| // Remove an observer; this should clear all caches. |
| tracker->RemoveObserver(&observer2); |
| |
| // Close the tracker database and clear all caches. |
| // Then make sure that DatabaseOpened() still returns the correct result. |
| tracker->CloseTrackerDatabaseAndClearCaches(); |
| tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, |
| &database_size); |
| EXPECT_EQ(1, database_size); |
| tracker->DatabaseClosed(kOrigin1, kDB1); |
| |
| // Remove all observers. |
| tracker->RemoveObserver(&observer1); |
| |
| // Trying to delete a database in use should fail |
| tracker->DatabaseOpened(kOrigin1, kDB3, kDescription, 0, |
| &database_size); |
| EXPECT_FALSE(tracker->DeleteClosedDatabase(kOrigin1, kDB3)); |
| origin1_info = tracker->GetCachedOriginInfo(kOrigin1); |
| EXPECT_TRUE(origin1_info); |
| EXPECT_EQ(4, origin1_info->GetDatabaseSize(kDB3)); |
| tracker->DatabaseClosed(kOrigin1, kDB3); |
| |
| // Delete a database and make sure the space used by that origin is updated |
| EXPECT_TRUE(tracker->DeleteClosedDatabase(kOrigin1, kDB3)); |
| origin1_info = tracker->GetCachedOriginInfo(kOrigin1); |
| EXPECT_TRUE(origin1_info); |
| EXPECT_EQ(1, origin1_info->GetDatabaseSize(kDB1)); |
| EXPECT_EQ(0, origin1_info->GetDatabaseSize(kDB3)); |
| |
| // Get all data for all origins |
| std::vector<OriginInfo> origins_info; |
| EXPECT_TRUE(tracker->GetAllOriginsInfo(&origins_info)); |
| EXPECT_EQ(size_t(2), origins_info.size()); |
| EXPECT_EQ(kOrigin1, origins_info[0].GetOrigin()); |
| EXPECT_EQ(1, origins_info[0].TotalSize()); |
| EXPECT_EQ(1, origins_info[0].GetDatabaseSize(kDB1)); |
| EXPECT_EQ(0, origins_info[0].GetDatabaseSize(kDB3)); |
| |
| EXPECT_EQ(kOrigin2, origins_info[1].GetOrigin()); |
| EXPECT_EQ(2, origins_info[1].TotalSize()); |
| |
| // Trying to delete an origin with databases in use should fail |
| tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, |
| &database_size); |
| EXPECT_FALSE(tracker->DeleteOrigin(kOrigin1, false)); |
| origin1_info = tracker->GetCachedOriginInfo(kOrigin1); |
| EXPECT_TRUE(origin1_info); |
| EXPECT_EQ(1, origin1_info->GetDatabaseSize(kDB1)); |
| tracker->DatabaseClosed(kOrigin1, kDB1); |
| |
| // Delete an origin that doesn't have any database in use |
| EXPECT_TRUE(tracker->DeleteOrigin(kOrigin1, false)); |
| origins_info.clear(); |
| EXPECT_TRUE(tracker->GetAllOriginsInfo(&origins_info)); |
| EXPECT_EQ(size_t(1), origins_info.size()); |
| EXPECT_EQ(kOrigin2, origins_info[0].GetOrigin()); |
| |
| origin1_info = tracker->GetCachedOriginInfo(kOrigin1); |
| EXPECT_TRUE(origin1_info); |
| EXPECT_EQ(0, origin1_info->TotalSize()); |
| } |
| |
| static void DatabaseTrackerQuotaIntegration() { |
| const GURL kOrigin(kOrigin1Url); |
| const string16 kOriginId = DatabaseUtil::GetOriginIdentifier(kOrigin); |
| const string16 kName = ASCIIToUTF16("name"); |
| const string16 kDescription = ASCIIToUTF16("description"); |
| |
| ScopedTempDir temp_dir; |
| ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| |
| // Initialize the tracker with a QuotaManagerProxy |
| scoped_refptr<TestQuotaManagerProxy> test_quota_proxy( |
| new TestQuotaManagerProxy); |
| scoped_refptr<DatabaseTracker> tracker( |
| new DatabaseTracker(temp_dir.path(), false /* incognito */, false, |
| NULL, test_quota_proxy, NULL)); |
| EXPECT_TRUE(test_quota_proxy->registered_client_); |
| |
| // Create a database and modify it a couple of times, close it, |
| // then delete it. Observe the tracker notifies accordingly. |
| |
| int64 database_size = 0; |
| tracker->DatabaseOpened(kOriginId, kName, kDescription, 0, |
| &database_size); |
| EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); |
| test_quota_proxy->reset(); |
| |
| FilePath db_file(tracker->GetFullDBFilePath(kOriginId, kName)); |
| EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); |
| EXPECT_TRUE(EnsureFileOfSize(db_file, 10)); |
| tracker->DatabaseModified(kOriginId, kName); |
| EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, 10)); |
| test_quota_proxy->reset(); |
| |
| EXPECT_TRUE(EnsureFileOfSize(db_file, 100)); |
| tracker->DatabaseModified(kOriginId, kName); |
| EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, 90)); |
| test_quota_proxy->reset(); |
| |
| tracker->DatabaseClosed(kOriginId, kName); |
| EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); |
| EXPECT_EQ(net::OK, tracker->DeleteDatabase(kOriginId, kName, NULL)); |
| EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, -100)); |
| test_quota_proxy->reset(); |
| |
| // Create a database and modify it, try to delete it while open, |
| // then close it (at which time deletion will actually occur). |
| // Observe the tracker notifies accordingly. |
| |
| tracker->DatabaseOpened(kOriginId, kName, kDescription, 0, |
| &database_size); |
| EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); |
| test_quota_proxy->reset(); |
| |
| db_file = tracker->GetFullDBFilePath(kOriginId, kName); |
| EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); |
| EXPECT_TRUE(EnsureFileOfSize(db_file, 100)); |
| tracker->DatabaseModified(kOriginId, kName); |
| EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, 100)); |
| test_quota_proxy->reset(); |
| |
| EXPECT_EQ(net::ERR_IO_PENDING, |
| tracker->DeleteDatabase(kOriginId, kName, NULL)); |
| EXPECT_FALSE(test_quota_proxy->WasModificationNotified(kOrigin, -100)); |
| |
| tracker->DatabaseClosed(kOriginId, kName); |
| EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); |
| EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, -100)); |
| test_quota_proxy->reset(); |
| |
| // Create a database and up the file size without telling |
| // the tracker about the modification, than simulate a |
| // a renderer crash. |
| // Observe the tracker notifies accordingly. |
| |
| tracker->DatabaseOpened(kOriginId, kName, kDescription, 0, |
| &database_size); |
| EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); |
| test_quota_proxy->reset(); |
| db_file = tracker->GetFullDBFilePath(kOriginId, kName); |
| EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); |
| EXPECT_TRUE(EnsureFileOfSize(db_file, 100)); |
| DatabaseConnections crashed_renderer_connections; |
| crashed_renderer_connections.AddConnection(kOriginId, kName); |
| EXPECT_FALSE(test_quota_proxy->WasModificationNotified(kOrigin, 100)); |
| tracker->CloseDatabases(crashed_renderer_connections); |
| EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, 100)); |
| |
| // Cleanup. |
| crashed_renderer_connections.RemoveAllConnections(); |
| test_quota_proxy->SimulateQuotaManagerDestroyed(); |
| } |
| |
| static void DatabaseTrackerClearLocalStateOnExit() { |
| int64 database_size = 0; |
| const string16 kOrigin1 = |
| DatabaseUtil::GetOriginIdentifier(GURL(kOrigin1Url)); |
| const string16 kOrigin2 = |
| DatabaseUtil::GetOriginIdentifier(GURL(kOrigin2Url)); |
| const string16 kDB1 = ASCIIToUTF16("db1"); |
| const string16 kDB2 = ASCIIToUTF16("db2"); |
| const string16 kDB3 = ASCIIToUTF16("db3"); |
| const string16 kDescription = ASCIIToUTF16("database_description"); |
| |
| // Initialize the tracker database. |
| ScopedTempDir temp_dir; |
| ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
| FilePath origin1_db_dir; |
| { |
| scoped_refptr<quota::MockSpecialStoragePolicy> special_storage_policy = |
| new quota::MockSpecialStoragePolicy; |
| special_storage_policy->AddProtected(GURL(kOrigin2Url)); |
| scoped_refptr<DatabaseTracker> tracker( |
| new DatabaseTracker( |
| temp_dir.path(), false, true, |
| special_storage_policy, NULL, |
| base::MessageLoopProxy::current())); |
| |
| // Open three new databases. |
| tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, |
| &database_size); |
| EXPECT_EQ(0, database_size); |
| tracker->DatabaseOpened(kOrigin2, kDB2, kDescription, 0, |
| &database_size); |
| EXPECT_EQ(0, database_size); |
| tracker->DatabaseOpened(kOrigin1, kDB3, kDescription, 0, |
| &database_size); |
| EXPECT_EQ(0, database_size); |
| |
| // Write some data to each file. |
| FilePath db_file; |
| db_file = tracker->GetFullDBFilePath(kOrigin1, kDB1); |
| EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); |
| EXPECT_TRUE(EnsureFileOfSize(db_file, 1)); |
| |
| db_file = tracker->GetFullDBFilePath(kOrigin2, kDB2); |
| EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); |
| EXPECT_TRUE(EnsureFileOfSize(db_file, 2)); |
| |
| db_file = tracker->GetFullDBFilePath(kOrigin1, kDB3); |
| EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); |
| EXPECT_TRUE(EnsureFileOfSize(db_file, 3)); |
| |
| // Store the origin database directory as long as it still exists. |
| origin1_db_dir = tracker->GetFullDBFilePath(kOrigin1, kDB3).DirName(); |
| |
| tracker->DatabaseModified(kOrigin1, kDB1); |
| tracker->DatabaseModified(kOrigin2, kDB2); |
| tracker->DatabaseModified(kOrigin1, kDB3); |
| |
| // Close all databases but one database. |
| tracker->DatabaseClosed(kOrigin1, kDB1); |
| tracker->DatabaseClosed(kOrigin2, kDB2); |
| |
| // Keep an open file handle to the last database. |
| base::PlatformFile file_handle = base::CreatePlatformFile( |
| tracker->GetFullDBFilePath(kOrigin1, kDB3), |
| base::PLATFORM_FILE_READ | |
| base::PLATFORM_FILE_WRITE | |
| base::PLATFORM_FILE_EXCLUSIVE_READ | |
| base::PLATFORM_FILE_EXCLUSIVE_WRITE | |
| base::PLATFORM_FILE_OPEN_ALWAYS | |
| base::PLATFORM_FILE_SHARE_DELETE, |
| NULL, NULL); |
| |
| tracker->Shutdown(); |
| |
| base::ClosePlatformFile(file_handle); |
| tracker->DatabaseClosed(kOrigin1, kDB3); |
| } |
| |
| // At this point, the database tracker should be gone. Create a new one. |
| scoped_refptr<quota::MockSpecialStoragePolicy> special_storage_policy = |
| new quota::MockSpecialStoragePolicy; |
| special_storage_policy->AddProtected(GURL(kOrigin2Url)); |
| scoped_refptr<DatabaseTracker> tracker( |
| new DatabaseTracker(temp_dir.path(), false, false, |
| special_storage_policy, NULL, NULL)); |
| |
| // Get all data for all origins. |
| std::vector<OriginInfo> origins_info; |
| EXPECT_TRUE(tracker->GetAllOriginsInfo(&origins_info)); |
| EXPECT_EQ(size_t(1), origins_info.size()); |
| EXPECT_EQ(kOrigin2, origins_info[0].GetOrigin()); |
| EXPECT_EQ(FilePath(), tracker->GetFullDBFilePath(kOrigin1, kDB1)); |
| EXPECT_TRUE( |
| file_util::PathExists(tracker->GetFullDBFilePath(kOrigin2, kDB2))); |
| EXPECT_EQ(FilePath(), tracker->GetFullDBFilePath(kOrigin1, kDB3)); |
| |
| // The origin directory should be gone as well. |
| EXPECT_FALSE(file_util::PathExists(origin1_db_dir)); |
| } |
| }; |
| |
| TEST(DatabaseTrackerTest, DeleteOpenDatabase) { |
| DatabaseTracker_TestHelper_Test::TestDeleteOpenDatabase(false); |
| } |
| |
| TEST(DatabaseTrackerTest, DeleteOpenDatabaseIncognitoMode) { |
| DatabaseTracker_TestHelper_Test::TestDeleteOpenDatabase(true); |
| } |
| |
| TEST(DatabaseTrackerTest, DatabaseTracker) { |
| DatabaseTracker_TestHelper_Test::TestDatabaseTracker(false); |
| } |
| |
| TEST(DatabaseTrackerTest, DatabaseTrackerIncognitoMode) { |
| DatabaseTracker_TestHelper_Test::TestDatabaseTracker(true); |
| } |
| |
| TEST(DatabaseTrackerTest, DatabaseTrackerQuotaIntegration) { |
| // There is no difference in behavior between incognito and not. |
| DatabaseTracker_TestHelper_Test::DatabaseTrackerQuotaIntegration(); |
| } |
| |
| TEST(DatabaseTrackerTest, DatabaseTrackerClearLocalStateOnExit) { |
| // Only works for regular mode. |
| DatabaseTracker_TestHelper_Test::DatabaseTrackerClearLocalStateOnExit(); |
| } |
| |
| } // namespace webkit_database |