Added unit test for WebShareTargetPickerView.

chrome/browser/ui/views/webshare/webshare_target_picker_view_unittest.cc

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