1 // Copyright 2018 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 "chrome/browser/offline_pages/visuals_decoder_impl.h"
6
7 #include "base/base64.h"
8 #include "base/test/bind.h"
9 #include "base/test/mock_callback.h"
10 #include "chrome/browser/image_fetcher/image_decoder_impl.h"
11 #include "components/image_fetcher/core/mock_image_decoder.h"
12 #include "testing/gtest/include/gtest/gtest.h"
13 #include "ui/gfx/geometry/size.h"
14 #include "ui/gfx/image/image.h"
15 #include "ui/gfx/image/image_unittest_util.h"
16
17 namespace offline_pages {
18 namespace {
19 using testing::_;
20 using testing::Invoke;
21
22 const char kImageData[] = "abc123";
23
24 class VisualsDecoderImplTest : public testing::Test {
25 public:
SetUp()26 void SetUp() override {
27 auto decoder = std::make_unique<image_fetcher::MockImageDecoder>();
28 image_decoder = decoder.get();
29 visuals_decoder = std::make_unique<VisualsDecoderImpl>(std::move(decoder));
30 }
31 image_fetcher::MockImageDecoder* image_decoder;
32 std::unique_ptr<VisualsDecoderImpl> visuals_decoder;
33 };
34
TEST_F(VisualsDecoderImplTest,Success)35 TEST_F(VisualsDecoderImplTest, Success) {
36 const int kImageWidth = 4;
37 const int kImageHeight = 2;
38 const gfx::Image kDecodedImage =
39 gfx::test::CreateImage(kImageWidth, kImageHeight);
40 EXPECT_CALL(*image_decoder, DecodeImage(kImageData, _, _))
41 .WillOnce(testing::WithArg<2>(
42 Invoke([&](image_fetcher::ImageDecodedCallback callback) {
43 std::move(callback).Run(kDecodedImage);
44 })));
45 base::MockCallback<VisualsDecoder::DecodeComplete> complete_callback;
46 EXPECT_CALL(complete_callback, Run(_))
47 .WillOnce(Invoke([&](const gfx::Image& image) {
48 EXPECT_EQ(kImageHeight, image.Width());
49 EXPECT_EQ(kImageHeight, image.Height());
50 }));
51 visuals_decoder->DecodeAndCropImage(kImageData, complete_callback.Get());
52 }
53
TEST_F(VisualsDecoderImplTest,DecodeFail)54 TEST_F(VisualsDecoderImplTest, DecodeFail) {
55 const gfx::Image kDecodedImage = gfx::Image();
56 EXPECT_CALL(*image_decoder, DecodeImage(kImageData, _, _))
57 .WillOnce(testing::WithArg<2>(
58 Invoke([&](image_fetcher::ImageDecodedCallback callback) {
59 std::move(callback).Run(kDecodedImage);
60 })));
61 base::MockCallback<VisualsDecoder::DecodeComplete> complete_callback;
62 EXPECT_CALL(complete_callback, Run(_))
63 .WillOnce(Invoke(
64 [](const gfx::Image& image) { EXPECT_TRUE(image.IsEmpty()); }));
65 visuals_decoder->DecodeAndCropImage(kImageData, complete_callback.Get());
66 }
67
68 } // namespace
69 } // namespace offline_pages
70