Pass the Translate error type to the bubble

chrome/browser/ui/views/translate/translate_bubble_view_unittest.cc

 // Copyright 2013 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/ui/views/translate/translate_bubble_view.h"
 
 #include "base/memory/scoped_ptr.h"
 #include "chrome/browser/ui/translate/translate_bubble_model.h"
 #include "chrome/browser/ui/translate/translate_bubble_view_state_transition.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "ui/views/controls/button/checkbox.h"
 #include "ui/views/controls/combobox/combobox.h"
 #include "ui/views/test/views_test_base.h"
 #include "ui/views/widget/widget.h"
 
 namespace {
 
 class MockTranslateBubbleModel : public TranslateBubbleModel {
  public:
   explicit MockTranslateBubbleModel(TranslateBubbleModel::ViewState view_state)
       : view_state_transition_(view_state),
+        error_type_(TranslateErrors::NONE),
         original_language_index_(0),
         target_language_index_(1),
         never_translate_language_(false),
         never_translate_site_(false),
         should_always_translate_(false),
         set_always_translate_called_count_(0),
         translate_called_(false),
         revert_translation_called_(false),
         translation_declined_called_(false) {
   }
 
   virtual TranslateBubbleModel::ViewState GetViewState() const OVERRIDE {
     return view_state_transition_.view_state();
   }
 
   virtual void SetViewState(TranslateBubbleModel::ViewState view_state)
       OVERRIDE {
     view_state_transition_.SetViewState(view_state);
   }
 
+  virtual TranslateErrors::Type GetErrorType() const OVERRIDE {
+    return error_type_;
+  }
+
+  virtual void SetErrorType(TranslateErrors::Type error_type) OVERRIDE {
+    error_type_ = error_type;
+  }
+
   virtual void GoBackFromAdvanced() OVERRIDE {
     view_state_transition_.GoBackFromAdvanced();
   }
 
   virtual int GetNumberOfLanguages() const OVERRIDE {
     return 1000;
   }
 
   virtual string16 GetLanguageNameAt(int index) const OVERRIDE {
     return string16();
@@ skipping 38 matching lines @@
 
   virtual void RevertTranslation() OVERRIDE {
     revert_translation_called_ = true;
   }
 
   virtual void TranslationDeclined() OVERRIDE {
     translation_declined_called_ = true;
   }
 
   TranslateBubbleViewStateTransition view_state_transition_;
+  TranslateErrors::Type error_type_;
   int original_language_index_;
   int target_language_index_;
   bool never_translate_language_;
   bool never_translate_site_;
   bool should_always_translate_;
   int set_always_translate_called_count_;
   bool translate_called_;
   bool revert_translation_called_;
   bool translation_declined_called_;
 };
@@ skipping 61 matching lines @@
   bubble_->SwitchView(TranslateBubbleModel::VIEW_STATE_AFTER_TRANSLATE);
 
   // Click the "Show original" button to revert translation.
   EXPECT_FALSE(mock_model_->revert_translation_called_);
   bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_SHOW_ORIGINAL);
   EXPECT_TRUE(mock_model_->revert_translation_called_);
 }
 
 TEST_F(TranslateBubbleViewTest, TryAgainButton) {
   bubble_->SwitchView(TranslateBubbleModel::VIEW_STATE_ERROR);
+  bubble_->model()->SetErrorType(TranslateErrors::NETWORK);
+
+  EXPECT_EQ(TranslateErrors::NETWORK, bubble_->model()->GetErrorType());
 
   // Click the "Try again" button to translate.
   EXPECT_FALSE(mock_model_->translate_called_);
   bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_TRY_AGAIN);
   EXPECT_TRUE(mock_model_->translate_called_);
 }
 
 TEST_F(TranslateBubbleViewTest, AlwaysTranslateCheckboxAndCancelButton) {
   bubble_->SwitchView(TranslateBubbleModel::VIEW_STATE_ADVANCED);
 
@@ skipping 83 matching lines @@
 
 TEST_F(TranslateBubbleViewTest, CancelButtonReturningError) {
   bubble_->SwitchView(TranslateBubbleModel::VIEW_STATE_ERROR);
   bubble_->SwitchView(TranslateBubbleModel::VIEW_STATE_ADVANCED);
 
   // Click the "Cancel" button to go back.
   EXPECT_EQ(TranslateBubbleModel::VIEW_STATE_ADVANCED, bubble_->GetViewState());
   bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_CANCEL);
   EXPECT_EQ(TranslateBubbleModel::VIEW_STATE_ERROR, bubble_->GetViewState());
 }