mac: Add tests for reading images from NSPasteboard.

ui/base/clipboard/clipboard_mac_unittest.mm

Index: ui/base/clipboard/clipboard_mac_unittest.mm
diff --git a/ui/base/clipboard/clipboard_mac_unittest.mm b/ui/base/clipboard/clipboard_mac_unittest.mm
new file mode 100644
index 0000000000000000000000000000000000000000..72ce910bbda09bcd2f05617139a3e3c4f06648ea
--- /dev/null
+++ b/ui/base/clipboard/clipboard_mac_unittest.mm
@@ -0,0 +1,81 @@
+// Copyright 2016 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.
+
+#import "ui/base/clipboard/clipboard_mac.h"
+
+#include "base/mac/scoped_cftyperef.h"
+#include "base/mac/scoped_nsobject.h"
+#include "base/memory/free_deleter.h"
+#include "base/memory/ref_counted.h"
+#include "testing/platform_test.h"
+#include "third_party/skia/include/core/SkBitmap.h"
+#include "ui/base/clipboard/clipboard_types.h"
+#include "ui/base/clipboard/clipboard_util_mac.h"
+
+namespace ui {
+
+class ClipboardMacTest : public PlatformTest {
+ public:
+  ClipboardMacTest() { }
+
+  base::scoped_nsobject<NSImage> CreateImage(int32_t width,
+                                             int32_t height,
+                                             bool retina) {
+    int32_t pixel_width = retina ? width * 2 : width;
+    int32_t pixel_height = retina ? height * 2 : height;
+
+    // It seems more natural to create an NSBitmapImageRep and set it as the
+    // representation for an NSImage. This doesn't work, because when the
+    // result is written, and then read from an NSPasteboard, the new NSImage
+    // loses its "retina-ness".
+    std::unique_ptr<uint8_t, base::FreeDeleter> buffer(
+        static_cast<uint8_t*>(calloc(pixel_width * pixel_height * 4, 1)));

[Avi (use Gerrit), 2016/08/26 19:24:05]

Or calloc(pixel_width * pixel_height, 4) if we're playing by calloc's semantics
for the arguments (which btw are silly).

[erikchen, 2016/08/26 19:36:23]

your wish is my command.

+    base::ScopedCFTypeRef<CGDataProviderRef> provider(
+        CGDataProviderCreateWithData(
+            nullptr, buffer.get(), (pixel_width * pixel_height * 4), nullptr));
+    base::ScopedCFTypeRef<CGColorSpaceRef> color_space(
+        CGColorSpaceCreateWithName(kCGColorSpaceSRGB));
+    base::ScopedCFTypeRef<CGImageRef> image_ref(
+        CGImageCreate(pixel_width, pixel_height, 8, 32, 4 * pixel_width,
+                      color_space.get(), kCGBitmapByteOrderDefault,
+                      provider.get(), nullptr, NO, kCGRenderingIntentDefault));
+    return base::scoped_nsobject<NSImage>([[NSImage alloc]
+        initWithCGImage:image_ref.get()
+                   size:NSMakeSize(width, height)]);
+  }
+};
+
+TEST_F(ClipboardMacTest, ReadImageRetina) {
+  int32_t width = 99;
+  int32_t height = 101;
+  scoped_refptr<ui::UniquePasteboard> pasteboard = new ui::UniquePasteboard;
+  base::scoped_nsobject<NSImage> image = CreateImage(width, height, true);
+  [pasteboard->get() writeObjects:@[ image.get() ]];
+
+  ui::Clipboard* clipboard = ui::Clipboard::GetForCurrentThread();
+  ui::ClipboardMac* clipboard_mac = static_cast<ui::ClipboardMac*>(clipboard);
+
+  SkBitmap bitmap = clipboard_mac->ReadImage(ui::CLIPBOARD_TYPE_COPY_PASTE,
+                                             pasteboard->get());
+  EXPECT_EQ(width, bitmap.width());
+  EXPECT_EQ(height, bitmap.height());
+}
+
+TEST_F(ClipboardMacTest, ReadImageNonRetina) {
+  int32_t width = 99;
+  int32_t height = 101;
+  scoped_refptr<ui::UniquePasteboard> pasteboard = new ui::UniquePasteboard;
+  base::scoped_nsobject<NSImage> image = CreateImage(width, height, false);
+  [pasteboard->get() writeObjects:@[ image.get() ]];
+
+  ui::Clipboard* clipboard = ui::Clipboard::GetForCurrentThread();
+  ui::ClipboardMac* clipboard_mac = static_cast<ui::ClipboardMac*>(clipboard);
+
+  SkBitmap bitmap = clipboard_mac->ReadImage(ui::CLIPBOARD_TYPE_COPY_PASTE,
+                                             pasteboard->get());
+  EXPECT_EQ(width, bitmap.width());
+  EXPECT_EQ(height, bitmap.height());
+}
+
+}  // namespace ui