components/enhanced_bookmarks/test_image_store.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "components/enhanced_bookmarks/test_image_store.h"
   6
   7 #include "third_party/skia/include/core/SkBitmap.h"
   8 #include "ui/gfx/geometry/size.h"
   9 #include "url/gurl.h"
  10
  11 TestImageStore::TestImageStore() {
  12 }
  13
  14 bool TestImageStore::HasKey(const GURL& page_url) {
  15   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
  16
  17   return store_.find(page_url) != store_.end();
  18 }
  19
  20 void TestImageStore::Insert(
  21     const GURL& page_url,
  22     scoped_refptr<enhanced_bookmarks::ImageRecord> image_record) {
  23   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
  24
  25   Erase(page_url);
  26   store_.insert(std::make_pair(page_url, image_record));
  27 }
  28
  29 void TestImageStore::Erase(const GURL& page_url) {
  30   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
  31
  32   store_.erase(page_url);
  33 }
  34
  35 scoped_refptr<enhanced_bookmarks::ImageRecord> TestImageStore::Get(
  36     const GURL& page_url) {
  37   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
  38
  39   if (store_.find(page_url) == store_.end()) {
  40     return scoped_refptr<enhanced_bookmarks::ImageRecord>(
  41         new enhanced_bookmarks::ImageRecord());
  42   }
  43
  44   return store_[page_url];
  45 }
  46
  47 gfx::Size TestImageStore::GetSize(const GURL& page_url) {
  48   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
  49
  50   ImageMap::const_iterator pair_enumerator = store_.find(page_url);
  51   if (pair_enumerator == store_.end())
  52     return gfx::Size();
  53
  54   return store_[page_url]->image->Size();
  55 }
  56
  57 void TestImageStore::GetAllPageUrls(std::set<GURL>* urls) {
  58   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
  59   DCHECK(urls->empty());
  60
  61   for (ImageMap::const_iterator it = store_.begin(); it != store_.end(); ++it)
  62     urls->insert(it->first);
  63 }
  64
  65 void TestImageStore::ClearAll() {
  66   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
  67
  68   store_.clear();
  69 }
  70
  71 int64 TestImageStore::GetStoreSizeInBytes() {
  72   // Not 100% accurate, but it's for testing so the actual value is not
  73   // important.
  74   int64 size = sizeof(store_);
  75   for (ImageMap::const_iterator it = store_.begin(); it != store_.end(); ++it) {
  76     size += sizeof(it->first);
  77     size += it->first.spec().length();
  78     size += sizeof(it->second);
  79     SkBitmap bitmap = it->second->image->AsBitmap();
  80     size += bitmap.getSize();
  81     size += it->second->url.spec().length();
  82     size += sizeof(it->second->dominant_color);
  83   }
  84   return size;
  85 }
  86
  87 TestImageStore::~TestImageStore() {
  88   DCHECK(sequence_checker_.CalledOnValidSequencedThread());
  89 }