Index: ios/chrome/browser/ui/toolbar/test_toolbar_model_ios.h |
diff --git a/ios/chrome/browser/ui/toolbar/test_toolbar_model_ios.h b/ios/chrome/browser/ui/toolbar/test_toolbar_model_ios.h |
index 18153d4b4638c822b7cc87dcd12ee01be4caa8f4..5e63a410b2cb0edeb7d79ab9f78348ff7aa062b6 100644 |
--- a/ios/chrome/browser/ui/toolbar/test_toolbar_model_ios.h |
+++ b/ios/chrome/browser/ui/toolbar/test_toolbar_model_ios.h |
@@ -18,8 +18,10 @@ class TestToolbarModelIOS : public ToolbarModelIOS { |
TestToolbarModelIOS(); |
~TestToolbarModelIOS() override; |
+ // Getter to the TestToolbarModel to set its values in tests. |
+ TestToolbarModel* GetToolbarModel(); |
+ |
// ToolbarModelIOS implementation: |
- ToolbarModel* GetToolbarModel() override; |
bool IsLoading() override; |
CGFloat GetLoadProgressFraction() override; |
bool CanGoBack() override; |
@@ -29,6 +31,15 @@ class TestToolbarModelIOS : public ToolbarModelIOS { |
bool IsCurrentTabBookmarkedByUser() override; |
bool ShouldDisplayHintText() override; |
+ base::string16 GetFormattedURL(size_t* prefix_end) const override; |
+ GURL GetURL() const override; |
+ security_state::SecurityLevel GetSecurityLevel( |
+ bool ignore_editing) const override; |
+ gfx::VectorIconId GetVectorIcon() const override; |
+ base::string16 GetSecureVerboseText() const override; |
+ base::string16 GetEVCertName() const override; |
+ bool ShouldDisplayURL() const override; |
+ |
void set_is_loading(bool is_loading) { is_loading_ = is_loading; } |
void set_load_progress_fraction(CGFloat load_progress_fraction) { |
load_progress_fraction_ = load_progress_fraction; |