| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 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 <set> | 5 #include <set> |
| 6 | 6 |
| 7 #include "base/test/test_simple_task_runner.h" | 7 #include "base/test/test_simple_task_runner.h" |
| 8 #include "content/browser/indexed_db/indexed_db_active_blob_registry.h" | 8 #include "content/browser/indexed_db/indexed_db_active_blob_registry.h" |
| 9 #include "content/browser/indexed_db/indexed_db_backing_store.h" | 9 #include "content/browser/indexed_db/indexed_db_backing_store.h" |
| 10 #include "content/browser/indexed_db/indexed_db_factory_impl.h" | |
| 11 #include "content/browser/indexed_db/indexed_db_fake_backing_store.h" | 10 #include "content/browser/indexed_db/indexed_db_fake_backing_store.h" |
| 11 #include "content/browser/indexed_db/mock_indexed_db_factory.h" |
| 12 #include "testing/gtest/include/gtest/gtest.h" | 12 #include "testing/gtest/include/gtest/gtest.h" |
| 13 | 13 |
| 14 namespace content { | 14 namespace content { |
| 15 | 15 |
| 16 namespace { | 16 namespace { |
| 17 | 17 |
| 18 class MockIDBFactory : public IndexedDBFactoryImpl { | 18 class MockIDBFactory : public MockIndexedDBFactory { |
| 19 public: | 19 public: |
| 20 MockIDBFactory() : IndexedDBFactoryImpl(NULL), duplicate_calls_(false) {} | 20 MockIDBFactory() : duplicate_calls_(false) {} |
| 21 | 21 |
| 22 virtual void ReportOutstandingBlobs(const GURL& origin_url, | 22 virtual void ReportOutstandingBlobs(const GURL& origin_url, |
| 23 bool blobs_outstanding) OVERRIDE { | 23 bool blobs_outstanding) OVERRIDE { |
| 24 if (blobs_outstanding) { | 24 if (blobs_outstanding) { |
| 25 if (origins_.count(origin_url)) { | 25 if (origins_.count(origin_url)) { |
| 26 duplicate_calls_ = true; | 26 duplicate_calls_ = true; |
| 27 } else { | 27 } else { |
| 28 origins_.insert(origin_url); | 28 origins_.insert(origin_url); |
| 29 } | 29 } |
| 30 } else { | 30 } else { |
| 31 if (!origins_.count(origin_url)) { | 31 if (!origins_.count(origin_url)) { |
| 32 duplicate_calls_ = true; | 32 duplicate_calls_ = true; |
| 33 } else { | 33 } else { |
| 34 origins_.erase(origin_url); | 34 origins_.erase(origin_url); |
| 35 } | 35 } |
| 36 } | 36 } |
| 37 } | 37 } |
| 38 | 38 |
| 39 bool CheckNoOriginsInUse() const { | 39 bool CheckNoOriginsInUse() const { |
| 40 return !duplicate_calls_ && !origins_.size(); | 40 return !duplicate_calls_ && !origins_.size(); |
| 41 } | 41 } |
| 42 |
| 42 bool CheckSingleOriginInUse(const GURL& origin) const { | 43 bool CheckSingleOriginInUse(const GURL& origin) const { |
| 43 return !duplicate_calls_ && origins_.size() == 1 && origins_.count(origin); | 44 return !duplicate_calls_ && origins_.size() == 1 && origins_.count(origin); |
| 44 } | 45 } |
| 45 | 46 |
| 46 protected: | 47 private: |
| 47 virtual ~MockIDBFactory() {} | 48 virtual ~MockIDBFactory() {} |
| 48 | 49 |
| 49 private: | |
| 50 std::set<GURL> origins_; | 50 std::set<GURL> origins_; |
| 51 bool duplicate_calls_; | 51 bool duplicate_calls_; |
| 52 | 52 |
| 53 DISALLOW_COPY_AND_ASSIGN(MockIDBFactory); | 53 DISALLOW_COPY_AND_ASSIGN(MockIDBFactory); |
| 54 }; | 54 }; |
| 55 | 55 |
| 56 class MockIDBBackingStore : public IndexedDBFakeBackingStore { | 56 class MockIDBBackingStore : public IndexedDBFakeBackingStore { |
| 57 public: | 57 public: |
| 58 MockIDBBackingStore(IndexedDBFactory* factory, base::TaskRunner* task_runner) | 58 MockIDBBackingStore(IndexedDBFactory* factory, base::TaskRunner* task_runner) |
| 59 : IndexedDBFakeBackingStore(factory, task_runner), | 59 : IndexedDBFakeBackingStore(factory, task_runner), |
| (...skipping 205 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 265 RunUntilIdle(); | 265 RunUntilIdle(); |
| 266 | 266 |
| 267 // Nothing changes. | 267 // Nothing changes. |
| 268 EXPECT_TRUE(factory()->CheckSingleOriginInUse(backing_store()->origin_url())); | 268 EXPECT_TRUE(factory()->CheckSingleOriginInUse(backing_store()->origin_url())); |
| 269 EXPECT_TRUE(backing_store()->CheckUnusedBlobsEmpty()); | 269 EXPECT_TRUE(backing_store()->CheckUnusedBlobsEmpty()); |
| 270 } | 270 } |
| 271 | 271 |
| 272 } // namespace | 272 } // namespace |
| 273 | 273 |
| 274 } // namespace content | 274 } // namespace content |
| OLD | NEW |