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 #ifndef COMPONENTS_ENHANCED_BOOKMARKS_TEST_IMAGE_STORE_H_ | 5 #ifndef COMPONENTS_ENHANCED_BOOKMARKS_TEST_IMAGE_STORE_H_ |
6 #define COMPONENTS_ENHANCED_BOOKMARKS_TEST_IMAGE_STORE_H_ | 6 #define COMPONENTS_ENHANCED_BOOKMARKS_TEST_IMAGE_STORE_H_ |
7 | 7 |
8 #include "components/enhanced_bookmarks/image_store.h" | 8 #include "components/enhanced_bookmarks/image_store.h" |
9 | 9 |
10 // The TestImageStore is an implementation of ImageStore that keeps all its | 10 // The TestImageStore is an implementation of ImageStore that keeps all its |
11 // data in memory. When deallocated all the associations are lost. | 11 // data in memory. When deallocated all the associations are lost. |
12 // Used in tests. | 12 // Used in tests. |
13 class TestImageStore : public ImageStore { | 13 class TestImageStore : public ImageStore { |
14 public: | 14 public: |
15 TestImageStore(); | 15 TestImageStore(); |
16 bool HasKey(const GURL& page_url) override; | 16 bool HasKey(const GURL& page_url) override; |
17 void Insert(const GURL& page_url, | 17 void Insert( |
18 const enhanced_bookmarks::ImageRecord& image) override; | 18 const GURL& page_url, |
| 19 scoped_refptr<enhanced_bookmarks::ImageRecord> image_record) override; |
19 void Erase(const GURL& page_url) override; | 20 void Erase(const GURL& page_url) override; |
20 enhanced_bookmarks::ImageRecord Get(const GURL& page_url) override; | 21 scoped_refptr<enhanced_bookmarks::ImageRecord> Get( |
| 22 const GURL& page_url) override; |
21 gfx::Size GetSize(const GURL& page_url) override; | 23 gfx::Size GetSize(const GURL& page_url) override; |
22 void GetAllPageUrls(std::set<GURL>* urls) override; | 24 void GetAllPageUrls(std::set<GURL>* urls) override; |
23 void ClearAll() override; | 25 void ClearAll() override; |
24 int64 GetStoreSizeInBytes() override; | 26 int64 GetStoreSizeInBytes() override; |
25 | 27 |
26 protected: | 28 protected: |
27 ~TestImageStore() override; | 29 ~TestImageStore() override; |
28 | 30 |
29 private: | 31 private: |
30 typedef std::map<const GURL, enhanced_bookmarks::ImageRecord> ImageMap; | 32 typedef std::map<const GURL, scoped_refptr<enhanced_bookmarks::ImageRecord>> |
| 33 ImageMap; |
31 ImageMap store_; | 34 ImageMap store_; |
32 | 35 |
33 DISALLOW_COPY_AND_ASSIGN(TestImageStore); | 36 DISALLOW_COPY_AND_ASSIGN(TestImageStore); |
34 }; | 37 }; |
35 | 38 |
36 #endif // COMPONENTS_ENHANCED_BOOKMARKS_TEST_IMAGE_STORE_H_ | 39 #endif // COMPONENTS_ENHANCED_BOOKMARKS_TEST_IMAGE_STORE_H_ |
OLD | NEW |