Index: chrome/browser/chromeos/login/screens/mock_error_screen.cc |
diff --git a/chrome/browser/chromeos/login/screens/mock_error_screen.cc b/chrome/browser/chromeos/login/screens/mock_error_screen.cc |
index 5f0ff041962317a38a5908c91b3f8afbe4e5bea7..5d8d6572331f424a89bfe07e62b48d0881f3a3bd 100644 |
--- a/chrome/browser/chromeos/login/screens/mock_error_screen.cc |
+++ b/chrome/browser/chromeos/login/screens/mock_error_screen.cc |
@@ -4,20 +4,36 @@ |
#include "chrome/browser/chromeos/login/screens/mock_error_screen.h" |
+using ::testing::AtLeast; |
+using ::testing::_; |
+ |
namespace chromeos { |
MockErrorScreen::MockErrorScreen(BaseScreenDelegate* base_screen_delegate, |
- ErrorScreenActor* actor) |
- : ErrorScreen(base_screen_delegate, actor) { |
+ NetworkErrorView* view) |
+ : ErrorScreen(base_screen_delegate, view) { |
} |
MockErrorScreen::~MockErrorScreen() { |
} |
-MockErrorScreenActor::MockErrorScreenActor() { |
+MockNetworkErrorView::MockNetworkErrorView() : model_(nullptr) { |
+ EXPECT_CALL(*this, MockBind(_)).Times(AtLeast(1)); |
+} |
+ |
+MockNetworkErrorView::~MockNetworkErrorView() { |
+ if (model_) |
+ model_->OnViewDestroyed(this); |
+} |
+ |
+void MockNetworkErrorView::Bind(NetworkErrorModel& model) { |
+ model_ = &model; |
+ MockBind(model); |
} |
-MockErrorScreenActor::~MockErrorScreenActor() { |
+void MockNetworkErrorView::Unbind() { |
+ model_ = nullptr; |
+ MockUnbind(); |
} |
} // namespace chromeos |