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(const GURL& page_url, |
18 const GURL& image_url, | 18 const enhanced_bookmarks::ImageRecord& image) override; |
19 const gfx::Image& image) override; | |
20 void Erase(const GURL& page_url) override; | 19 void Erase(const GURL& page_url) override; |
21 std::pair<gfx::Image, GURL> Get(const GURL& page_url) override; | 20 enhanced_bookmarks::ImageRecord Get(const GURL& page_url) override; |
22 gfx::Size GetSize(const GURL& page_url) override; | 21 gfx::Size GetSize(const GURL& page_url) override; |
23 void GetAllPageUrls(std::set<GURL>* urls) override; | 22 void GetAllPageUrls(std::set<GURL>* urls) override; |
24 void ClearAll() override; | 23 void ClearAll() override; |
25 int64 GetStoreSizeInBytes() override; | 24 int64 GetStoreSizeInBytes() override; |
26 | 25 |
27 protected: | 26 protected: |
28 ~TestImageStore() override; | 27 ~TestImageStore() override; |
29 | 28 |
30 private: | 29 private: |
31 typedef std::map<const GURL, std::pair<gfx::Image, const GURL> > ImageMap; | 30 typedef std::map<const GURL, enhanced_bookmarks::ImageRecord> ImageMap; |
32 ImageMap store_; | 31 ImageMap store_; |
33 | 32 |
34 DISALLOW_COPY_AND_ASSIGN(TestImageStore); | 33 DISALLOW_COPY_AND_ASSIGN(TestImageStore); |
35 }; | 34 }; |
36 | 35 |
37 #endif // COMPONENTS_ENHANCED_BOOKMARKS_TEST_IMAGE_STORE_H_ | 36 #endif // COMPONENTS_ENHANCED_BOOKMARKS_TEST_IMAGE_STORE_H_ |
OLD | NEW |