Translate: Add TranslateBubbleView

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 "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:
+  MockTranslateBubbleModel(TranslateBubbleModel::ViewType view_type)
+      : TranslateBubbleModel(view_type),
+        original_language_index_(0),
+        target_language_index_(1),
+        is_language_blocked_(false),
+        is_site_blacklisted_(false),
+        should_always_translate_(false),
+        set_always_translate_called_count_(0),
+        translate_called_(false),
+        revert_translation_called_(false),
+        translation_declined_called_(false) {
+  }
+
+  virtual int GetNumberOfLanguages() const OVERRIDE {
+    return 1000;
+  }
+
+  virtual string16 GetLanguageNameAt(int index) const OVERRIDE {
+    return string16();
+  }
+
+  virtual int GetOriginalLanguageIndex() const OVERRIDE {
+    return original_language_index_;
+  }
+
+  virtual void SetOriginalLanguageIndex(int index) OVERRIDE {
+    original_language_index_ = index;
+  }
+
+  virtual int GetTargetLanguageIndex() const OVERRIDE {
+    return target_language_index_;
+  }
+
+  virtual void SetTargetLanguageIndex(int index) OVERRIDE {
+    target_language_index_ = index;
+  }
+
+  virtual void SetLanguageBlocked(bool value) OVERRIDE {
+    is_language_blocked_ = value;
+  }
+
+  virtual void SetSiteBlacklist(bool value) OVERRIDE {
+    is_site_blacklisted_ = value;
+  }
+
+  virtual bool ShouldAlwaysTranslate() const OVERRIDE {
+    return should_always_translate_;
+  }
+
+  virtual void SetAlwaysTranslate(bool value) OVERRIDE {
+    should_always_translate_ = value;
+    set_always_translate_called_count_++;
+  }
+
+  virtual void Translate() OVERRIDE {
+    translate_called_ = true;
+  }
+
+  virtual void RevertTranslation() OVERRIDE {
+    revert_translation_called_ = true;
+  }
+
+  virtual void TranslationDeclined() OVERRIDE {
+    translation_declined_called_ = true;
+  }
+
+  int original_language_index_;
+  int target_language_index_;
+  bool is_language_blocked_;
+  bool is_site_blacklisted_;
+  bool should_always_translate_;
+  int set_always_translate_called_count_;
+  bool translate_called_;
+  bool revert_translation_called_;
+  bool translation_declined_called_;
+};
+
+}  // namespace
+
+class TranslateBubbleViewTest : public views::ViewsTestBase {
+ public:
+  TranslateBubbleViewTest() {
+  }
+
+ protected:
+  virtual void SetUp() OVERRIDE {
+    views::ViewsTestBase::SetUp();
+
+    // The bubble needs the parent as an anchor.
+    views::Widget::InitParams params =
+        CreateParams(views::Widget::InitParams::TYPE_WINDOW);
+    params.ownership = views::Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET;
+
+    anchor_widget_.reset(new views::Widget());
+    anchor_widget_->Init(params);
+    anchor_widget_->Show();
+
+    mock_model_ = new MockTranslateBubbleModel(
+        TranslateBubbleModel::BEFORE_TRANSLATE);
+    scoped_ptr<TranslateBubbleModel> model(mock_model_);
+    bubble_ = new TranslateBubbleView(anchor_widget_->GetContentsView(),
+                                      model.Pass(),
+                                      false,
+                                      NULL);
+    views::BubbleDelegateView::CreateBubble(bubble_)->Show();
+  }
+
+  virtual void TearDown() OVERRIDE {
+    bubble_->GetWidget()->CloseNow();
+    anchor_widget_.reset();
+
+    views::ViewsTestBase::TearDown();
+  }
+
+  scoped_ptr<views::Widget> anchor_widget_;
+  MockTranslateBubbleModel* mock_model_;
+  TranslateBubbleView* bubble_;
+};
+
+TEST_F(TranslateBubbleViewTest, TranslateButton) {
+  EXPECT_FALSE(mock_model_->translate_called_);
+
+  // Press the "Translate" button.
+  bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_TRANSLATE);
+  EXPECT_TRUE(mock_model_->translate_called_);
+}
+
+TEST_F(TranslateBubbleViewTest, AdvancedLink) {
+  EXPECT_EQ(TranslateBubbleModel::BEFORE_TRANSLATE, bubble_->view_type());
+
+  // Click the "Advanced" link.
+  bubble_->HandleLinkClicked(TranslateBubbleView::LINK_ID_ADVANCED);
+  EXPECT_EQ(TranslateBubbleModel::ADVANCED, bubble_->view_type());
+}
+
+TEST_F(TranslateBubbleViewTest, ShowOriginalButton) {
+  bubble_->SwitchView(TranslateBubbleModel::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::ERROR);
+
+  // 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, AlwaysTranslateCheckbox) {
+  bubble_->SwitchView(TranslateBubbleModel::ADVANCED);
+
+  // Click the "Always Translate" checkbox. Changing the state of this checkbox
+  // should affect the model immediately.
+
+  // Check the initial state.
+  EXPECT_FALSE(mock_model_->should_always_translate_);
+  EXPECT_EQ(0, mock_model_->set_always_translate_called_count_);
+  EXPECT_FALSE(bubble_->always_translate_checkbox_->checked());
+
+  // Click the checkbox.
+  bubble_->always_translate_checkbox_->SetChecked(true);
+  bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_ALWAYS_TRANSLATE);
+  EXPECT_TRUE(mock_model_->should_always_translate_);
+  EXPECT_EQ(1, mock_model_->set_always_translate_called_count_);
+
+  // Click this again and check the state is reverted.
+  bubble_->always_translate_checkbox_->SetChecked(false);
+  bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_ALWAYS_TRANSLATE);
+  EXPECT_FALSE(mock_model_->should_always_translate_);
+  EXPECT_EQ(2, mock_model_->set_always_translate_called_count_);
+}
+
+TEST_F(TranslateBubbleViewTest, DoneButton) {
+  bubble_->SwitchView(TranslateBubbleModel::ADVANCED);
+
+  // Click the "Done" button to translate. The selected languages by the user
+  // are applied.
+  EXPECT_FALSE(mock_model_->translate_called_);
+  bubble_->source_language_combobox_->SetSelectedIndex(10);
+  bubble_->target_language_combobox_->SetSelectedIndex(20);
+  bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_DONE);
+  EXPECT_TRUE(mock_model_->translate_called_);
+  EXPECT_EQ(10, mock_model_->original_language_index_);
+  EXPECT_EQ(20, mock_model_->target_language_index_);
+}
+
+TEST_F(TranslateBubbleViewTest, CancelButtonReturningBeforeTranslate) {
+  bubble_->SwitchView(TranslateBubbleModel::BEFORE_TRANSLATE);
+  bubble_->SwitchView(TranslateBubbleModel::ADVANCED);
+
+  // Click the "Cancel" button to go back.
+  EXPECT_EQ(TranslateBubbleModel::ADVANCED, bubble_->view_type());
+  bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_CANCEL);
+  EXPECT_EQ(TranslateBubbleModel::BEFORE_TRANSLATE, bubble_->view_type());
+}
+
+TEST_F(TranslateBubbleViewTest, CancelButtonReturningAfterTranslate) {
+  bubble_->SwitchView(TranslateBubbleModel::AFTER_TRANSLATE);
+  bubble_->SwitchView(TranslateBubbleModel::ADVANCED);
+
+  // Click the "Cancel" button to go back.
+  EXPECT_EQ(TranslateBubbleModel::ADVANCED, bubble_->view_type());
+  bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_CANCEL);
+  EXPECT_EQ(TranslateBubbleModel::AFTER_TRANSLATE, bubble_->view_type());
+}
+
+TEST_F(TranslateBubbleViewTest, CancelButtonReturningError) {
+  bubble_->SwitchView(TranslateBubbleModel::ERROR);
+  bubble_->SwitchView(TranslateBubbleModel::ADVANCED);
+
+  // Click the "Cancel" button to go back.
+  EXPECT_EQ(TranslateBubbleModel::ADVANCED, bubble_->view_type());
+  bubble_->HandleButtonPressed(TranslateBubbleView::BUTTON_ID_CANCEL);
+  EXPECT_EQ(TranslateBubbleModel::ERROR, bubble_->view_type());
+}