Added InvokeUI browser tests for CastDialogView

Bug: 1060448
Change-Id: I04de559f42ed31d159e93f55b59940fd15d1721e
Reviewed-on: https://chromium-review.googlesource.com/c/chromium/src/+/2132635Reviewed-by: Takumi Fujimoto <takumif@chromium.org>
Commit-Queue: Thomas Lukaszewicz <tluk@chromium.org>
Cr-Commit-Position: refs/heads/master@{#756844}

chrome/browser/ui/views/media_router/cast_dialog_view_browsertest.cc

+// Copyright 2020 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/media_router/cast_dialog_view.h"
+
+#include "chrome/browser/ui/media_router/cast_dialog_controller.h"
+#include "chrome/browser/ui/media_router/cast_dialog_model.h"
+#include "chrome/browser/ui/media_router/ui_media_sink.h"
+#include "chrome/browser/ui/test/test_browser_dialog.h"
+#include "chrome/browser/ui/views/frame/browser_view.h"
+
+namespace {
+
+media_router::UIMediaSink CreateAvailableSink() {
+  media_router::UIMediaSink sink;
+  sink.id = "sink_available";
+  sink.friendly_name = base::UTF8ToUTF16("TestAvailableSink");
+  sink.state = media_router::UIMediaSinkState::AVAILABLE;
+  sink.cast_modes = {media_router::TAB_MIRROR, media_router::DESKTOP_MIRROR};
+  return sink;
+}
+
+media_router::UIMediaSink CreateConnectedSink() {
+  media_router::UIMediaSink sink;
+  sink.id = "sink_connected";
+  sink.friendly_name = base::UTF8ToUTF16("TestConnectedSink");
+  sink.state = media_router::UIMediaSinkState::CONNECTED;
+  sink.cast_modes = {media_router::TAB_MIRROR, media_router::DESKTOP_MIRROR};
+  sink.route = media_router::MediaRoute(
+      "route_id", media_router::MediaSource("https://example.com"), sink.id, "",
+      true, true);
+  return sink;
+}
+
+media_router::UIMediaSink CreateUnavailableSink() {
+  media_router::UIMediaSink sink;
+  sink.id = "sink_unavailable";
+  sink.friendly_name = base::UTF8ToUTF16("TestUnavailableSink");
+  sink.state = media_router::UIMediaSinkState::UNAVAILABLE;
+  sink.cast_modes = {media_router::TAB_MIRROR, media_router::DESKTOP_MIRROR};
+  return sink;
+}
+
+media_router::CastDialogModel CreateModelWithSinks(
+    std::vector<media_router::UIMediaSink> sinks) {
+  media_router::CastDialogModel model;
+  model.set_dialog_header(base::UTF8ToUTF16("Dialog header"));
+  model.set_media_sinks(std::move(sinks));
+  return model;
+}
+
+class MockCastDialogController : public media_router::CastDialogController {
+ public:
+  void AddObserver(Observer* observer) override {}
+  void RemoveObserver(Observer* observer) override {}
+  void StartCasting(const media_router::MediaSink::Id& sink_id,
+                    media_router::MediaCastMode cast_mode) override {}
+  void StopCasting(const media_router::MediaRoute::Id& route_id) override {}
+  void ChooseLocalFile(
+      base::OnceCallback<void(const ui::SelectedFileInfo*)> callback) override {
+  }
+  void ClearIssue(const media_router::Issue::Id& issue_id) override {}
+};
+
+}  // namespace
+
+class CastDialogViewBrowserTest : public DialogBrowserTest {
+ public:
+  CastDialogViewBrowserTest()
+      : controller_(std::make_unique<MockCastDialogController>()) {}
+
+  CastDialogViewBrowserTest(const CastDialogViewBrowserTest&) = delete;
+  CastDialogViewBrowserTest& operator=(const CastDialogViewBrowserTest&) =
+      delete;
+
+  // DialogBrowserTest:
+  void PreShow() override {
+    media_router::CastDialogView::ShowDialogCenteredForBrowserWindow(
+        controller_.get(), browser(), base::Time::Now(),
+        media_router::MediaRouterDialogOpenOrigin::TOOLBAR);
+  }
+
+  void ShowUi(const std::string& name) override {
+    media_router::CastDialogModel model;
+    if (name == "Available") {
+      model = CreateModelWithSinks({CreateAvailableSink()});
+    } else if (name == "Connected") {
+      model = CreateModelWithSinks({CreateConnectedSink()});
+    } else if (name == "Unavailable") {
+      model = CreateModelWithSinks({CreateUnavailableSink()});
+    } else if (name == "Mixed") {
+      model = CreateModelWithSinks({
+          CreateAvailableSink(),
+          CreateConnectedSink(),
+          CreateUnavailableSink(),
+      });
+    }
+    media_router::CastDialogView* dialog =
+        media_router::CastDialogView::GetInstance();
+    dialog->OnModelUpdated(model);
+  }
+
+ private:
+  std::unique_ptr<MockCastDialogController> controller_;
+};
+
+IN_PROC_BROWSER_TEST_F(CastDialogViewBrowserTest, InvokeUi_Available) {
+  ShowAndVerifyUi();
+}
+
+IN_PROC_BROWSER_TEST_F(CastDialogViewBrowserTest, InvokeUi_Connected) {
+  ShowAndVerifyUi();
+}
+
+IN_PROC_BROWSER_TEST_F(CastDialogViewBrowserTest, InvokeUi_Unavailable) {
+  ShowAndVerifyUi();
+}
+
+IN_PROC_BROWSER_TEST_F(CastDialogViewBrowserTest, InvokeUi_Mixed) {
+  ShowAndVerifyUi();
+}
+
+IN_PROC_BROWSER_TEST_F(CastDialogViewBrowserTest, InvokeUi_NoSinks) {
+  ShowAndVerifyUi();
+}

chrome/test/BUILD.gn

@@ -1949,6 +1949,7 @@ if (!is_android) {
         "../browser/ui/views/location_bar/location_bar_view_browsertest.cc",
         "../browser/ui/views/location_bar/location_icon_view_browsertest.cc",
         "../browser/ui/views/location_bar/zoom_bubble_view_browsertest.cc",
+        "../browser/ui/views/media_router/cast_dialog_view_browsertest.cc",
         "../browser/ui/views/media_router/media_router_dialog_controller_views_browsertest.cc",
         "../browser/ui/views/media_router/media_router_ui_browsertest.cc",
         "../browser/ui/views/media_router/presentation_receiver_window_view_browsertest.cc",