Index: chrome/browser/ui/toolbar/test_toolbar_model.h |
diff --git a/chrome/browser/ui/toolbar/test_toolbar_model.h b/chrome/browser/ui/toolbar/test_toolbar_model.h |
index f426056d222716f1376a3139a839bb507900c82b..6d2d5a652a17fcc1b24348e2acbcff2c78195d1a 100644 |
--- a/chrome/browser/ui/toolbar/test_toolbar_model.h |
+++ b/chrome/browser/ui/toolbar/test_toolbar_model.h |
@@ -27,6 +27,7 @@ class TestToolbarModel : public ToolbarModel { |
virtual int GetIconForSecurityLevel(SecurityLevel level) const OVERRIDE; |
virtual base::string16 GetEVCertName() const OVERRIDE; |
virtual bool ShouldDisplayURL() const OVERRIDE; |
+ virtual bool ShouldShowOriginChip() const OVERRIDE; |
void set_text(const base::string16& text) { text_ = text; } |
void set_url(const GURL& url) { url_ = url;} |
@@ -47,6 +48,9 @@ class TestToolbarModel : public ToolbarModel { |
void set_should_display_url(bool should_display_url) { |
should_display_url_ = should_display_url; |
} |
+ void set_should_show_origin_chip(bool should_show_origin_chip) { |
+ should_show_origin_chip_ = should_show_origin_chip; |
+ } |
private: |
virtual bool WouldOmitURLDueToOriginChip() const OVERRIDE; |
@@ -59,6 +63,7 @@ class TestToolbarModel : public ToolbarModel { |
int icon_; |
base::string16 ev_cert_name_; |
bool should_display_url_; |
+ bool should_show_origin_chip_; |
DISALLOW_COPY_AND_ASSIGN(TestToolbarModel); |
}; |