OLD | NEW |
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/file_util.h" | 5 #include "base/file_util.h" |
6 #include "base/files/scoped_temp_dir.h" | 6 #include "base/files/scoped_temp_dir.h" |
7 #include "base/test/test_simple_task_runner.h" | 7 #include "base/test/test_simple_task_runner.h" |
8 #include "base/threading/thread.h" | 8 #include "base/threading/thread.h" |
9 #include "content/browser/browser_thread_impl.h" | 9 #include "content/browser/browser_thread_impl.h" |
10 #include "content/browser/indexed_db/indexed_db_connection.h" | 10 #include "content/browser/indexed_db/indexed_db_connection.h" |
(...skipping 98 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
109 } | 109 } |
110 | 110 |
111 // Make sure we wait until the destructor has run. | 111 // Make sure we wait until the destructor has run. |
112 message_loop_.RunUntilIdle(); | 112 message_loop_.RunUntilIdle(); |
113 | 113 |
114 // No data was cleared because of SetForceKeepSessionState. | 114 // No data was cleared because of SetForceKeepSessionState. |
115 EXPECT_TRUE(base::DirectoryExists(normal_path)); | 115 EXPECT_TRUE(base::DirectoryExists(normal_path)); |
116 EXPECT_TRUE(base::DirectoryExists(session_only_path)); | 116 EXPECT_TRUE(base::DirectoryExists(session_only_path)); |
117 } | 117 } |
118 | 118 |
119 class ForceCloseDBCallbacks : public IndexedDBCallbacks { | 119 class MockConnection : public IndexedDBConnection { |
120 public: | 120 public: |
121 ForceCloseDBCallbacks(scoped_refptr<IndexedDBContextImpl> idb_context, | 121 explicit MockConnection(bool expect_force_close) |
122 const GURL& origin_url) | 122 : IndexedDBConnection(NULL, NULL), |
123 : IndexedDBCallbacks(NULL, 0, 0), | 123 expect_force_close_(expect_force_close), |
124 idb_context_(idb_context), | 124 force_close_called_(false) {} |
125 origin_url_(origin_url), | |
126 connection_(NULL) {} | |
127 | 125 |
128 virtual void OnSuccess() OVERRIDE {} | 126 virtual ~MockConnection() { |
129 virtual void OnSuccess(const std::vector<base::string16>&) OVERRIDE {} | 127 EXPECT_TRUE(force_close_called_ == expect_force_close_); |
130 virtual void OnSuccess(scoped_ptr<IndexedDBConnection> connection, | |
131 const IndexedDBDatabaseMetadata& metadata) OVERRIDE { | |
132 connection_ = connection.release(); | |
133 idb_context_->ConnectionOpened(origin_url_, connection_); | |
134 } | 128 } |
135 | 129 |
136 IndexedDBConnection* connection() { return connection_; } | 130 virtual void ForceClose() OVERRIDE { |
| 131 ASSERT_TRUE(expect_force_close_); |
| 132 force_close_called_ = true; |
| 133 } |
137 | 134 |
138 protected: | 135 virtual bool IsConnected() OVERRIDE { |
139 virtual ~ForceCloseDBCallbacks() {} | 136 return !force_close_called_; |
| 137 } |
140 | 138 |
141 private: | 139 private: |
142 scoped_refptr<IndexedDBContextImpl> idb_context_; | 140 bool expect_force_close_; |
143 GURL origin_url_; | 141 bool force_close_called_; |
144 IndexedDBConnection* connection_; | |
145 DISALLOW_COPY_AND_ASSIGN(ForceCloseDBCallbacks); | |
146 }; | 142 }; |
147 | 143 |
148 TEST_F(IndexedDBTest, ForceCloseOpenDatabasesOnDelete) { | 144 TEST_F(IndexedDBTest, ForceCloseOpenDatabasesOnDelete) { |
149 base::ScopedTempDir temp_dir; | 145 base::ScopedTempDir temp_dir; |
150 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 146 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
151 | 147 |
152 scoped_refptr<MockIndexedDBDatabaseCallbacks> open_db_callbacks( | |
153 new MockIndexedDBDatabaseCallbacks()); | |
154 scoped_refptr<MockIndexedDBDatabaseCallbacks> closed_db_callbacks( | |
155 new MockIndexedDBDatabaseCallbacks()); | |
156 | |
157 base::FilePath test_path; | 148 base::FilePath test_path; |
158 | 149 |
159 // Create the scope which will ensure we run the destructor of the context. | 150 // Create the scope which will ensure we run the destructor of the context. |
160 { | 151 { |
161 TestBrowserContext browser_context; | 152 TestBrowserContext browser_context; |
162 | 153 |
163 const GURL kTestOrigin("http://test/"); | 154 const GURL kTestOrigin("http://test/"); |
164 | 155 |
165 scoped_refptr<IndexedDBContextImpl> idb_context = new IndexedDBContextImpl( | 156 scoped_refptr<IndexedDBContextImpl> idb_context = new IndexedDBContextImpl( |
166 temp_dir.path(), special_storage_policy_, NULL, task_runner_); | 157 temp_dir.path(), special_storage_policy_, NULL, task_runner_); |
167 | 158 |
168 scoped_refptr<ForceCloseDBCallbacks> open_callbacks = | |
169 new ForceCloseDBCallbacks(idb_context, kTestOrigin); | |
170 | |
171 scoped_refptr<ForceCloseDBCallbacks> closed_callbacks = | |
172 new ForceCloseDBCallbacks(idb_context, kTestOrigin); | |
173 | |
174 IndexedDBFactory* factory = idb_context->GetIDBFactory(); | |
175 | |
176 test_path = idb_context->GetFilePathForTesting( | 159 test_path = idb_context->GetFilePathForTesting( |
177 webkit_database::GetIdentifierFromOrigin(kTestOrigin)); | 160 webkit_database::GetIdentifierFromOrigin(kTestOrigin)); |
| 161 ASSERT_TRUE(base::CreateDirectory(test_path)); |
178 | 162 |
179 factory->Open(base::ASCIIToUTF16("opendb"), | 163 const bool kExpectForceClose = true; |
180 0, | |
181 0, | |
182 open_callbacks, | |
183 open_db_callbacks, | |
184 kTestOrigin, | |
185 idb_context->data_path()); | |
186 factory->Open(base::ASCIIToUTF16("closeddb"), | |
187 0, | |
188 0, | |
189 closed_callbacks, | |
190 closed_db_callbacks, | |
191 kTestOrigin, | |
192 idb_context->data_path()); | |
193 | 164 |
194 closed_callbacks->connection()->Close(); | 165 MockConnection connection1(kExpectForceClose); |
| 166 idb_context->TaskRunner()->PostTask( |
| 167 FROM_HERE, |
| 168 base::Bind(&IndexedDBContextImpl::ConnectionOpened, |
| 169 idb_context, |
| 170 kTestOrigin, |
| 171 &connection1)); |
| 172 |
| 173 MockConnection connection2(!kExpectForceClose); |
| 174 idb_context->TaskRunner()->PostTask( |
| 175 FROM_HERE, |
| 176 base::Bind(&IndexedDBContextImpl::ConnectionOpened, |
| 177 idb_context, |
| 178 kTestOrigin, |
| 179 &connection2)); |
| 180 idb_context->TaskRunner()->PostTask( |
| 181 FROM_HERE, |
| 182 base::Bind(&IndexedDBContextImpl::ConnectionClosed, |
| 183 idb_context, |
| 184 kTestOrigin, |
| 185 &connection2)); |
195 | 186 |
196 idb_context->TaskRunner()->PostTask( | 187 idb_context->TaskRunner()->PostTask( |
197 FROM_HERE, | 188 FROM_HERE, |
198 base::Bind( | 189 base::Bind( |
199 &IndexedDBContextImpl::DeleteForOrigin, idb_context, kTestOrigin)); | 190 &IndexedDBContextImpl::DeleteForOrigin, idb_context, kTestOrigin)); |
200 FlushIndexedDBTaskRunner(); | 191 FlushIndexedDBTaskRunner(); |
201 message_loop_.RunUntilIdle(); | 192 message_loop_.RunUntilIdle(); |
202 } | 193 } |
203 | 194 |
204 // Make sure we wait until the destructor has run. | 195 // Make sure we wait until the destructor has run. |
205 message_loop_.RunUntilIdle(); | 196 message_loop_.RunUntilIdle(); |
206 | 197 |
207 EXPECT_TRUE(open_db_callbacks->forced_close_called()); | |
208 EXPECT_FALSE(closed_db_callbacks->forced_close_called()); | |
209 EXPECT_FALSE(base::DirectoryExists(test_path)); | 198 EXPECT_FALSE(base::DirectoryExists(test_path)); |
210 } | 199 } |
211 | 200 |
212 TEST_F(IndexedDBTest, DeleteFailsIfDirectoryLocked) { | 201 TEST_F(IndexedDBTest, DeleteFailsIfDirectoryLocked) { |
213 base::ScopedTempDir temp_dir; | 202 base::ScopedTempDir temp_dir; |
214 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); | 203 ASSERT_TRUE(temp_dir.CreateUniqueTempDir()); |
215 const GURL kTestOrigin("http://test/"); | 204 const GURL kTestOrigin("http://test/"); |
216 | 205 |
217 scoped_refptr<IndexedDBContextImpl> idb_context = new IndexedDBContextImpl( | 206 scoped_refptr<IndexedDBContextImpl> idb_context = new IndexedDBContextImpl( |
218 temp_dir.path(), special_storage_policy_, NULL, task_runner_); | 207 temp_dir.path(), special_storage_policy_, NULL, task_runner_); |
(...skipping 46 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
265 EXPECT_TRUE(factory->IsBackingStoreOpen(kTestOrigin)); | 254 EXPECT_TRUE(factory->IsBackingStoreOpen(kTestOrigin)); |
266 | 255 |
267 // Simulate the write failure. | 256 // Simulate the write failure. |
268 callbacks->connection()->database()->TransactionCommitFailed(); | 257 callbacks->connection()->database()->TransactionCommitFailed(); |
269 | 258 |
270 EXPECT_TRUE(db_callbacks->forced_close_called()); | 259 EXPECT_TRUE(db_callbacks->forced_close_called()); |
271 EXPECT_FALSE(factory->IsBackingStoreOpen(kTestOrigin)); | 260 EXPECT_FALSE(factory->IsBackingStoreOpen(kTestOrigin)); |
272 } | 261 } |
273 | 262 |
274 } // namespace content | 263 } // namespace content |
OLD | NEW |