Upstream Chrome on iOS source code [9/11].

ios/chrome/browser/ui/toolbar/test_toolbar_model_ios.h

+// Copyright 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.
+
+#ifndef IOS_CHROME_BROWSER_UI_TOOLBAR_TEST_TOOLBAR_MODEL_IOS_H_
+#define IOS_CHROME_BROWSER_UI_TOOLBAR_TEST_TOOLBAR_MODEL_IOS_H_
+
+#import <UIKit/UIKit.h>
+
+#include <memory>
+
+#include "base/mac/scoped_nsobject.h"
+#include "components/toolbar/test_toolbar_model.h"
+#include "ios/chrome/browser/ui/toolbar/toolbar_model_ios.h"
+
+class TestToolbarModelIOS : public ToolbarModelIOS {
+ public:
+  TestToolbarModelIOS();
+  ~TestToolbarModelIOS() override;
+
+  // ToolbarModelIOS implementation:
+  ToolbarModel* GetToolbarModel() override;
+  bool IsLoading() override;
+  CGFloat GetLoadProgressFraction() override;
+  bool CanGoBack() override;
+  bool CanGoForward() override;
+  bool IsCurrentTabNativePage() override;
+  bool IsCurrentTabBookmarked() override;
+  bool IsCurrentTabBookmarkedByUser() override;
+  bool ShouldDisplayHintText() 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;
+  }
+  void set_can_go_back(bool can_go_back) { can_go_back_ = can_go_back; }
+  void set_can_go_forward(bool can_go_forward) {
+    can_go_forward_ = can_go_forward;
+  }
+  void set_is_current_tab_native_page(bool is_current_tab_native_page) {
+    is_current_tab_native_page_ = is_current_tab_native_page;
+  }
+  void set_is_current_tab_bookmarked(bool is_current_tab_bookmarked) {
+    is_current_tab_bookmarked_ = is_current_tab_bookmarked;
+  }
+
+ private:
+  std::unique_ptr<TestToolbarModel> test_toolbar_model_;
+  bool is_loading_;
+  CGFloat load_progress_fraction_;
+  bool can_go_back_;
+  bool can_go_forward_;
+  bool is_current_tab_native_page_;
+  bool is_current_tab_bookmarked_;
+};
+
+#endif  // IOS_CHROME_BROWSER_UI_TOOLBAR_TEST_TOOLBAR_MODEL_IOS_H_