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 <string>
6
7 #include "chrome/browser/image_holder.h"
8 #include "testing/gtest/include/gtest/gtest.h"
9
10 namespace {
11
12 const char kIconUrl1[] = "http://www.google.com/icon1.jpg";
13 const char kIconUrl2[] = "http://www.google.com/icon2.jpg";
14
15 class TestDelegate : public chrome::ImageHolderDelegate {
16 public:
TestDelegate()17 TestDelegate() : on_fetch_complete_called_(false) {}
OnFetchComplete()18 virtual void OnFetchComplete() OVERRIDE {
19 on_fetch_complete_called_ = true;
20 }
21 bool on_fetch_complete_called_;
22 };
23
24 } // namespace.
25
26 namespace chrome {
27
28 typedef testing::Test ImageHolderTest;
29
TEST_F(ImageHolderTest,CreateBitmapFetcherTest)30 TEST_F(ImageHolderTest, CreateBitmapFetcherTest) {
31 TestDelegate delegate;
32 ImageHolder image_holder(GURL(kIconUrl1), GURL(kIconUrl2), NULL, &delegate);
33
34 EXPECT_EQ(GURL(kIconUrl1), image_holder.fetchers_[0]->url());
35 EXPECT_EQ(GURL(kIconUrl2), image_holder.fetchers_[1]->url());
36 EXPECT_EQ(static_cast<unsigned int>(2), image_holder.fetchers_.size());
37
38 // Adding a dup of an existing URL shouldn't change anything.
39 image_holder.CreateBitmapFetcher(GURL(kIconUrl2));
40 EXPECT_EQ(GURL(kIconUrl1), image_holder.fetchers_[0]->url());
41 EXPECT_EQ(GURL(kIconUrl2), image_holder.fetchers_[1]->url());
42 EXPECT_EQ(static_cast<unsigned int>(2), image_holder.fetchers_.size());
43 }
44
TEST_F(ImageHolderTest,OnFetchCompleteTest)45 TEST_F(ImageHolderTest, OnFetchCompleteTest) {
46 TestDelegate delegate;
47 ImageHolder image_holder(GURL(kIconUrl1), GURL(), NULL, &delegate);
48
49 // Put a real bitmap into "bitmap". 2x2 bitmap of green 32 bit pixels.
50 SkBitmap bitmap;
51 bitmap.setConfig(SkBitmap::kARGB_8888_Config, 2, 2);
52 bitmap.allocPixels();
53 bitmap.eraseColor(SK_ColorGREEN);
54
55 image_holder.OnFetchComplete(GURL(kIconUrl1), &bitmap);
56
57 // Expect that the app icon has some data in it.
58 EXPECT_FALSE(image_holder.low_dpi_image().IsEmpty());
59
60 // Expect that we reported the fetch done to the delegate.
61 EXPECT_TRUE(delegate.on_fetch_complete_called_);
62 }
63
TEST_F(ImageHolderTest,IsFetchingDoneTest)64 TEST_F(ImageHolderTest, IsFetchingDoneTest) {
65 TestDelegate delegate;
66 ImageHolder image_holder1(GURL(kIconUrl1), GURL(kIconUrl2), NULL, &delegate);
67 ImageHolder image_holder2(GURL(kIconUrl1), GURL(), NULL, &delegate);
68 ImageHolder image_holder3(GURL(), GURL(kIconUrl2), NULL, &delegate);
69 ImageHolder image_holder4(GURL(), GURL(), NULL, &delegate);
70
71 // Initially, image holder 4 with no URLs should report done, but no others.
72 EXPECT_FALSE(image_holder1.IsFetchingDone());
73 EXPECT_FALSE(image_holder2.IsFetchingDone());
74 EXPECT_FALSE(image_holder3.IsFetchingDone());
75 EXPECT_TRUE(image_holder4.IsFetchingDone());
76
77 // Put a real bitmap into "bitmap". 2x2 bitmap of green 32 bit pixels.
78 SkBitmap bitmap;
79 bitmap.setConfig(SkBitmap::kARGB_8888_Config, 2, 2);
80 bitmap.allocPixels();
81 bitmap.eraseColor(SK_ColorGREEN);
82
83 // Add the first icon, and image holder 2 should now also report done.
84 image_holder1.OnFetchComplete(GURL(kIconUrl1), &bitmap);
85 image_holder2.OnFetchComplete(GURL(kIconUrl1), &bitmap);
86 image_holder3.OnFetchComplete(GURL(kIconUrl1), &bitmap);
87 image_holder4.OnFetchComplete(GURL(kIconUrl1), &bitmap);
88 EXPECT_FALSE(image_holder1.IsFetchingDone());
89 EXPECT_TRUE(image_holder2.IsFetchingDone());
90 EXPECT_FALSE(image_holder3.IsFetchingDone());
91 EXPECT_TRUE(image_holder4.IsFetchingDone());
92
93 // Add the second image, and now all 4 should report done.
94 image_holder1.OnFetchComplete(GURL(kIconUrl2), &bitmap);
95 image_holder2.OnFetchComplete(GURL(kIconUrl2), &bitmap);
96 image_holder3.OnFetchComplete(GURL(kIconUrl2), &bitmap);
97 image_holder4.OnFetchComplete(GURL(kIconUrl2), &bitmap);
98 EXPECT_TRUE(image_holder1.IsFetchingDone());
99 EXPECT_TRUE(image_holder2.IsFetchingDone());
100 EXPECT_TRUE(image_holder3.IsFetchingDone());
101 EXPECT_TRUE(image_holder4.IsFetchingDone());
102 }
103
104 } // namespace chrome.
105
