[Media>Session] Pause all players for non-routed frames when receiving PAUSE action

content/browser/media/session/media_session_impl_service_routing_unittest.cc

 // Copyright 2016 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 "content/browser/media/session/media_session_impl.h"
 
 #include <map>
 #include <memory>
 
 #include "base/memory/ptr_util.h"
+#include "base/run_loop.h"
 #include "base/strings/utf_string_conversions.h"
 #include "content/browser/media/session/media_session_player_observer.h"
 #include "content/browser/media/session/media_session_service_impl.h"
 #include "content/browser/media/session/mock_media_session_observer.h"
 #include "content/test/test_render_view_host.h"
 #include "content/test/test_web_contents.h"
 #include "media/base/media_content_type.h"
+#include "third_party/WebKit/public/platform/modules/mediasession/media_session.mojom.h"
 
 using ::testing::_;
 using ::testing::AnyNumber;
 using ::testing::Eq;
+using ::testing::InvokeWithoutArgs;
 
 namespace content {
 
 namespace {
 
 static const int kPlayerId = 0;
 
 class MockMediaSessionServiceImpl : public MediaSessionServiceImpl {
  public:
   explicit MockMediaSessionServiceImpl(RenderFrameHost* rfh)
       : MediaSessionServiceImpl(rfh) {}
   ~MockMediaSessionServiceImpl() override = default;
 };
 
+class MockMediaSessionClient : public blink::mojom::MediaSessionClient {
+ public:
+  MockMediaSessionClient()
+      : binding_(this) {}
+
+  blink::mojom::MediaSessionClientPtr CreateInterfacePtrAndBind() {
+    return binding_.CreateInterfacePtrAndBind();
+  }
+
+  MOCK_METHOD1(DidReceiveAction, void(blink::mojom::MediaSessionAction action));
+
+ private:
+  mojo::Binding<blink::mojom::MediaSessionClient> binding_;
+
+  DISALLOW_COPY_AND_ASSIGN(MockMediaSessionClient);
+};
+
 class MockMediaSessionPlayerObserver : public MediaSessionPlayerObserver {
  public:
   explicit MockMediaSessionPlayerObserver(RenderFrameHost* rfh)
       : render_frame_host_(rfh) {}
 
   ~MockMediaSessionPlayerObserver() override = default;
 
-  void OnSuspend(int player_id) override {}
-  void OnResume(int player_id) override {}
-  void OnSetVolumeMultiplier(int player_id, double volume_multiplier) override {
-  }
+  MOCK_METHOD1(OnSuspend, void(int player_id));
+  MOCK_METHOD1(OnResume, void(int player_id));
+  MOCK_METHOD2(OnSetVolumeMultiplier,
+               void(int player_id, double volume_multiplier));
+
   RenderFrameHost* GetRenderFrameHost() const override {
     return render_frame_host_;
   }
 
  private:
   RenderFrameHost* render_frame_host_;
 };
 
 }  // anonymous namespace
 
@@ skipping 13 matching lines @@
     sub_frame_ = main_frame_->AppendChild("sub_frame");
 
     player_in_main_frame_.reset(
         new MockMediaSessionPlayerObserver(main_frame_));
     player_in_sub_frame_.reset(new MockMediaSessionPlayerObserver(sub_frame_));
   }
 
   void TearDown() override {
     mock_media_session_observer_.reset();
     services_.clear();
+    clients_.clear();
 
     RenderViewHostImplTestHarness::TearDown();
   }
 
  protected:
   MockMediaSessionObserver* mock_media_session_observer() {
     return mock_media_session_observer_.get();
   }
 
   void CreateServiceForFrame(TestRenderFrameHost* frame) {
     services_[frame] = base::MakeUnique<MockMediaSessionServiceImpl>(frame);
+    clients_[frame] = base::MakeUnique<MockMediaSessionClient>();
+    services_[frame]->SetClient(clients_[frame]->CreateInterfacePtrAndBind());
   }
 
   void DestroyServiceForFrame(TestRenderFrameHost* frame) {
     services_.erase(frame);
+    clients_.erase(frame);
+  }
+
+  MockMediaSessionClient* GetClientForFrame(TestRenderFrameHost* frame) {
+    auto iter = clients_.find(frame);
+    return (iter != clients_.end()) ? iter->second.get() : nullptr;
   }
 
   void StartPlayerForFrame(TestRenderFrameHost* frame) {
     players_[frame] = base::MakeUnique<MockMediaSessionPlayerObserver>(frame);
     MediaSessionImpl::Get(contents())
         ->AddPlayer(players_[frame].get(), kPlayerId,
                     media::MediaContentType::Persistent);
   }
 
   void ClearPlayersForFrame(TestRenderFrameHost* frame) {
     if (!players_.count(frame))
       return;
 
     MediaSessionImpl::Get(contents())
         ->RemovePlayer(players_[frame].get(), kPlayerId);
   }
 
+  MockMediaSessionPlayerObserver* GetPlayerForFrame(
+      TestRenderFrameHost* frame) {
+    auto iter = players_.find(frame);
+    return (iter != players_.end()) ? iter->second.get() : nullptr;
+  }
+
   MediaSessionServiceImpl* ComputeServiceForRouting() {
     return MediaSessionImpl::Get(contents())->ComputeServiceForRouting();
   }
 
   TestRenderFrameHost* main_frame_;
   TestRenderFrameHost* sub_frame_;
 
   std::unique_ptr<MockMediaSessionPlayerObserver> player_in_main_frame_;
   std::unique_ptr<MockMediaSessionPlayerObserver> player_in_sub_frame_;
 
   std::unique_ptr<MockMediaSessionObserver> mock_media_session_observer_;
 
   using ServiceMap = std::map<TestRenderFrameHost*,
                               std::unique_ptr<MockMediaSessionServiceImpl>>;
   ServiceMap services_;
 
+  using ClientMap = std::map<TestRenderFrameHost*,
+                             std::unique_ptr<MockMediaSessionClient>>;
+  ClientMap clients_;
+
   using PlayerMap = std::map<TestRenderFrameHost*,
                              std::unique_ptr<MockMediaSessionPlayerObserver>>;
   PlayerMap players_;
 };
 
 TEST_F(MediaSessionImplServiceRoutingTest, NoFrameProducesAudio) {
   CreateServiceForFrame(main_frame_);
   CreateServiceForFrame(sub_frame_);
 
   ASSERT_EQ(nullptr, ComputeServiceForRouting());
@@ skipping 174 matching lines @@
 
   CreateServiceForFrame(main_frame_);
 
   services_[main_frame_]->SetMetadata(expected_metadata);
   services_[main_frame_]->EnableAction(blink::mojom::MediaSessionAction::PLAY);
 
   StartPlayerForFrame(main_frame_);
   ClearPlayersForFrame(main_frame_);
 }
 
+TEST_F(MediaSessionImplServiceRoutingTest,
+       TestPauseBehaviorWhenMainFrameIsRouted) {
+  base::RunLoop run_loop;
+
+  StartPlayerForFrame(main_frame_);
+  StartPlayerForFrame(sub_frame_);
+
+  CreateServiceForFrame(main_frame_);
+
+  EXPECT_CALL(*GetPlayerForFrame(sub_frame_), OnSuspend(_));
+  EXPECT_CALL(*GetClientForFrame(main_frame_),
+              DidReceiveAction(blink::mojom::MediaSessionAction::PAUSE))
+      .WillOnce(InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit));
+
+  services_[main_frame_]->EnableAction(blink::mojom::MediaSessionAction::PAUSE);
+
+  MediaSessionImpl::Get(contents())->DidReceiveAction(
+      blink::mojom::MediaSessionAction::PAUSE);
+
+  run_loop.Run();
+}
+
+TEST_F(MediaSessionImplServiceRoutingTest,
+       TestPauseBehaviorWhenSubFrameIsRouted) {
+  base::RunLoop run_loop;
+
+  StartPlayerForFrame(main_frame_);
+  StartPlayerForFrame(sub_frame_);
+
+  CreateServiceForFrame(sub_frame_);
+
+  EXPECT_CALL(*GetPlayerForFrame(main_frame_), OnSuspend(_));
+  EXPECT_CALL(*GetClientForFrame(sub_frame_),
+              DidReceiveAction(blink::mojom::MediaSessionAction::PAUSE))
+      .WillOnce(InvokeWithoutArgs(&run_loop, &base::RunLoop::Quit));
+
+  services_[sub_frame_]->EnableAction(blink::mojom::MediaSessionAction::PAUSE);
+
+  MediaSessionImpl::Get(contents())->DidReceiveAction(
+      blink::mojom::MediaSessionAction::PAUSE);
+
+  run_loop.Run();
+}
+
 }  // namespace content