[Media Router] Custom Controls 1 - Add MediaStatus, MediaRouteController, and mojo interfaces

chrome/browser/media/router/mojo/media_route_controller_unittest.cc

Index: chrome/browser/media/router/mojo/media_route_controller_unittest.cc
diff --git a/chrome/browser/media/router/mojo/media_route_controller_unittest.cc b/chrome/browser/media/router/mojo/media_route_controller_unittest.cc
new file mode 100644
index 0000000000000000000000000000000000000000..b0167b79eef29e93dda83e4ea6bc50007fb5fd12
--- /dev/null
+++ b/chrome/browser/media/router/mojo/media_route_controller_unittest.cc
@@ -0,0 +1,136 @@
+// 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/media/router/mojo/media_route_controller.h"
+
+#include <string>
+#include <utility>
+
+#include "base/run_loop.h"
+#include "base/time/time.h"
+#include "chrome/browser/media/router/media_route_status_observer.h"
+#include "chrome/browser/media/router/mojo/media_router_mojo_test.h"
+#include "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using ::testing::_;
+using ::testing::Invoke;
+using ::testing::StrictMock;
+
+namespace media_router {
+
+class MockMediaController : public mojom::MediaController {
+ public:
+  MOCK_METHOD0(Play, void());
+  MOCK_METHOD0(Pause, void());
+  MOCK_METHOD1(SetMute, void(bool mute));
+  MOCK_METHOD1(SetVolume, void(float volume));
+  MOCK_METHOD1(Seek, void(base::TimeDelta time));
+};
+
+class MockMediaRouteStatusObserver : public MediaRouteStatusObserver {
+ public:
+  MOCK_METHOD1(OnMediaStatusUpdated, void(const MediaStatus& status));
+};
+
+class MediaRouteControllerUnitTest : public ::testing::Test {
+ public:
+  MediaRouteControllerUnitTest() {}
+  ~MediaRouteControllerUnitTest() override {}
+
+  void SetUp() override {
+    mojom::MediaControllerPtr media_controller_ptr;
+    mock_media_controller_ = base::MakeUnique<MockMediaController>();
+    media_controller_binding_ =
+        base::MakeUnique<mojo::Binding<mojom::MediaController>>(
+            mock_media_controller_.get(),
+            mojo::MakeRequest(&media_controller_ptr));
+
+    controller_ = base::MakeUnique<MediaRouteController>(
+        std::move(media_controller_ptr),
+        base::Bind(&MediaRouteControllerUnitTest::DestroyController,
+                   base::Unretained(this)));
+  }
+
+  void DestroyController() {
+    controller_.reset();
+    OnDestroyController();
+  }
+  MOCK_METHOD0(OnDestroyController, void());
+
+ protected:
+  std::unique_ptr<MockMediaController> mock_media_controller_;
+  std::unique_ptr<mojo::Binding<mojom::MediaController>>
+      media_controller_binding_;
+  std::unique_ptr<MediaRouteController> controller_;
+
+  content::TestBrowserThreadBundle test_thread_bundle_;
+
+ private:
+  DISALLOW_COPY_AND_ASSIGN(MediaRouteControllerUnitTest);
+};
+
+TEST_F(MediaRouteControllerUnitTest, ForwardControllerCommands) {
+  const float volume = 0.5;
+  const base::TimeDelta time = base::TimeDelta::FromSeconds(42);
+
+  EXPECT_CALL(*mock_media_controller_, Play());
+  controller_->Play();
+  EXPECT_CALL(*mock_media_controller_, Pause());
+  controller_->Pause();
+  EXPECT_CALL(*mock_media_controller_, SetMute(true));
+  controller_->SetMute(true);
+  EXPECT_CALL(*mock_media_controller_, SetVolume(volume));
+  controller_->SetVolume(volume);
+  EXPECT_CALL(*mock_media_controller_, Seek(time));
+  controller_->Seek(time);
+}
+
+TEST_F(MediaRouteControllerUnitTest, NotifyMediaRouteStatusObservers) {
+  StrictMock<MockMediaRouteStatusObserver> observer1;
+  StrictMock<MockMediaRouteStatusObserver> observer2;
+  StrictMock<MockMediaRouteStatusObserver> observer3;
+
+  MediaStatus status;
+  status.title = "test media status";
+
+  controller_->AddObserver(&observer1);
+  controller_->AddObserver(&observer2);
+
+  // Get a mojo pointer for |controller_|, so that we can notify it of status
+  // updates via mojo.
+  mojom::MediaStatusObserverPtr mojo_observer = controller_->GetObserverPtr();
+
+  base::RunLoop run_loop1;
+  EXPECT_CALL(observer1, OnMediaStatusUpdated(status));
+  EXPECT_CALL(observer2, OnMediaStatusUpdated(status));
+  mojo_observer->OnMediaStatusUpdated(status);
+  run_loop1.RunUntilIdle();
+
+  controller_->RemoveObserver(&observer1);
+  controller_->AddObserver(&observer3);
+
+  base::RunLoop run_loop2;
+  EXPECT_CALL(observer2, OnMediaStatusUpdated(status));
+  EXPECT_CALL(observer3, OnMediaStatusUpdated(status));
+  mojo_observer->OnMediaStatusUpdated(status);
+  run_loop2.RunUntilIdle();
+}
+
+TEST_F(MediaRouteControllerUnitTest, DestroyControllerOnNoObservers) {
+  MockMediaRouteStatusObserver observer1;
+  MockMediaRouteStatusObserver observer2;
+  controller_->AddObserver(&observer1);
+  controller_->AddObserver(&observer2);
+
+  EXPECT_CALL(*this, OnDestroyController()).Times(0);
+  controller_->RemoveObserver(&observer1);
+
+  // OnDestroyController() should be called when |controller_| no longer has any
+  // observers.
+  EXPECT_CALL(*this, OnDestroyController()).Times(1);
+  controller_->RemoveObserver(&observer2);
+}
+
+}  // namespace media_router