[Cast,Android] Handle some PresentationAPI edge cases

chrome/browser/media/android/router/media_router_android_unittest.cc

Index: chrome/browser/media/android/router/media_router_android_unittest.cc
diff --git a/chrome/browser/media/android/router/media_router_android_unittest.cc b/chrome/browser/media/android/router/media_router_android_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..afcd557ccb0a3a2f3da5d39754b4ffdb3f4d648c
--- /dev/null
+++ b/chrome/browser/media/android/router/media_router_android_unittest.cc
@@ -0,0 +1,123 @@
+// 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 <memory>
+
+#include "base/test/mock_callback.h"
+#include "chrome/browser/media/android/router/media_router_android.h"
+#include "chrome/browser/media/android/router/media_router_android_bridge.h"
+#include "chrome/browser/media/router/test_helper.h"
+#include "content/public/browser/presentation_service_delegate.h"
+#include "content/public/common/presentation_info.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "url/origin.h"
+
+using testing::_;
+using testing::Expectation;
+using testing::Return;
+
+namespace media_router {
+
+class MockMediaRouterAndroidBridge : public MediaRouterAndroidBridge {
+ public:
+  MockMediaRouterAndroidBridge() : MediaRouterAndroidBridge(nullptr) {}
+  ~MockMediaRouterAndroidBridge() = default;
+
+  MOCK_METHOD7(CreateRoute,
+               void(const MediaSource::Id&,
+                    const MediaSink::Id&,
+                    const std::string&,
+                    const url::Origin&,
+                    int,
+                    bool,
+                    int));
+  MOCK_METHOD5(JoinRoute,
+               void(const MediaSource::Id&,
+                    const std::string&,
+                    const url::Origin&,
+                    int,
+                    int));
+  MOCK_METHOD1(TerminateRoute, void(const MediaRoute::Id&));
+  MOCK_METHOD3(SendRouteMessage,
+               void(const MediaRoute::Id&, const std::string&, int));
+  MOCK_METHOD1(DetachRoute, void(const MediaRoute::Id&));
+  MOCK_METHOD1(StartObservingMediaSinks, bool(const MediaSource::Id&));
+  MOCK_METHOD1(StopObservingMediaSinks, void(const MediaSource::Id&));
+};
+
+class MediaRouterAndroidTest : public testing::Test {
+ public:
+  void SetUp() override {
+    mock_bridge_ = new MockMediaRouterAndroidBridge();
+    router_.reset(new MediaRouterAndroid(nullptr));
+    router_->SetMediaRouterBridgeForTest(mock_bridge_);
+  }
+
+  void TearDown() override {}

[mlamouri (slow - plz ping), 2017/04/11 13:18:12]

nit: could you just drop this?

[whywhat, 2017/04/11 16:36:12]

Done.

+
+ protected:
+  std::unique_ptr<MediaRouterAndroid> router_;
+  MockMediaRouterAndroidBridge* mock_bridge_;
+};
+
+TEST_F(MediaRouterAndroidTest, DetachRoute) {
+  base::MockCallback<content::PresentationConnectionStateChangedCallback>
+      callback;
+  content::PresentationConnectionStateChangeInfo change_info_closed(
+      content::PRESENTATION_CONNECTION_STATE_CLOSED);
+  EXPECT_CALL(callback, Run(StateChangeInfoEquals(change_info_closed)));
+
+  Expectation createRouteExpectation =
+      EXPECT_CALL(*mock_bridge_, CreateRoute(_, _, _, _, _, _, 1))
+          .WillOnce(Return());
+  EXPECT_CALL(*mock_bridge_, DetachRoute("route"))
+      .After(createRouteExpectation)
+      .WillOnce(Return());
+
+  router_->CreateRoute("source", "sink", url::Origin(), nullptr,
+                       std::vector<MediaRouteResponseCallback>(),
+                       base::TimeDelta(), false);
+  router_->OnRouteCreated("route", "sink", 1, false);
+
+  EXPECT_NE(nullptr, router_->FindRouteBySource("source"));
+
+  std::unique_ptr<PresentationConnectionStateSubscription> subscription =
+      router_->AddPresentationConnectionStateChangedCallback("route",
+                                                             callback.Get());
+  router_->DetachRoute("route");
+
+  EXPECT_EQ(nullptr, router_->FindRouteBySource("source"));
+}
+
+TEST_F(MediaRouterAndroidTest, OnRouteClosed) {
+  base::MockCallback<content::PresentationConnectionStateChangedCallback>
+      callback;
+  content::PresentationConnectionStateChangeInfo change_info_closed(
+      content::PRESENTATION_CONNECTION_STATE_TERMINATED);
+  EXPECT_CALL(callback, Run(StateChangeInfoEquals(change_info_closed)));
+
+  Expectation createRouteExpectation =
+      EXPECT_CALL(*mock_bridge_, CreateRoute(_, _, _, _, _, _, 1))
+          .WillOnce(Return());
+  EXPECT_CALL(*mock_bridge_, DetachRoute("route"))
+      .After(createRouteExpectation)
+      .WillOnce(Return());
+
+  router_->CreateRoute("source", "sink", url::Origin(), nullptr,
+                       std::vector<MediaRouteResponseCallback>(),
+                       base::TimeDelta(), false);
+  router_->OnRouteCreated("route", "sink", 1, false);
+
+  EXPECT_NE(nullptr, router_->FindRouteBySource("source"));
+
+  std::unique_ptr<PresentationConnectionStateSubscription> subscription =
+      router_->AddPresentationConnectionStateChangedCallback("route",
+                                                             callback.Get());
+  router_->OnRouteClosed("route");
+
+  EXPECT_EQ(nullptr, router_->FindRouteBySource("source"));
+}
+
+}  // namespace media_router