diff options
Diffstat (limited to 'chrome/browser/ui/cocoa/wrench_menu_button_cell_unittest.mm')
-rw-r--r-- | chrome/browser/ui/cocoa/wrench_menu_button_cell_unittest.mm | 51 |
1 files changed, 51 insertions, 0 deletions
diff --git a/chrome/browser/ui/cocoa/wrench_menu_button_cell_unittest.mm b/chrome/browser/ui/cocoa/wrench_menu_button_cell_unittest.mm new file mode 100644 index 0000000..7ab1588 --- /dev/null +++ b/chrome/browser/ui/cocoa/wrench_menu_button_cell_unittest.mm @@ -0,0 +1,51 @@ +// Copyright (c) 2010 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 "base/scoped_nsobject.h" +#include "chrome/app/chrome_command_ids.h" +#import "chrome/browser/ui/cocoa/cocoa_test_helper.h" +#import "chrome/browser/ui/cocoa/wrench_menu_button_cell.h" +#include "testing/gtest/include/gtest/gtest.h" +#include "testing/platform_test.h" + +@interface TestWrenchMenuButton : NSButton +@end +@implementation TestWrenchMenuButton ++ (Class)cellClass { + return [WrenchMenuButtonCell class]; +} +@end + +namespace { + +class WrenchMenuButtonCellTest : public CocoaTest { + public: + void SetUp() { + CocoaTest::SetUp(); + + NSRect frame = NSMakeRect(10, 10, 50, 19); + button_.reset([[TestWrenchMenuButton alloc] initWithFrame:frame]); + [button_ setBezelStyle:NSSmallSquareBezelStyle]; + [[button_ cell] setControlSize:NSSmallControlSize]; + [button_ setTitle:@"Allays"]; + [button_ setButtonType:NSMomentaryPushInButton]; + } + + scoped_nsobject<NSButton> button_; +}; + +TEST_F(WrenchMenuButtonCellTest, Draw) { + ASSERT_TRUE(button_.get()); + [[test_window() contentView] addSubview:button_.get()]; + [button_ setNeedsDisplay:YES]; +} + +TEST_F(WrenchMenuButtonCellTest, DrawHighlight) { + ASSERT_TRUE(button_.get()); + [[test_window() contentView] addSubview:button_.get()]; + [button_ highlight:YES]; + [button_ setNeedsDisplay:YES]; +} + +} // namespace |