Add VectorIconButton functionality to ImageButton.

ui/views/controls/button/image_button_unittest.cc

 // Copyright (c) 2012 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 "ui/views/controls/button/image_button.h"
+
 #include "base/macros.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/base/layout.h"
+#include "ui/vector_icons/vector_icons.h"
 #include "ui/views/border.h"
-#include "ui/views/controls/button/image_button.h"
 #include "ui/views/test/views_test_base.h"
 
 namespace {
 
 gfx::ImageSkia CreateTestImage(int width, int height) {
   SkBitmap bitmap;
   bitmap.allocN32Pixels(width, height);
   return gfx::ImageSkia::CreateFrom1xBitmap(bitmap);
 }
 
@@ skipping 172 matching lines @@
   EXPECT_EQ(0, parent.pref_size_changed_calls());
 
   button.SetImage(CustomButton::STATE_HOVERED, &second_image);
   EXPECT_EQ(0, parent.pref_size_changed_calls());
 
   // Changing normal state image size leads to a change in preferred size.
   button.SetImage(CustomButton::STATE_NORMAL, &second_image);
   EXPECT_EQ(1, parent.pref_size_changed_calls());
 }
 
+TEST_F(ImageButtonTest, Padding) {
+  ImageButton button(nullptr);
+  gfx::ImageSkia image = CreateTestImage(5, 10);
+  button.SetImage(CustomButton::STATE_NORMAL, &image);
+  button.SetPadding(gfx::Insets(2, 5, 7, 13));
+  EXPECT_EQ("23x19", button.GetPreferredSize().ToString());
+}
+
+namespace {
+class TestImageButtonDelegate : public ImageButtonDelegate {
+  void ButtonPressed(Button* sender, const ui::Event& event) override {}
+};
+}
+
+TEST_F(ImageButtonTest, VectorIcon) {
+  TestImageButtonDelegate delegate;
+  ImageButton button(&delegate);
+  button.SetVectorIcon(ui::kCloseIcon);
+  EXPECT_FALSE(button.GetImage(CustomButton::STATE_NORMAL).isNull());
+  EXPECT_FALSE(button.GetImage(CustomButton::STATE_DISABLED).isNull());
+}
+
 }  // namespace views