| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #import <Cocoa/Cocoa.h> | 5 #import <Cocoa/Cocoa.h> |
| 6 | 6 |
| 7 #include "base/message_loop.h" | 7 #include "base/message_loop.h" |
| 8 #include "base/sys_string_conversions.h" | 8 #include "base/sys_string_conversions.h" |
| 9 #include "base/utf_string_conversions.h" | 9 #include "base/utf_string_conversions.h" |
| 10 #include "chrome/browser/ui/cocoa/cocoa_test_helper.h" | 10 #include "chrome/browser/ui/cocoa/cocoa_test_helper.h" |
| (...skipping 15 matching lines...) Expand all Loading... |
| 26 class Delegate : public ui::SimpleMenuModel::Delegate { | 26 class Delegate : public ui::SimpleMenuModel::Delegate { |
| 27 public: | 27 public: |
| 28 Delegate() | 28 Delegate() |
| 29 : execute_count_(0), | 29 : execute_count_(0), |
| 30 enable_count_(0), | 30 enable_count_(0), |
| 31 menu_to_close_(nil), | 31 menu_to_close_(nil), |
| 32 did_show_(false), | 32 did_show_(false), |
| 33 did_close_(false) { | 33 did_close_(false) { |
| 34 } | 34 } |
| 35 | 35 |
| 36 virtual bool IsCommandIdChecked(int command_id) const { return false; } | 36 virtual bool IsCommandIdChecked(int command_id) const OVERRIDE { |
| 37 virtual bool IsCommandIdEnabled(int command_id) const { | 37 return false; |
| 38 } |
| 39 virtual bool IsCommandIdEnabled(int command_id) const OVERRIDE { |
| 38 ++enable_count_; | 40 ++enable_count_; |
| 39 return true; | 41 return true; |
| 40 } | 42 } |
| 41 virtual bool GetAcceleratorForCommandId( | 43 virtual bool GetAcceleratorForCommandId( |
| 42 int command_id, | 44 int command_id, |
| 43 ui::Accelerator* accelerator) { return false; } | 45 ui::Accelerator* accelerator) OVERRIDE { return false; } |
| 44 virtual void ExecuteCommand(int command_id) { ++execute_count_; } | 46 virtual void ExecuteCommand(int command_id) OVERRIDE { ++execute_count_; } |
| 45 | 47 |
| 46 virtual void MenuWillShow(ui::SimpleMenuModel* /*source*/) { | 48 virtual void MenuWillShow(ui::SimpleMenuModel* /*source*/) OVERRIDE { |
| 47 EXPECT_FALSE(did_show_); | 49 EXPECT_FALSE(did_show_); |
| 48 EXPECT_FALSE(did_close_); | 50 EXPECT_FALSE(did_close_); |
| 49 did_show_ = true; | 51 did_show_ = true; |
| 50 NSArray* modes = [NSArray arrayWithObjects:NSEventTrackingRunLoopMode, | 52 NSArray* modes = [NSArray arrayWithObjects:NSEventTrackingRunLoopMode, |
| 51 NSDefaultRunLoopMode, | 53 NSDefaultRunLoopMode, |
| 52 nil]; | 54 nil]; |
| 53 [menu_to_close_ performSelector:@selector(cancelTracking) | 55 [menu_to_close_ performSelector:@selector(cancelTracking) |
| 54 withObject:nil | 56 withObject:nil |
| 55 afterDelay:0.1 | 57 afterDelay:0.1 |
| 56 inModes:modes]; | 58 inModes:modes]; |
| 57 } | 59 } |
| 58 | 60 |
| 59 virtual void MenuClosed(ui::SimpleMenuModel* /*source*/) { | 61 virtual void MenuClosed(ui::SimpleMenuModel* /*source*/) OVERRIDE { |
| 60 EXPECT_TRUE(did_show_); | 62 EXPECT_TRUE(did_show_); |
| 61 EXPECT_FALSE(did_close_); | 63 EXPECT_FALSE(did_close_); |
| 62 did_close_ = true; | 64 did_close_ = true; |
| 63 } | 65 } |
| 64 | 66 |
| 65 int execute_count_; | 67 int execute_count_; |
| 66 mutable int enable_count_; | 68 mutable int enable_count_; |
| 67 // The menu on which to call |-cancelTracking| after a short delay in | 69 // The menu on which to call |-cancelTracking| after a short delay in |
| 68 // MenuWillShow. | 70 // MenuWillShow. |
| 69 NSMenu* menu_to_close_; | 71 NSMenu* menu_to_close_; |
| 70 bool did_show_; | 72 bool did_show_; |
| 71 bool did_close_; | 73 bool did_close_; |
| 72 }; | 74 }; |
| 73 | 75 |
| 74 // Just like Delegate, except the items are treated as "dynamic" so updates to | 76 // Just like Delegate, except the items are treated as "dynamic" so updates to |
| 75 // the label/icon in the model are reflected in the menu. | 77 // the label/icon in the model are reflected in the menu. |
| 76 class DynamicDelegate : public Delegate { | 78 class DynamicDelegate : public Delegate { |
| 77 public: | 79 public: |
| 78 DynamicDelegate() {} | 80 DynamicDelegate() {} |
| 79 virtual bool IsItemForCommandIdDynamic(int command_id) const { return true; } | 81 virtual bool IsItemForCommandIdDynamic(int command_id) const OVERRIDE { |
| 80 virtual string16 GetLabelForCommandId(int command_id) const { return label_; } | 82 return true; |
| 81 virtual bool GetIconForCommandId(int command_id, gfx::Image* icon) const { | 83 } |
| 84 virtual string16 GetLabelForCommandId(int command_id) const OVERRIDE { |
| 85 return label_; |
| 86 } |
| 87 virtual bool GetIconForCommandId( |
| 88 int command_id, |
| 89 gfx::Image* icon) const OVERRIDE { |
| 82 if (icon_.IsEmpty()) { | 90 if (icon_.IsEmpty()) { |
| 83 return false; | 91 return false; |
| 84 } else { | 92 } else { |
| 85 *icon = icon_; | 93 *icon = icon_; |
| 86 return true; | 94 return true; |
| 87 } | 95 } |
| 88 } | 96 } |
| 89 void SetDynamicLabel(string16 label) { label_ = label; } | 97 void SetDynamicLabel(string16 label) { label_ = label; } |
| 90 void SetDynamicIcon(const gfx::Image& icon) { icon_ = icon; } | 98 void SetDynamicIcon(const gfx::Image& icon) { icon_ = icon; } |
| 91 | 99 |
| (...skipping 235 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 327 EXPECT_FALSE(delegate.did_close_); | 335 EXPECT_FALSE(delegate.did_close_); |
| 328 | 336 |
| 329 // Pump the task that notifies the delegate. | 337 // Pump the task that notifies the delegate. |
| 330 message_loop.RunUntilIdle(); | 338 message_loop.RunUntilIdle(); |
| 331 | 339 |
| 332 // Expect that the delegate got notified properly. | 340 // Expect that the delegate got notified properly. |
| 333 EXPECT_TRUE(delegate.did_close_); | 341 EXPECT_TRUE(delegate.did_close_); |
| 334 } | 342 } |
| 335 | 343 |
| 336 } // namespace | 344 } // namespace |
| OLD | NEW |