Index: chrome/browser/ui/window_snapshot/window_snapshot_mac_unittest.mm |
diff --git a/chrome/browser/ui/window_snapshot/window_snapshot_mac_unittest.mm b/chrome/browser/ui/window_snapshot/window_snapshot_mac_unittest.mm |
new file mode 100755 |
index 0000000000000000000000000000000000000000..a7a377b0ee1503d97b8f8a6f2c4391c727e885f7 |
--- /dev/null |
+++ b/chrome/browser/ui/window_snapshot/window_snapshot_mac_unittest.mm |
@@ -0,0 +1,49 @@ |
+// 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 "chrome/browser/ui/window_snapshot/window_snapshot.h" |
+ |
+#import <Cocoa/Cocoa.h> |
+ |
+#include "base/scoped_nsobject.h" |
+#include "base/scoped_ptr.h" |
+#include "base/test/mock_chrome_application_mac.h" |
+#include "gfx/rect.h" |
+#include "testing/platform_test.h" |
+ |
+namespace browser { |
+namespace { |
+ |
+typedef PlatformTest GrabWindowSnapshotTest; |
+ |
+TEST_F(GrabWindowSnapshotTest, TestGrabWindowSnapshot) { |
+ // Launch a test window so we can take a snapshot. |
+ NSRect frame = NSMakeRect(0, 0, 400, 400); |
+ scoped_nsobject<NSWindow> window( |
+ [[NSWindow alloc] initWithContentRect:frame |
+ styleMask:NSBorderlessWindowMask |
+ backing:NSBackingStoreBuffered |
+ defer:NO]); |
+ [window setBackgroundColor:[NSColor whiteColor]]; |
+ [window makeKeyAndOrderFront:NSApp]; |
+ |
+ scoped_ptr<std::vector<unsigned char> > png_representation( |
+ new std::vector<unsigned char>); |
+ browser::GrabWindowSnapshot(window, png_representation.get()); |
+ |
+ // Copy png back into NSData object so we can make sure we grabbed a png. |
+ scoped_nsobject<NSData> image_data( |
+ [[NSData alloc] initWithBytes:&(*png_representation)[0] |
+ length:png_representation->size()]); |
+ NSBitmapImageRep* rep = [NSBitmapImageRep imageRepWithData:image_data.get()]; |
+ EXPECT_TRUE([rep isKindOfClass:[NSBitmapImageRep class]]); |
+ EXPECT_TRUE(CGImageGetWidth([rep CGImage]) == 400); |
+ NSColor* color = [rep colorAtX:200 y:200]; |
+ CGFloat red = 0, green = 0, blue = 0, alpha = 0; |
+ [color getRed:&red green:&green blue:&blue alpha:&alpha]; |
+ EXPECT_GE(red + green + blue, 3.0); |
+} |
+ |
+} // namespace |
+} // namespace browser |