1 // Copyright (c) 2011 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 "base/memory/scoped_nsobject.h" 6 #include "chrome/app/chrome_command_ids.h" 7 #import "chrome/browser/ui/cocoa/cocoa_test_helper.h" 8 #import "chrome/browser/ui/cocoa/wrench_menu/wrench_menu_button_cell.h" 9 #include "testing/gtest/include/gtest/gtest.h" 10 #include "testing/platform_test.h" 11 12 @interface TestWrenchMenuButton : NSButton 13 @end 14 @implementation TestWrenchMenuButton 15 + (Class)cellClass { 16 return [WrenchMenuButtonCell class]; 17 } 18 @end 19 20 namespace { 21 22 class WrenchMenuButtonCellTest : public CocoaTest { 23 public: 24 void SetUp() { 25 CocoaTest::SetUp(); 26 27 NSRect frame = NSMakeRect(10, 10, 50, 19); 28 button_.reset([[TestWrenchMenuButton alloc] initWithFrame:frame]); 29 [button_ setBezelStyle:NSSmallSquareBezelStyle]; 30 [[button_ cell] setControlSize:NSSmallControlSize]; 31 [button_ setTitle:@"Allays"]; 32 [button_ setButtonType:NSMomentaryPushInButton]; 33 } 34 35 scoped_nsobject<NSButton> button_; 36 }; 37 38 TEST_F(WrenchMenuButtonCellTest, Draw) { 39 ASSERT_TRUE(button_.get()); 40 [[test_window() contentView] addSubview:button_.get()]; 41 [button_ setNeedsDisplay:YES]; 42 } 43 44 TEST_F(WrenchMenuButtonCellTest, DrawHighlight) { 45 ASSERT_TRUE(button_.get()); 46 [[test_window() contentView] addSubview:button_.get()]; 47 [button_ highlight:YES]; 48 [button_ setNeedsDisplay:YES]; 49 } 50 51 } // namespace 52