Index: chrome/browser/ui/views/webshare/webshare_target_picker_view_unittest.cc |
diff --git a/chrome/browser/ui/views/webshare/webshare_target_picker_view_unittest.cc b/chrome/browser/ui/views/webshare/webshare_target_picker_view_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..a8d0478933ce2529cd5b6aacee956db940d20856 |
--- /dev/null |
+++ b/chrome/browser/ui/views/webshare/webshare_target_picker_view_unittest.cc |
@@ -0,0 +1,173 @@ |
+// Copyright 2017 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/webshare/webshare_target_picker_view.h" |
+ |
+#include <string> |
+#include <vector> |
+ |
+#include "base/bind.h" |
+#include "base/callback.h" |
+#include "base/optional.h" |
+#include "base/run_loop.h" |
+#include "base/strings/string16.h" |
+#include "base/strings/utf_string_conversions.h" |
+#include "chrome/browser/ui/views/chrome_constrained_window_views_client.h" |
+#include "chrome/test/base/browser_with_test_window_test.h" |
+#include "components/constrained_window/constrained_window_views.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+#include "ui/views/controls/table/table_view.h" |
+#include "ui/views/test/views_test_base.h" |
+#include "ui/views/widget/widget.h" |
+#include "ui/views/window/dialog_client_view.h" |
+#include "ui/views/window/dialog_delegate.h" |
+#include "url/gurl.h" |
+ |
+namespace { |
+ |
+class WebShareTargetPickerViewTest : public views::ViewsTestBase { |
+ public: |
+ WebShareTargetPickerViewTest() {} |
+ |
+ void SetUp() override { |
+ ViewsTestBase::SetUp(); |
+ |
+ SetConstrainedWindowViewsClient(CreateChromeConstrainedWindowViewsClient()); |
+ |
+ // Create the parent widget. |
+ views::Widget::InitParams params = |
+ CreateParams(views::Widget::InitParams::TYPE_WINDOW); |
+ params.ownership = views::Widget::InitParams::WIDGET_OWNS_NATIVE_WIDGET; |
+ |
+ parent_widget_.reset(new views::Widget()); |
+ parent_widget_->Init(params); |
+ parent_widget_->Show(); |
+ } |
+ |
+ void TearDown() override { |
+ if (view_) |
+ view_->GetWidget()->CloseNow(); |
+ parent_widget_->CloseNow(); |
+ quit_closure_ = base::Closure(); |
+ constrained_window::SetConstrainedWindowViewsClient(nullptr); |
+ |
+ ViewsTestBase::TearDown(); |
+ } |
+ |
+ protected: |
+ // Creates the WebShareTargetPickerView (available as view()). |
+ void CreateView(const std::vector<std::pair<base::string16, GURL>>& targets) { |
+ view_ = new WebShareTargetPickerView( |
+ targets, base::Bind(&WebShareTargetPickerViewTest::OnCallback, |
+ base::Unretained(this))); |
+ constrained_window::CreateBrowserModalDialogViews( |
+ view_, parent_widget_->GetNativeWindow()) |
+ ->Show(); |
+ } |
+ |
+ // Sets the closure that will be called when the dialog is closed. This is |
+ // used in tests to quit the RunLoop. |
+ void SetQuitClosure(base::Closure&& quit_closure) { |
+ quit_closure_ = std::move(quit_closure); |
+ } |
+ |
+ // The view under test. |
+ WebShareTargetPickerView* view() { return view_; } |
+ // The table inside the view (for inspection). |
+ views::TableView* table() { return view_->table_; } |
+ |
+ // The result that was returned to the dialog's callback. |
+ const base::Optional<std::string>& result() { return result_; } |
+ |
+ private: |
+ void OnCallback(base::Optional<std::string> result) { |
+ result_ = result; |
+ if (quit_closure_) |
+ quit_closure_.Run(); |
+ } |
+ |
+ std::unique_ptr<views::Widget> parent_widget_; |
+ WebShareTargetPickerView* view_ = nullptr; |
+ |
+ base::Optional<std::string> result_; |
+ |
+ base::Closure quit_closure_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(WebShareTargetPickerViewTest); |
+}; |
+ |
+// Table with 0 targets. Choose to cancel. |
+TEST_F(WebShareTargetPickerViewTest, EmptyListCancel) { |
+ CreateView(std::vector<std::pair<base::string16, GURL>>()); |
+ EXPECT_EQ(0, table()->RowCount()); |
+ EXPECT_EQ(-1, table()->FirstSelectedRow()); // Nothing selected. |
+ EXPECT_FALSE(view()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK)); |
+ |
+ base::RunLoop run_loop; |
+ SetQuitClosure(run_loop.QuitClosure()); |
+ |
+ view()->Cancel(); |
+ |
+ run_loop.Run(); |
+ |
+ EXPECT_EQ(base::nullopt, result()); |
+} |
+ |
+// Table with 2 targets. Choose second target and share. |
+TEST_F(WebShareTargetPickerViewTest, ChooseItem) { |
+ std::vector<std::pair<base::string16, GURL>> targets{ |
+ std::make_pair(base::ASCIIToUTF16("App One"), |
+ GURL("https://appone.com/path/bits")), |
+ std::make_pair(base::ASCIIToUTF16("App Two"), |
+ GURL("https://apptwo.xyz"))}; |
+ CreateView(targets); |
+ EXPECT_EQ(2, table()->RowCount()); |
+ EXPECT_EQ(base::ASCIIToUTF16("App One (https://appone.com/)"), |
+ table()->model()->GetText(0, 0)); |
+ EXPECT_EQ(base::ASCIIToUTF16("App Two (https://apptwo.xyz/)"), |
+ table()->model()->GetText(1, 0)); |
+ EXPECT_EQ(0, table()->FirstSelectedRow()); |
+ EXPECT_TRUE(view()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK)); |
+ |
+ // Deselect and ensure OK button is disabled. |
+ table()->Select(-1); |
+ EXPECT_FALSE(view()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK)); |
+ |
+ // Select the second app and ensure the OK button is enabled. |
+ table()->Select(1); |
+ EXPECT_TRUE(view()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK)); |
+ |
+ base::RunLoop run_loop; |
+ SetQuitClosure(run_loop.QuitClosure()); |
+ |
+ view()->Accept(); |
+ |
+ run_loop.Run(); |
+ |
+ EXPECT_EQ(base::Optional<std::string>("https://apptwo.xyz/"), result()); |
+} |
+ |
+// Table with 1 target. Select using double-click. |
+TEST_F(WebShareTargetPickerViewTest, ChooseItemWithDoubleClick) { |
+ std::vector<std::pair<base::string16, GURL>> targets{std::make_pair( |
+ base::ASCIIToUTF16("App One"), GURL("https://appone.com/path/bits"))}; |
+ CreateView(targets); |
+ EXPECT_EQ(1, table()->RowCount()); |
+ EXPECT_EQ(base::ASCIIToUTF16("App One (https://appone.com/)"), |
+ table()->model()->GetText(0, 0)); |
+ EXPECT_EQ(0, table()->FirstSelectedRow()); |
+ EXPECT_TRUE(view()->IsDialogButtonEnabled(ui::DIALOG_BUTTON_OK)); |
+ |
+ base::RunLoop run_loop; |
+ SetQuitClosure(run_loop.QuitClosure()); |
+ |
+ view()->OnDoubleClick(); |
+ |
+ run_loop.Run(); |
+ |
+ EXPECT_EQ(base::Optional<std::string>("https://appone.com/path/bits"), |
+ result()); |
+} |
+ |
+} // namespace |