1 files changed, 163 insertions, 0 deletions
diff --git a/chrome/browser/cocoa/infobar_test_helper.h b/chrome/browser/cocoa/infobar_test_helper.h
new file mode 100644
index 0000000..d83bdc1
--- /dev/null
+++ b/chrome/browser/cocoa/infobar_test_helper.h
@@ -0,0 +1,163 @@
+// Copyright (c) 2009 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 "chrome/browser/tab_contents/infobar_delegate.h"
+
+namespace {
+const wchar_t* kMockAlertInfoBarMessage = L"MockAlertInfoBarMessage";
+const wchar_t* kMockLinkInfoBarMessage = L"MockLinkInfoBarMessage";
+const wchar_t* kMockLinkInfoBarLink = L"http://dev.chromium.org";
+const wchar_t* kMockConfirmInfoBarMessage = L"MockConfirmInfoBarMessage";
+}
+
+//////////////////////////////////////////////////////////////////////////
+// Mock InfoBarDelgates
+
+class MockAlertInfoBarDelegate : public AlertInfoBarDelegate {
+ public:
+  explicit MockAlertInfoBarDelegate()
+      : AlertInfoBarDelegate(NULL),
+        message_text_accessed(false),
+        icon_accessed(false),
+        closed(false) {
+  }
+
+  virtual std::wstring GetMessageText() const {
+    message_text_accessed = true;
+    return kMockAlertInfoBarMessage;
+  }
+
+  virtual SkBitmap* GetIcon() const {
+    icon_accessed = true;
+    return NULL;
+  }
+
+  virtual void InfoBarClosed() {
+    closed = true;
+  }
+
+  // These are declared mutable to get around const-ness issues.
+  mutable bool message_text_accessed;
+  mutable bool icon_accessed;
+  bool closed;
+};
+
+class MockLinkInfoBarDelegate : public LinkInfoBarDelegate {
+ public:
+  explicit MockLinkInfoBarDelegate()
+      : LinkInfoBarDelegate(NULL),
+        message_text_accessed(false),
+        link_text_accessed(false),
+        icon_accessed(false),
+        link_clicked(false),
+        closed(false),
+        closes_on_action(true) {
+  }
+
+  virtual std::wstring GetMessageTextWithOffset(size_t* link_offset) const {
+    message_text_accessed = true;
+    return kMockLinkInfoBarMessage;
+  }
+
+  virtual std::wstring GetLinkText() const {
+    link_text_accessed = true;
+    return kMockLinkInfoBarLink;
+  }
+
+  virtual SkBitmap* GetIcon() const {
+    icon_accessed = true;
+    return NULL;
+  }
+
+  virtual bool LinkClicked(WindowOpenDisposition disposition) {
+    link_clicked = true;
+    return closes_on_action;
+  }
+
+  virtual void InfoBarClosed() {
+    closed = true;
+  }
+
+  // These are declared mutable to get around const-ness issues.
+  mutable bool message_text_accessed;
+  mutable bool link_text_accessed;
+  mutable bool icon_accessed;
+  bool link_clicked;
+  bool closed;
+
+  // Determines whether the infobar closes when an action is taken or not.
+  bool closes_on_action;
+};
+
+class MockConfirmInfoBarDelegate : public ConfirmInfoBarDelegate {
+ public:
+  explicit MockConfirmInfoBarDelegate()
+      : ConfirmInfoBarDelegate(NULL),
+        message_text_accessed(false),
+        link_text_accessed(false),
+        icon_accessed(false),
+        ok_clicked(false),
+        cancel_clicked(false),
+        link_clicked(false),
+        closed(false),
+        closes_on_action(true) {
+  }
+
+  virtual int GetButtons() const {
+    return (BUTTON_OK | BUTTON_CANCEL);
+  }
+
+  virtual std::wstring GetButtonLabel(InfoBarButton button) const {
+    if (button == BUTTON_OK)
+      return L"OK";
+    else
+      return L"Cancel";
+  }
+
+  virtual bool Accept() {
+    ok_clicked = true;
+    return closes_on_action;
+  }
+
+  virtual bool Cancel() {
+    cancel_clicked = true;
+    return closes_on_action;
+  }
+
+  virtual std::wstring GetMessageText() const {
+    message_text_accessed = true;
+    return kMockConfirmInfoBarMessage;
+  }
+
+  virtual SkBitmap* GetIcon() const {
+    icon_accessed = true;
+    return NULL;
+  }
+
+  virtual void InfoBarClosed() {
+    closed = true;
+  }
+
+  virtual std::wstring GetLinkText() {
+    link_text_accessed = true;
+    return std::wstring();
+  }
+
+  virtual bool LinkClicked(WindowOpenDisposition disposition) {
+    link_clicked = true;
+    return closes_on_action;
+  }
+
+  // These are declared mutable to get around const-ness issues.
+  mutable bool message_text_accessed;
+  mutable bool link_text_accessed;
+  mutable bool icon_accessed;
+  bool ok_clicked;
+  bool cancel_clicked;
+  bool link_clicked;
+  bool closed;
+
+  // Determines whether the infobar closes when an action is taken or not.
+  bool closes_on_action;
+};