| 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 |