1 // Copyright 2014 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.
5 #include "ui/views/controls/menu/menu_item_view.h"
7 #include "base/strings/string16.h"
8 #include "base/strings/utf_string_conversions.h"
9 #include "testing/gtest/include/gtest/gtest.h"
10 #include "ui/views/controls/menu/submenu_view.h"
11 #include "ui/views/view.h"
15 // A simple View class that will match its height to the available width.
16 class SquareView : public views::View {
19 virtual ~SquareView() {}
22 virtual gfx::Size GetPreferredSize() const OVERRIDE {
23 return gfx::Size(1, 1);
25 virtual int GetHeightForWidth(int width) const OVERRIDE {
30 // A MenuItemView implementation with a public destructor (so we can clean up
32 class TestMenuItemView : public views::MenuItemView {
34 TestMenuItemView() : views::MenuItemView(NULL) {}
35 virtual ~TestMenuItemView() {}
40 TEST(MenuItemViewUnitTest, TestMenuItemViewWithFlexibleWidthChild) {
41 TestMenuItemView root_menu;
42 root_menu.set_owned_by_client();
44 // Append a normal MenuItemView.
45 views::MenuItemView* label_view =
46 root_menu.AppendMenuItemWithLabel(1, base::ASCIIToUTF16("item 1"));
48 // Append a second MenuItemView that has a child SquareView.
49 views::MenuItemView* flexible_view =
50 root_menu.AppendMenuItemWithLabel(2, base::string16());
51 flexible_view->AddChildView(new SquareView());
52 // Set margins to 0 so that we know width should match height.
53 flexible_view->SetMargins(0, 0);
55 views::SubmenuView* submenu = root_menu.GetSubmenu();
57 // The first item should be the label view.
58 ASSERT_EQ(label_view, submenu->GetMenuItemAt(0));
59 gfx::Size label_size = label_view->GetPreferredSize();
61 // The second item should be the flexible view.
62 ASSERT_EQ(flexible_view, submenu->GetMenuItemAt(1));
63 gfx::Size flexible_size = flexible_view->GetPreferredSize();
65 // The flexible view's "preferred size" should be 1x1...
66 EXPECT_EQ(flexible_size, gfx::Size(1, 1));
68 // ...but it should use whatever space is available to make a square.
69 int flex_height = flexible_view->GetHeightForWidth(label_size.width());
70 EXPECT_EQ(label_size.width(), flex_height);
72 // The submenu should be tall enough to allow for both menu items at the given
74 EXPECT_EQ(label_size.height() + flex_height,
75 submenu->GetPreferredSize().height());