| // Copyright 2014 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "chrome/browser/bitmap_fetcher/bitmap_fetcher_service.h" |
| |
| #include "chrome/browser/bitmap_fetcher/bitmap_fetcher.h" |
| #include "chrome/test/base/testing_profile.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace { |
| |
| class TestObserver : public BitmapFetcherService::Observer { |
| public: |
| explicit TestObserver(BitmapFetcherService::Observer* target) |
| : target_(target) {} |
| virtual ~TestObserver() {} |
| |
| virtual void OnImageChanged(BitmapFetcherService::RequestId request_id, |
| const SkBitmap& answers_image) OVERRIDE { |
| target_->OnImageChanged(request_id, answers_image); |
| } |
| |
| BitmapFetcherService::Observer* target_; |
| }; |
| |
| class TestService : public BitmapFetcherService { |
| public: |
| explicit TestService(content::BrowserContext* context) |
| : BitmapFetcherService(context) {} |
| virtual ~TestService() {} |
| |
| // Create a fetcher, but don't start downloading. That allows side-stepping |
| // the decode step, which requires a utility process. |
| virtual chrome::BitmapFetcher* CreateFetcher(const GURL& url) OVERRIDE { |
| return new chrome::BitmapFetcher(url, this); |
| } |
| }; |
| |
| } // namespace |
| |
| class BitmapFetcherServiceTest : public testing::Test, |
| public BitmapFetcherService::Observer { |
| public: |
| virtual void SetUp() OVERRIDE { |
| service_.reset(new TestService(&profile_)); |
| requestsFinished_ = 0; |
| imagesChanged_ = 0; |
| url1_ = GURL("http://example.org/sample-image-1.png"); |
| url2_ = GURL("http://example.org/sample-image-2.png"); |
| } |
| |
| const ScopedVector<BitmapFetcherRequest>& requests() { |
| return service_->requests_; |
| } |
| const ScopedVector<chrome::BitmapFetcher>& active_fetchers() { |
| return service_->active_fetchers_; |
| } |
| size_t cache_size() { return service_->cache_.size(); } |
| |
| virtual void OnImageChanged(BitmapFetcherService::RequestId request_id, |
| const SkBitmap& answers_image) OVERRIDE { |
| if (answers_image.empty()) |
| requestsFinished_++; |
| else |
| imagesChanged_++; |
| } |
| |
| // Simulate finishing a URL fetch and decode for the given fetcher. |
| void CompleteFetch(const GURL& url) { |
| const chrome::BitmapFetcher* fetcher = service_->FindFetcherForUrl(url); |
| ASSERT_TRUE(NULL != fetcher); |
| |
| // Create a non-empty bitmap. |
| SkBitmap image; |
| image.setConfig(SkBitmap::kARGB_8888_Config, 2, 2); |
| image.allocPixels(); |
| image.eraseColor(SK_ColorGREEN); |
| |
| const_cast<chrome::BitmapFetcher*>(fetcher)->OnImageDecoded(NULL, image); |
| } |
| |
| void FailFetch(const GURL& url) { |
| const chrome::BitmapFetcher* fetcher = service_->FindFetcherForUrl(url); |
| ASSERT_TRUE(NULL != fetcher); |
| const_cast<chrome::BitmapFetcher*>(fetcher) |
| ->OnImageDecoded(NULL, SkBitmap()); |
| } |
| |
| protected: |
| scoped_ptr<BitmapFetcherService> service_; |
| |
| int imagesChanged_; |
| int requestsFinished_; |
| |
| GURL url1_; |
| GURL url2_; |
| |
| private: |
| TestingProfile profile_; |
| }; |
| |
| TEST_F(BitmapFetcherServiceTest, CancelInvalidRequest) { |
| service_->CancelRequest(BitmapFetcherService::REQUEST_ID_INVALID); |
| service_->CancelRequest(23); |
| } |
| |
| TEST_F(BitmapFetcherServiceTest, OnlyFirstRequestCreatesFetcher) { |
| EXPECT_EQ(0U, active_fetchers().size()); |
| |
| service_->RequestImage(url1_, new TestObserver(this)); |
| EXPECT_EQ(1U, active_fetchers().size()); |
| |
| service_->RequestImage(url1_, new TestObserver(this)); |
| EXPECT_EQ(1U, active_fetchers().size()); |
| } |
| |
| TEST_F(BitmapFetcherServiceTest, CompletedFetchNotifiesAllObservers) { |
| service_->RequestImage(url1_, new TestObserver(this)); |
| service_->RequestImage(url1_, new TestObserver(this)); |
| service_->RequestImage(url1_, new TestObserver(this)); |
| service_->RequestImage(url1_, new TestObserver(this)); |
| EXPECT_EQ(1U, active_fetchers().size()); |
| EXPECT_EQ(4U, requests().size()); |
| |
| CompleteFetch(url1_); |
| EXPECT_EQ(4, imagesChanged_); |
| EXPECT_EQ(4, requestsFinished_); |
| } |
| |
| TEST_F(BitmapFetcherServiceTest, CancelRequest) { |
| service_->RequestImage(url1_, new TestObserver(this)); |
| service_->RequestImage(url1_, new TestObserver(this)); |
| BitmapFetcherService::RequestId requestId = |
| service_->RequestImage(url2_, new TestObserver(this)); |
| service_->RequestImage(url1_, new TestObserver(this)); |
| service_->RequestImage(url1_, new TestObserver(this)); |
| EXPECT_EQ(5U, requests().size()); |
| |
| service_->CancelRequest(requestId); |
| EXPECT_EQ(4U, requests().size()); |
| |
| CompleteFetch(url2_); |
| EXPECT_EQ(0, imagesChanged_); |
| |
| CompleteFetch(url1_); |
| EXPECT_EQ(4, imagesChanged_); |
| } |
| |
| TEST_F(BitmapFetcherServiceTest, FailedRequestsDontEnterCache) { |
| service_->RequestImage(url1_, new TestObserver(this)); |
| service_->RequestImage(url2_, new TestObserver(this)); |
| EXPECT_EQ(0U, cache_size()); |
| |
| CompleteFetch(url1_); |
| EXPECT_EQ(1U, cache_size()); |
| |
| FailFetch(url2_); |
| EXPECT_EQ(1U, cache_size()); |
| } |