summaryrefslogtreecommitdiffstats
path: root/ui/gfx/image_mac_unittest.mm
blob: 7d19e739621f6da75576ff228c7df2de3ef40b4b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
// Copyright (c) 2011 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 <Cocoa/Cocoa.h>

#include "base/logging.h"
#include "base/memory/scoped_nsobject.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/skia/include/core/SkBitmap.h"
#include "ui/gfx/image.h"
#include "ui/gfx/image_unittest_util.h"

namespace {

class ImageMacTest : public testing::Test {
 public:
  size_t GetRepCount(const gfx::Image& image) {
    return image.representations_.size();
  }

  void CreateBitmapImageRep(int width, int height, NSImageRep** image_rep) {
    scoped_nsobject<NSImage> image([[NSImage alloc]
        initWithSize:NSMakeSize(width, height)]);
    [image lockFocus];
    [[NSColor redColor] set];
    NSRectFill(NSMakeRect(0, 0, width, height));
    [image unlockFocus];
    EXPECT_TRUE([[[image representations] lastObject]
        isKindOfClass:[NSImageRep class]]);
    *image_rep = [[image representations] lastObject];
  }
};

namespace gt = gfx::test;

TEST_F(ImageMacTest, MultiResolutionNSImageToSkBitmap) {
  const int width1 = 10;
  const int height1 = 12;
  const int width2 = 20;
  const int height2 = 24;

  NSImageRep* image_rep_1;
  CreateBitmapImageRep(width1, height1, &image_rep_1);
  NSImageRep* image_rep_2;
  CreateBitmapImageRep(width2, height2, &image_rep_2);
  scoped_nsobject<NSImage> ns_image([[NSImage alloc]
      initWithSize:NSMakeSize(width1, height1)]);
  [ns_image addRepresentation:image_rep_1];
  [ns_image addRepresentation:image_rep_2];

  gfx::Image image(ns_image.release());

  EXPECT_EQ(1u, GetRepCount(image));
  EXPECT_EQ(2u, image.GetNumberOfSkBitmaps());

  const SkBitmap* bitmap1 = image.GetSkBitmapAtIndex(0);
  EXPECT_TRUE(bitmap1);
  const SkBitmap* bitmap2 = image.GetSkBitmapAtIndex(1);
  EXPECT_TRUE(bitmap2);

  if (bitmap1->width() == width1) {
    EXPECT_EQ(bitmap1->height(), height1);
    EXPECT_EQ(bitmap2->width(), width2);
    EXPECT_EQ(bitmap2->height(), height2);
  } else {
    EXPECT_EQ(bitmap1->width(), width2);
    EXPECT_EQ(bitmap1->height(), height2);
    EXPECT_EQ(bitmap2->width(), width1);
    EXPECT_EQ(bitmap2->height(), height1);
  }

  // GetNumberOfSkBitmaps and GetSkBitmapAtIndex should create a second
  // representation.
  EXPECT_EQ(2u, GetRepCount(image));
}

TEST_F(ImageMacTest, MultiResolutionSkBitmapToNSImage) {
  const int width1 = 10;
  const int height1 = 12;
  const int width2 = 20;
  const int height2 = 24;

  std::vector<const SkBitmap*> bitmaps;
  bitmaps.push_back(gt::CreateBitmap(width1, height1));
  bitmaps.push_back(gt::CreateBitmap(width2, height2));
  gfx::Image image(bitmaps);

  EXPECT_EQ(1u, GetRepCount(image));
  EXPECT_EQ(2u, image.GetNumberOfSkBitmaps());

  NSImage* ns_image = image;
  EXPECT_TRUE(ns_image);

  EXPECT_EQ(2u, [[image representations] count]);
  NSImageRep* image_rep_1 = [[image representations] objectAtIndex:0];
  NSImageRep* image_rep_2 = [[image representations] objectAtIndex:1];

  if ([image_rep_1 size].width == width1) {
    EXPECT_EQ([image_rep_1 size].height, height1);
    EXPECT_EQ([image_rep_2 size].width, width2);
    EXPECT_EQ([image_rep_2 size].height, height2);
  } else {
    EXPECT_EQ([image_rep_1 size].width, width2);
    EXPECT_EQ([image_rep_1 size].height, height2);
    EXPECT_EQ([image_rep_2 size].width, width1);
    EXPECT_EQ([image_rep_2 size].height, height1);
  }

  // Cast to NSImage* should create a second representation.
  EXPECT_EQ(2u, GetRepCount(image));
}

} // namespace