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..9895e452c27a6ed34636ce006e603c7de051a235
--- /dev/null
+++ b/chrome/browser/ui/views/webshare/webshare_target_picker_view_unittest.cc
@@ -0,0 +1,102 @@
+// 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 <vector>
+
+#include "base/bind.h"
+#include "base/callback.h"
+#include "base/run_loop.h"
+#include "base/strings/string16.h"
+#include "base/strings/utf_string_conversions.h"
+#include "chrome/browser/ui/views/webshare/webshare_target_picker_view.h"

[sky, 2017/02/06 17:11:24]

This should be your first include (just like web_share_target_picker_view.cc).

[Matt Giuca, 2017/02/06 23:17:22]

Done.

+#include "chrome/test/base/browser_with_test_window_test.h"
+#include "chrome/test/base/chrome_render_view_host_test_harness.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "ui/views/controls/table/table_view.h"
+#include "ui/views/window/dialog_client_view.h"
+#include "ui/views/window/dialog_delegate.h"
+
+namespace {
+
+class WebShareTargetPickerViewTest : public ChromeRenderViewHostTestHarness {

[sky, 2017/02/06 17:11:24]

Based on capitalization all of the files in this directory should be web_share,
not webshare_.

[Matt Giuca, 2017/02/06 23:17:22]

Hmm, that's a good point, but we have directories and files called "webshare" in
a LOT of places (c/b/webshare, c/b/ui/views/webshare,
WebKit/public/platform/modules/webshare,
android/java/src/org/chromium/chrome/browser/webshare/, etc, plus all the files
within). I think for consistency that ship has sailed, though perhaps at some
point we should rename all the classes to Webshare instead of WebShare.

Can we not do this now? Connie and myself both have a few CLs out that will
conflict if we make major changes.

+ public:
+  void SetUp() override {
+    ChromeRenderViewHostTestHarness::SetUp();
+  }
+
+  void TearDown() override {
+    ChromeRenderViewHostTestHarness::TearDown();
+    view_.reset();
+    quit_closure_ = base::Closure();
+  }
+
+ protected:
+  // Creates the WebShareTargetPickerView (available as view()).
+  void CreateView(const std::vector<base::string16>& targets) {
+    view_.reset(new WebShareTargetPickerView(
+        targets, base::Bind(&WebShareTargetPickerViewTest::callback,
+                            base::Unretained(this))));
+  }
+
+  // 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_.get(); }
+  // The table inside the view (for inspection).
+  views::TableView* table() { return view_->table_; }
+
+  // The result that was returned to the dialog's callback.
+  SharePickerResult result() { return result_; }
+
+ private:
+  void callback(SharePickerResult result) {

[sky, 2017/02/06 17:11:24]

OnCallback?

[Matt Giuca, 2017/02/06 23:17:21]

Done.

+    result_ = result;
+    if (quit_closure_)
+      quit_closure_.Run();
+  }
+
+  std::unique_ptr<WebShareTargetPickerView> view_;
+
+  SharePickerResult result_;
+
+  base::Closure quit_closure_;
+};

[sky, 2017/02/06 17:11:24]

DISALLOW...

[Matt Giuca, 2017/02/06 23:17:22]

Done.

+
+// Table with 0 targets. Choose to cancel.
+TEST_F(WebShareTargetPickerViewTest, EmptyListCancel) {
+  CreateView(std::vector<base::string16>());
+  EXPECT_EQ(0, table()->RowCount());
+
+  view()->Cancel();
+
+  base::RunLoop run_loop;
+  SetQuitClosure(run_loop.QuitClosure());
+
+  EXPECT_EQ(SharePickerResult::CANCEL, result());
+}
+
+// Table with 2 targets. Choose second target and share.
+TEST_F(WebShareTargetPickerViewTest, ChooseItem) {
+  std::vector<base::string16> targets{base::ASCIIToUTF16("App One"),
+                                      base::ASCIIToUTF16("App Two")};
+  CreateView(targets);
+  EXPECT_EQ(2, table()->RowCount());
+  EXPECT_EQ(base::ASCIIToUTF16("App One"), table()->model()->GetText(0, 0));
+  EXPECT_EQ(base::ASCIIToUTF16("App Two"), table()->model()->GetText(1, 0));
+
+  // Choose the second app.
+  table()->Select(1);
+  view()->Accept();
+
+  base::RunLoop run_loop;
+  SetQuitClosure(run_loop.QuitClosure());
+
+  EXPECT_EQ(SharePickerResult::SHARE, result());
+}
+
+}  // namespace