database_tracker_unittest.cc revision 424c4d7b64af9d0d8fd9624f381f469654d5e3d2
15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2012 The Chromium Authors. All rights reserved. 25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be 35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file. 45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/file_util.h" 62a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/file_path.h" 72a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)#include "base/files/scoped_temp_dir.h" 85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/memory/scoped_ptr.h" 9ca12bfac764ba476d6cd062bf1dde12cc64c3f40Ben Murdoch#include "base/message_loop/message_loop.h" 10868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/message_loop/message_loop_proxy.h" 115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/platform_file.h" 12868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/utf_string_conversions.h" 13eb525c5499e34cc9c4b825d6d9e75bb07cc06aceBen Murdoch#include "base/time/time.h" 145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/net_errors.h" 155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "net/base/test_completion_callback.h" 165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "testing/gtest/include/gtest/gtest.h" 175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "third_party/sqlite/sqlite3.h" 1890dce4d38c5ff5333bea97d859d4e484e27edf0cTorne (Richard Coles)#include "webkit/browser/database/database_tracker.h" 19868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "webkit/browser/quota/mock_special_storage_policy.h" 20868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "webkit/browser/quota/quota_manager.h" 217d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles)#include "webkit/common/database/database_identifier.h" 225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace { 245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kOrigin1Url[] = "http://origin1"; 265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)const char kOrigin2Url[] = "http://protected_origin2"; 275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class TestObserver : public webkit_database::DatabaseTracker::Observer { 295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestObserver() 315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : new_notification_received_(false), 325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) observe_size_changes_(true), 335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) observe_scheduled_deletions_(true) { 345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestObserver(bool observe_size_changes, bool observe_scheduled_deletions) 365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : new_notification_received_(false), 375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) observe_size_changes_(observe_size_changes), 385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) observe_scheduled_deletions_(observe_scheduled_deletions) { 395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~TestObserver() {} 427d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) virtual void OnDatabaseSizeChanged(const std::string& origin_identifier, 43c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16& database_name, 442a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int64 database_size) OVERRIDE { 455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!observe_size_changes_) 465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new_notification_received_ = true; 485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin_identifier_ = origin_identifier; 495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) database_name_ = database_name; 505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) database_size_ = database_size; 515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 522a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void OnDatabaseScheduledForDeletion( 537d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string& origin_identifier, 54c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16& database_name) OVERRIDE { 555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!observe_scheduled_deletions_) 565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return; 575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new_notification_received_ = true; 585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin_identifier_ = origin_identifier; 595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) database_name_ = database_name; 605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool DidReceiveNewNotification() { 625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool temp_new_notification_received = new_notification_received_; 635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new_notification_received_ = false; 645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return temp_new_notification_received; 655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 667d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) std::string GetNotificationOriginIdentifier() { 67c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) return origin_identifier_; 68c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) } 69c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::string16 GetNotificationDatabaseName() { return database_name_; } 705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 GetNotificationDatabaseSize() { return database_size_; } 715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) private: 735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool new_notification_received_; 745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool observe_size_changes_; 755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool observe_scheduled_deletions_; 767d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) std::string origin_identifier_; 77c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) base::string16 database_name_; 785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 database_size_; 795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)void CheckNotificationReceived(TestObserver* observer, 827d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string& expected_origin_identifier, 83c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16& expected_database_name, 845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 expected_database_size) { 855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(observer->DidReceiveNewNotification()); 865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(expected_origin_identifier, 875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) observer->GetNotificationOriginIdentifier()); 885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(expected_database_name, 895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) observer->GetNotificationDatabaseName()); 905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(expected_database_size, 915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) observer->GetNotificationDatabaseSize()); 925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class TestQuotaManagerProxy : public quota::QuotaManagerProxy { 955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestQuotaManagerProxy() 975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) : QuotaManagerProxy(NULL, NULL), 985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) registered_client_(NULL) { 995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void RegisterClient(quota::QuotaClient* client) OVERRIDE { 1025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(registered_client_); 1035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) registered_client_ = client; 1045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void NotifyStorageAccessed(quota::QuotaClient::ID client_id, 1075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& origin, 1082a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) quota::StorageType type) OVERRIDE { 1095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(quota::QuotaClient::kDatabase, client_id); 1105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(quota::kStorageTypeTemporary, type); 1115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) accesses_[origin] += 1; 1125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual void NotifyStorageModified(quota::QuotaClient::ID client_id, 1155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL& origin, 1165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) quota::StorageType type, 1172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) int64 delta) OVERRIDE { 1185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(quota::QuotaClient::kDatabase, client_id); 1195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(quota::kStorageTypeTemporary, type); 1205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) modifications_[origin].first += 1; 1215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) modifications_[origin].second += delta; 1225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Not needed for our tests. 1252a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void NotifyOriginInUse(const GURL& origin) OVERRIDE {} 1262a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) virtual void NotifyOriginNoLongerInUse(const GURL& origin) OVERRIDE {} 1275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void SimulateQuotaManagerDestroyed() { 1295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (registered_client_) { 1305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) registered_client_->OnQuotaManagerDestroyed(); 1315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) registered_client_ = NULL; 1325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool WasAccessNotified(const GURL& origin) { 1365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return accesses_[origin] != 0; 1375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) bool WasModificationNotified(const GURL& origin, int64 amount) { 1405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return modifications_[origin].first != 0 && 1415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) modifications_[origin].second == amount; 1425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) void reset() { 1455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) accesses_.clear(); 1465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) modifications_.clear(); 1475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) quota::QuotaClient* registered_client_; 1505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Map from origin to count of access notifications. 1525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::map<GURL, int> accesses_; 1535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Map from origin to <count, sum of deltas> 1555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::map<GURL, std::pair<int, int64> > modifications_; 1565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) protected: 1585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) virtual ~TestQuotaManagerProxy() { 1595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(registered_client_); 1605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 1615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 1625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1642a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)bool EnsureFileOfSize(const base::FilePath& file_path, int64 length) { 1655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::PlatformFileError error_code(base::PLATFORM_FILE_ERROR_FAILED); 1665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::PlatformFile file = 1675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::CreatePlatformFile( 1685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) file_path, 1695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::PLATFORM_FILE_OPEN_ALWAYS | base::PLATFORM_FILE_WRITE, 1705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, 1715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &error_code); 1725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (error_code != base::PLATFORM_FILE_OK) 1735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return false; 1745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) if (!base::TruncatePlatformFile(file, length)) 1755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) error_code = base::PLATFORM_FILE_ERROR_FAILED; 1765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::ClosePlatformFile(file); 1775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) return error_code == base::PLATFORM_FILE_OK; 1785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 1795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace 1815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace webkit_database { 1835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 1845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// We declare a helper class, and make it a friend of DatabaseTracker using 1855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// the FRIEND_TEST() macro, and we implement all tests we want to run as 1865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static methods of this class. Then we make our TEST() targets call these 1875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// static functions. This allows us to run each test in normal mode and 1885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// incognito mode without writing the same code twice. 1895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)class DatabaseTracker_TestHelper_Test { 1905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) public: 1915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void TestDeleteOpenDatabase(bool incognito_mode) { 1925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize the tracker database. 1932a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir; 1945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 1955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<quota::MockSpecialStoragePolicy> special_storage_policy = 1965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new quota::MockSpecialStoragePolicy; 1975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) special_storage_policy->AddProtected(GURL(kOrigin2Url)); 1985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 199868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), 200868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) incognito_mode, 201868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) special_storage_policy.get(), 202868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) NULL, 203868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) NULL)); 2045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create and open three databases. 2065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 database_size = 0; 2077d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOrigin1 = 2087d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(GURL(kOrigin1Url)); 2097d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOrigin2 = 2107d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(GURL(kOrigin2Url)); 211c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB1 = ASCIIToUTF16("db1"); 212c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB2 = ASCIIToUTF16("db2"); 213c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB3 = ASCIIToUTF16("db3"); 214c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDescription = ASCIIToUTF16("database_description"); 2155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, 2175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 2185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin2, kDB2, kDescription, 0, 2195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 2205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin2, kDB3, kDescription, 0, 2215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 2225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 223424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory( 224424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->DatabaseDirectory().Append(base::FilePath::FromUTF16Unsafe( 225424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->GetOriginDirectory(kOrigin1))))); 226424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory( 227424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->DatabaseDirectory().Append(base::FilePath::FromUTF16Unsafe( 228424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->GetOriginDirectory(kOrigin2))))); 2295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, file_util::WriteFile( 2305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin1, kDB1), "a", 1)); 2315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(2, file_util::WriteFile( 2325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin2, kDB2), "aa", 2)); 2335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(3, file_util::WriteFile( 2345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin2, kDB3), "aaa", 3)); 2355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin1, kDB1); 2365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin2, kDB2); 2375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin2, kDB3); 2385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Delete db1. Should also delete origin1. 2405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestObserver observer; 2415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->AddObserver(&observer); 2425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::TestCompletionCallback callback; 2435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int result = tracker->DeleteDatabase(kOrigin1, kDB1, callback.callback()); 2445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(net::ERR_IO_PENDING, result); 2455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_FALSE(callback.have_result()); 2465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(observer.DidReceiveNewNotification()); 2475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(kOrigin1, observer.GetNotificationOriginIdentifier()); 2485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(kDB1, observer.GetNotificationDatabaseName()); 2495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB1); 2505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result = callback.GetResult(result); 2515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(net::OK, result); 2527dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::PathExists( 2537d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) tracker->DatabaseDirectory().AppendASCII(kOrigin1))); 2545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Recreate db1. 2565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, 2575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 258424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory( 259424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->DatabaseDirectory().Append(base::FilePath::FromUTF16Unsafe( 260424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->GetOriginDirectory(kOrigin1))))); 2615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, file_util::WriteFile( 2625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin1, kDB1), "a", 1)); 2635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin1, kDB1); 2645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Setup file modification times. db1 and db2 are modified now, db3 three 2665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // days ago. 2675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::SetLastModifiedTime( 2685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin1, kDB1), base::Time::Now())); 2695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::SetLastModifiedTime( 2705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin2, kDB2), base::Time::Now())); 2715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time three_days_ago = base::Time::Now(); 2725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) three_days_ago -= base::TimeDelta::FromDays(3); 2735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::SetLastModifiedTime( 2745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin2, kDB3), three_days_ago)); 2755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Delete databases modified since yesterday. db2 is whitelisted. 2775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) base::Time yesterday = base::Time::Now(); 2785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) yesterday -= base::TimeDelta::FromDays(1); 2795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result = tracker->DeleteDataModifiedSince( 2805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) yesterday, callback.callback()); 2815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(net::ERR_IO_PENDING, result); 2825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_FALSE(callback.have_result()); 2835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(observer.DidReceiveNewNotification()); 2845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB1); 2855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin2, kDB2); 2865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) result = callback.GetResult(result); 2875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(net::OK, result); 2887dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::PathExists( 2897d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) tracker->DatabaseDirectory().AppendASCII(kOrigin1))); 2905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE( 2917dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::PathExists(tracker->GetFullDBFilePath(kOrigin2, kDB2))); 2925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE( 2937dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::PathExists(tracker->GetFullDBFilePath(kOrigin2, kDB3))); 2945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin2, kDB3); 2965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->RemoveObserver(&observer); 2975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 2985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 2995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void TestDatabaseTracker(bool incognito_mode) { 3005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize the tracker database. 3012a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir; 3025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 3035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<quota::MockSpecialStoragePolicy> special_storage_policy = 3045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new quota::MockSpecialStoragePolicy; 3055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) special_storage_policy->AddProtected(GURL(kOrigin2Url)); 3065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 307868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), 308868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) incognito_mode, 309868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) special_storage_policy.get(), 310868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) NULL, 311868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) NULL)); 3125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Add two observers. 3145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestObserver observer1; 3155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestObserver observer2; 3165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->AddObserver(&observer1); 3175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->AddObserver(&observer2); 3185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Open three new databases. 3205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 database_size = 0; 3217d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOrigin1 = 3227d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(GURL(kOrigin1Url)); 3237d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOrigin2 = 3247d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(GURL(kOrigin2Url)); 325c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB1 = ASCIIToUTF16("db1"); 326c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB2 = ASCIIToUTF16("db2"); 327c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB3 = ASCIIToUTF16("db3"); 328c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDescription = ASCIIToUTF16("database_description"); 3295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get the info for kOrigin1 and kOrigin2 3315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker::CachedOriginInfo* origin1_info = 3325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetCachedOriginInfo(kOrigin1); 3335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker::CachedOriginInfo* origin2_info = 3345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetCachedOriginInfo(kOrigin1); 3355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(origin1_info); 3365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(origin2_info); 3375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, 3405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 3415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, database_size); 3425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin2, kDB2, kDescription, 0, 3435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 3445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, database_size); 3455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB3, kDescription, 0, 3465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 3475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, database_size); 3485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Write some data to each file and check that the listeners are 3505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // called with the appropriate values. 351424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory( 352424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->DatabaseDirectory().Append(base::FilePath::FromUTF16Unsafe( 353424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->GetOriginDirectory(kOrigin1))))); 354424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory( 355424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->DatabaseDirectory().Append(base::FilePath::FromUTF16Unsafe( 356424c4d7b64af9d0d8fd9624f381f469654d5e3d2Torne (Richard Coles) tracker->GetOriginDirectory(kOrigin2))))); 3575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, file_util::WriteFile( 3585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin1, kDB1), "a", 1)); 3595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(2, file_util::WriteFile( 3605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin2, kDB2), "aa", 2)); 3615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(4, file_util::WriteFile( 3625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->GetFullDBFilePath(kOrigin1, kDB3), "aaaa", 4)); 3635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin1, kDB1); 3645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CheckNotificationReceived(&observer1, kOrigin1, kDB1, 1); 3655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CheckNotificationReceived(&observer2, kOrigin1, kDB1, 1); 3665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin2, kDB2); 3675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CheckNotificationReceived(&observer1, kOrigin2, kDB2, 2); 3685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CheckNotificationReceived(&observer2, kOrigin2, kDB2, 2); 3695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin1, kDB3); 3705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CheckNotificationReceived(&observer1, kOrigin1, kDB3, 4); 3715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) CheckNotificationReceived(&observer2, kOrigin1, kDB3, 4); 3725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Close all databases 3745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB1); 3755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin2, kDB2); 3765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB3); 3775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Open an existing database and check the reported size 3795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, 3805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 3815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, database_size); 3825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB1); 3835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Remove an observer; this should clear all caches. 3855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->RemoveObserver(&observer2); 3865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Close the tracker database and clear all caches. 3885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Then make sure that DatabaseOpened() still returns the correct result. 3895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->CloseTrackerDatabaseAndClearCaches(); 3905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, 3915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 3925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, database_size); 3935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB1); 3945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Remove all observers. 3965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->RemoveObserver(&observer1); 3975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 3985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Trying to delete a database in use should fail 3995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB3, kDescription, 0, 4005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 4015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(tracker->DeleteClosedDatabase(kOrigin1, kDB3)); 4025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin1_info = tracker->GetCachedOriginInfo(kOrigin1); 4035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(origin1_info); 4045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(4, origin1_info->GetDatabaseSize(kDB3)); 4055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB3); 4065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Delete a database and make sure the space used by that origin is updated 4085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->DeleteClosedDatabase(kOrigin1, kDB3)); 4095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin1_info = tracker->GetCachedOriginInfo(kOrigin1); 4105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(origin1_info); 4115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, origin1_info->GetDatabaseSize(kDB1)); 4125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, origin1_info->GetDatabaseSize(kDB3)); 4135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get all data for all origins 4155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<OriginInfo> origins_info; 4165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetAllOriginsInfo(&origins_info)); 4175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(size_t(2), origins_info.size()); 4187d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) EXPECT_EQ(kOrigin1, origins_info[0].GetOriginIdentifier()); 4195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, origins_info[0].TotalSize()); 4205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, origins_info[0].GetDatabaseSize(kDB1)); 4215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, origins_info[0].GetDatabaseSize(kDB3)); 4225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4237d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) EXPECT_EQ(kOrigin2, origins_info[1].GetOriginIdentifier()); 4245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(2, origins_info[1].TotalSize()); 4255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Trying to delete an origin with databases in use should fail 4275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, 4285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 4295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(tracker->DeleteOrigin(kOrigin1, false)); 4305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin1_info = tracker->GetCachedOriginInfo(kOrigin1); 4315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(origin1_info); 4325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1, origin1_info->GetDatabaseSize(kDB1)); 4335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB1); 4345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Delete an origin that doesn't have any database in use 4365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->DeleteOrigin(kOrigin1, false)); 4375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origins_info.clear(); 4385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetAllOriginsInfo(&origins_info)); 4395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(size_t(1), origins_info.size()); 4407d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) EXPECT_EQ(kOrigin2, origins_info[0].GetOriginIdentifier()); 4415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin1_info = tracker->GetCachedOriginInfo(kOrigin1); 4435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(origin1_info); 4445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, origin1_info->TotalSize()); 4455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 4465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void DatabaseTrackerQuotaIntegration() { 4485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL kOrigin(kOrigin1Url); 4497d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOriginId = 4507d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(kOrigin); 451c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kName = ASCIIToUTF16("name"); 452c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDescription = ASCIIToUTF16("description"); 4535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4542a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir; 4555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 4565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize the tracker with a QuotaManagerProxy 4585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<TestQuotaManagerProxy> test_quota_proxy( 4595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new TestQuotaManagerProxy); 4605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 461868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), 462868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) false /* incognito */, 463868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) NULL, 464868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) test_quota_proxy.get(), 465868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) NULL)); 4665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->registered_client_); 4675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create a database and modify it a couple of times, close it, 4695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // then delete it. Observe the tracker notifies accordingly. 4705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 database_size = 0; 4725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOriginId, kName, kDescription, 0, 4735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 4745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); 4755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->reset(); 4765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4772a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath db_file(tracker->GetFullDBFilePath(kOriginId, kName)); 4785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); 4795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(db_file, 10)); 4805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOriginId, kName); 4815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, 10)); 4825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->reset(); 4835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(db_file, 100)); 4855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOriginId, kName); 4865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, 90)); 4875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->reset(); 4885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOriginId, kName); 4905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); 4915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(net::OK, tracker->DeleteDatabase( 4925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) kOriginId, kName, net::CompletionCallback())); 4935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, -100)); 4945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->reset(); 4955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 4965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create a database and modify it, try to delete it while open, 4975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // then close it (at which time deletion will actually occur). 4985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Observe the tracker notifies accordingly. 4995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOriginId, kName, kDescription, 0, 5015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 5025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); 5035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->reset(); 5045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db_file = tracker->GetFullDBFilePath(kOriginId, kName); 5065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); 5075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(db_file, 100)); 5085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOriginId, kName); 5095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, 100)); 5105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->reset(); 5115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(net::ERR_IO_PENDING, 5135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DeleteDatabase(kOriginId, kName, 5145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::CompletionCallback())); 5155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(test_quota_proxy->WasModificationNotified(kOrigin, -100)); 5165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOriginId, kName); 5185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); 5195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, -100)); 5205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->reset(); 5215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create a database and up the file size without telling 5235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // the tracker about the modification, than simulate a 5245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a renderer crash. 5255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Observe the tracker notifies accordingly. 5265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOriginId, kName, kDescription, 0, 5285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 5295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasAccessNotified(kOrigin)); 5305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->reset(); 5315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db_file = tracker->GetFullDBFilePath(kOriginId, kName); 5325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); 5335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(db_file, 100)); 5345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseConnections crashed_renderer_connections; 5355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crashed_renderer_connections.AddConnection(kOriginId, kName); 5365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(test_quota_proxy->WasModificationNotified(kOrigin, 100)); 5375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->CloseDatabases(crashed_renderer_connections); 5385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(test_quota_proxy->WasModificationNotified(kOrigin, 100)); 5395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Cleanup. 5415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) crashed_renderer_connections.RemoveAllConnections(); 5425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) test_quota_proxy->SimulateQuotaManagerDestroyed(); 5435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 5445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void DatabaseTrackerClearSessionOnlyDatabasesOnExit() { 5465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 database_size = 0; 5477d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOrigin1 = 5487d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(GURL(kOrigin1Url)); 5497d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOrigin2 = 5507d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(GURL(kOrigin2Url)); 551c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB1 = ASCIIToUTF16("db1"); 552c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB2 = ASCIIToUTF16("db2"); 553c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDescription = ASCIIToUTF16("database_description"); 5545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize the tracker database. 556b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) base::MessageLoop message_loop; 5572a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir; 5585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 5592a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath origin1_db_dir; 5602a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath origin2_db_dir; 5615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 5625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<quota::MockSpecialStoragePolicy> special_storage_policy = 5635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new quota::MockSpecialStoragePolicy; 5645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) special_storage_policy->AddSessionOnly(GURL(kOrigin2Url)); 5655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 566868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), 567868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) false, 568868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) special_storage_policy.get(), 569868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) NULL, 5707d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::MessageLoopProxy::current().get())); 5715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Open two new databases. 5735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, 5745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 5755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, database_size); 5765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin2, kDB2, kDescription, 0, 5775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 5785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, database_size); 5795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Write some data to each file. 5812a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath db_file; 5825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db_file = tracker->GetFullDBFilePath(kOrigin1, kDB1); 5835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); 5845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(db_file, 1)); 5855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db_file = tracker->GetFullDBFilePath(kOrigin2, kDB2); 5875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); 5885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(db_file, 2)); 5895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Store the origin database directories as long as they still exist. 5915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin1_db_dir = tracker->GetFullDBFilePath(kOrigin1, kDB1).DirName(); 5925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin2_db_dir = tracker->GetFullDBFilePath(kOrigin2, kDB2).DirName(); 5935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin1, kDB1); 5955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin2, kDB2); 5965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 5975821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Close all databases. 5985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB1); 5995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin2, kDB2); 6005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->Shutdown(); 6025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // At this point, the database tracker should be gone. Create a new one. 6055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 6065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), false, NULL, NULL, NULL)); 6075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get all data for all origins. 6095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<OriginInfo> origins_info; 6105821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetAllOriginsInfo(&origins_info)); 6115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // kOrigin1 was not session-only, so it survived. kOrigin2 was session-only 6125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // and it got deleted. 6135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(size_t(1), origins_info.size()); 6147d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) EXPECT_EQ(kOrigin1, origins_info[0].GetOriginIdentifier()); 6155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE( 6167dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::PathExists(tracker->GetFullDBFilePath(kOrigin1, kDB1))); 6172a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) EXPECT_EQ(base::FilePath(), tracker->GetFullDBFilePath(kOrigin2, kDB2)); 6185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // The origin directory of kOrigin1 remains, but the origin directory of 6205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // kOrigin2 is deleted. 6217dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::PathExists(origin1_db_dir)); 6227dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::PathExists(origin2_db_dir)); 6235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void DatabaseTrackerSetForceKeepSessionState() { 6265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 database_size = 0; 6277d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOrigin1 = 6287d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(GURL(kOrigin1Url)); 6297d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOrigin2 = 6307d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(GURL(kOrigin2Url)); 631c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB1 = ASCIIToUTF16("db1"); 632c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDB2 = ASCIIToUTF16("db2"); 633c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDescription = ASCIIToUTF16("database_description"); 6345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize the tracker database. 636b2df76ea8fec9e32f6f3718986dba0d95315b29cTorne (Richard Coles) base::MessageLoop message_loop; 6372a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir; 6385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 6392a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath origin1_db_dir; 6402a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath origin2_db_dir; 6415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) { 6425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<quota::MockSpecialStoragePolicy> special_storage_policy = 6435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new quota::MockSpecialStoragePolicy; 6445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) special_storage_policy->AddSessionOnly(GURL(kOrigin2Url)); 6455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 646868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), 647868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) false, 648868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) special_storage_policy.get(), 649868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles) NULL, 6507d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) base::MessageLoopProxy::current().get())); 6515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->SetForceKeepSessionState(); 6525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Open two new databases. 6545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin1, kDB1, kDescription, 0, 6555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 6565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, database_size); 6575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOrigin2, kDB2, kDescription, 0, 6585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 6595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, database_size); 6605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Write some data to each file. 6622a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath db_file; 6635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db_file = tracker->GetFullDBFilePath(kOrigin1, kDB1); 6645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); 6655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(db_file, 1)); 6665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) db_file = tracker->GetFullDBFilePath(kOrigin2, kDB2); 6685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(db_file.DirName())); 6695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(db_file, 2)); 6705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Store the origin database directories as long as they still exist. 6725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin1_db_dir = tracker->GetFullDBFilePath(kOrigin1, kDB1).DirName(); 6735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) origin2_db_dir = tracker->GetFullDBFilePath(kOrigin2, kDB2).DirName(); 6745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin1, kDB1); 6765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOrigin2, kDB2); 6775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Close all databases. 6795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin1, kDB1); 6805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOrigin2, kDB2); 6815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6825821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->Shutdown(); 6835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 6845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // At this point, the database tracker should be gone. Create a new one. 6865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 6875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), false, NULL, NULL, NULL)); 6885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Get all data for all origins. 6905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<OriginInfo> origins_info; 6915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetAllOriginsInfo(&origins_info)); 6925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // No origins were deleted. 6935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(size_t(2), origins_info.size()); 6945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE( 6957dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::PathExists(tracker->GetFullDBFilePath(kOrigin1, kDB1))); 6965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE( 6977dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch base::PathExists(tracker->GetFullDBFilePath(kOrigin2, kDB2))); 6985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 6997dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::PathExists(origin1_db_dir)); 7007dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::PathExists(origin2_db_dir)); 7015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7025821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void EmptyDatabaseNameIsValid() { 7045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL kOrigin(kOrigin1Url); 7057d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOriginId = 7067d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(kOrigin); 707c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kEmptyName; 708c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDescription(ASCIIToUTF16("description")); 709c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kChangedDescription( 710c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) ASCIIToUTF16("changed_description")); 7115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize a tracker database, no need to put it on disk. 7135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const bool kUseInMemoryTrackerDatabase = true; 7142a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir; 7155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 7165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 7175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), kUseInMemoryTrackerDatabase, 7185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, NULL, NULL)); 7195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Starts off with no databases. 7215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) std::vector<OriginInfo> infos; 7225821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetAllOriginsInfo(&infos)); 7235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(infos.empty()); 7245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create a db with an empty name. 7265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 database_size = -1; 7275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOriginId, kEmptyName, kDescription, 0, 7285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 7295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(0, database_size); 7305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseModified(kOriginId, kEmptyName); 7315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetAllOriginsInfo(&infos)); 7325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1u, infos.size()); 7335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(kDescription, infos[0].GetDatabaseDescription(kEmptyName)); 7345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(tracker->GetFullDBFilePath(kOriginId, kEmptyName).empty()); 7355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOriginId, kEmptyName, kChangedDescription, 0, 7365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 7375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) infos.clear(); 7385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetAllOriginsInfo(&infos)); 7395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(1u, infos.size()); 7405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(kChangedDescription, infos[0].GetDatabaseDescription(kEmptyName)); 7415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOriginId, kEmptyName); 7425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOriginId, kEmptyName); 7435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Deleting it should return to the initial state. 7455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_EQ(net::OK, tracker->DeleteDatabase(kOriginId, kEmptyName, 7465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) net::CompletionCallback())); 7475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) infos.clear(); 7485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetAllOriginsInfo(&infos)); 7495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(infos.empty()); 7505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 7515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) static void HandleSqliteError() { 7535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const GURL kOrigin(kOrigin1Url); 7547d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) const std::string kOriginId = 7557d4cd473f85ac64c3747c96c277f9e506a0d2246Torne (Richard Coles) webkit_database::GetIdentifierFromOrigin(kOrigin); 756c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kName(ASCIIToUTF16("name")); 757c2e0dbddbe15c98d52c4786dac06cb8952a8ae6dTorne (Richard Coles) const base::string16 kDescription(ASCIIToUTF16("description")); 7585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Initialize a tracker database, no need to put it on disk. 7605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) const bool kUseInMemoryTrackerDatabase = true; 7612a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::ScopedTempDir temp_dir; 7625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); 7635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) scoped_refptr<DatabaseTracker> tracker( 7645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) new DatabaseTracker(temp_dir.path(), kUseInMemoryTrackerDatabase, 7655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) NULL, NULL, NULL)); 7665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Setup to observe OnScheduledForDelete notifications. 7685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) TestObserver observer(false, true); 7695821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->AddObserver(&observer); 7705821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7715821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify does no harm when there is no such database. 7725821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->HandleSqliteError(kOriginId, kName, SQLITE_CORRUPT); 7735821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(tracker->IsDatabaseScheduledForDeletion(kOriginId, kName)); 7745821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(observer.DidReceiveNewNotification()); 7755821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7765821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // -------------------------------------------------------- 7775821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create a record of a database in the tracker db and create 7785821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a spoof_db_file on disk in the expected location. 7795821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) int64 database_size = 0; 7805821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOriginId, kName, kDescription, 0, 7815821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 7822a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath spoof_db_file = tracker->GetFullDBFilePath(kOriginId, kName); 7835821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(tracker->GetFullDBFilePath(kOriginId, kName).empty()); 7845821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(spoof_db_file.DirName())); 7855821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(spoof_db_file, 1)); 7865821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7875821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify does no harm with a non-error is reported. 7885821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->HandleSqliteError(kOriginId, kName, SQLITE_OK); 7895821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(tracker->IsDatabaseScheduledForDeletion(kOriginId, kName)); 7905821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(observer.DidReceiveNewNotification()); 7915821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7925821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify that with a connection open, the db is scheduled for deletion, 7935821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // but that the file still exists. 7945821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->HandleSqliteError(kOriginId, kName, SQLITE_CORRUPT); 7955821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->IsDatabaseScheduledForDeletion(kOriginId, kName)); 7965821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(observer.DidReceiveNewNotification()); 7977dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_TRUE(base::PathExists(spoof_db_file)); 7985821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 7995821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify that once closed, the file is deleted and the record in the 8005821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // tracker db is removed. 8015821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOriginId, kName); 8027dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::PathExists(spoof_db_file)); 8035821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetFullDBFilePath(kOriginId, kName).empty()); 8045821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8055821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // -------------------------------------------------------- 8065821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Create another record of a database in the tracker db and create 8075821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // a spoof_db_file on disk in the expected location. 8085821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseOpened(kOriginId, kName, kDescription, 0, 8095821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) &database_size); 8102a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles) base::FilePath spoof_db_file2 = tracker->GetFullDBFilePath(kOriginId, kName); 8115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(tracker->GetFullDBFilePath(kOriginId, kName).empty()); 8125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_NE(spoof_db_file, spoof_db_file2); 8135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(file_util::CreateDirectory(spoof_db_file2.DirName())); 8145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(EnsureFileOfSize(spoof_db_file2, 1)); 8155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Verify that with no connection open, the db is deleted immediately. 8175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->DatabaseClosed(kOriginId, kName); 8185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->HandleSqliteError(kOriginId, kName, SQLITE_CORRUPT); 8195821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(tracker->IsDatabaseScheduledForDeletion(kOriginId, kName)); 8205821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_FALSE(observer.DidReceiveNewNotification()); 8215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) EXPECT_TRUE(tracker->GetFullDBFilePath(kOriginId, kName).empty()); 8227dbb3d5cf0c15f500944d211057644d6a2f37371Ben Murdoch EXPECT_FALSE(base::PathExists(spoof_db_file2)); 8235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) tracker->RemoveObserver(&observer); 8255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) } 8265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}; 8275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, DeleteOpenDatabase) { 8295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test::TestDeleteOpenDatabase(false); 8305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, DeleteOpenDatabaseIncognitoMode) { 8335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test::TestDeleteOpenDatabase(true); 8345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, DatabaseTracker) { 8375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test::TestDatabaseTracker(false); 8385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, DatabaseTrackerIncognitoMode) { 8415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test::TestDatabaseTracker(true); 8425821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8435821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8445821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, DatabaseTrackerQuotaIntegration) { 8455821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // There is no difference in behavior between incognito and not. 8465821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test::DatabaseTrackerQuotaIntegration(); 8475821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8485821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8495821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, DatabaseTrackerClearSessionOnlyDatabasesOnExit) { 8505821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only works for regular mode. 8515821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test:: 8525821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTrackerClearSessionOnlyDatabasesOnExit(); 8535821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8545821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8555821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, DatabaseTrackerSetForceKeepSessionState) { 8565821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) // Only works for regular mode. 8575821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test::DatabaseTrackerSetForceKeepSessionState(); 8585821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8595821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8605821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, EmptyDatabaseNameIsValid) { 8615821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test::EmptyDatabaseNameIsValid(); 8625821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8635821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8645821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)TEST(DatabaseTrackerTest, HandleSqliteError) { 8655821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) DatabaseTracker_TestHelper_Test::HandleSqliteError(); 8665821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} 8675821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles) 8685821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)} // namespace webkit_database 869