1 files changed, 83 insertions, 0 deletions
diff --git a/chrome/browser/cocoa/cocoa_utils_unittest.mm b/chrome/browser/cocoa/cocoa_utils_unittest.mm
new file mode 100644
index 0000000..7812f2c
--- /dev/null
+++ b/chrome/browser/cocoa/cocoa_utils_unittest.mm
@@ -0,0 +1,83 @@
+// Copyright (c) 2009 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 "chrome/browser/cocoa/cocoa_utils.h"
+#import "chrome/browser/cocoa/cocoa_test_helper.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace {
+
+class CocoaUtilsTest : public testing::Test {
+ public:
+  CocoaTestHelper cocoa_helper_;  // Inits Cocoa, creates window, etc...
+
+  // If not red, is blue.
+  // If not tfbit (twenty-four-bit), is 444.
+  // If swap, swap R and B before testing (see TODOs in cocoa_utils.mm)
+  void ShapeHelper(int width, int height, bool isred, bool tfbit, bool swap);
+};
+
+void CocoaUtilsTest::ShapeHelper(int width, int height,
+                                 bool isred, bool tfbit, bool swap) {
+  SkBitmap thing;
+
+  if (tfbit)
+    thing.setConfig(SkBitmap::kARGB_8888_Config, width, height);
+  else
+    thing.setConfig(SkBitmap::kARGB_4444_Config, width, height);
+  thing.allocPixels();
+
+  if (isred)
+    thing.eraseRGB(0xff, 0, 0);
+  else
+    thing.eraseRGB(0, 0, 0xff);
+
+  // Confirm size
+  NSImage* image = CocoaUtils::SkBitmapToNSImage(thing);
+  EXPECT_DOUBLE_EQ([image size].width, (double)width);
+  EXPECT_DOUBLE_EQ([image size].height, (double)height);
+
+  // Get the color of a pixel and make sure it looks fine
+  [image lockFocus];
+
+  int x = width > 17 ? 17 : 0;
+  int y = height > 17 ? 17 : 0;
+  NSColor* color = NSReadPixel(NSMakePoint(x, y));
+  CGFloat red = 0, green = 0, blue = 0, alpha = 0;
+  [image unlockFocus];
+  [color getRed:&red green:&green blue:&blue alpha:&alpha];
+
+  if (swap) {
+    CGFloat tmp = red;
+    red = blue;
+    blue = tmp;
+  }
+
+  // Be tolerant of floating point rounding, gamma, etc.
+  if (isred) {
+    EXPECT_GT(red, 0.8);
+    EXPECT_LT(blue, 0.2);
+  } else {
+    EXPECT_LT(red, 0.2);
+    EXPECT_GT(blue, 0.8);
+  }
+  EXPECT_LT(green, 0.2);
+  EXPECT_GT(alpha, 0.9);
+}
+
+
+TEST_F(CocoaUtilsTest, BitmapToNSImage_RedSquare64x64) {
+  ShapeHelper(64, 64, true, true, true);
+}
+
+TEST_F(CocoaUtilsTest, BitmapToNSImage_BlueRectangle199x19) {
+  ShapeHelper(199, 19, false, true, true);
+}
+
+TEST_F(CocoaUtilsTest, BitmapToNSImage_BlueRectangle444) {
+  ShapeHelper(200, 200, false, false, false);
+}
+
+
+}  // namespace