Letting Flash join MediaSession (stack implementaion)

content/browser/media/session/audio_focus_manager_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/audio_focus_manager.h"
 
 #include "content/browser/media/session/media_session.h"
 #include "content/public/test/mock_render_process_host.h"
 #include "content/public/test/test_browser_context.h"
 #include "content/public/test/test_browser_thread.h"
@@ skipping 14 matching lines @@
     browser_context_.reset(new TestBrowserContext());
   }
 
   void TearDown() override {
     browser_context_.reset();
     SiteInstanceImpl::set_render_process_host_factory(nullptr);
     rph_factory_.reset();
   }
 
   WebContents* GetAudioFocusedContent() const {
-    if (!AudioFocusManager::GetInstance()->focus_entry_)
-      return nullptr;
-    return AudioFocusManager::GetInstance()->focus_entry_->web_contents();
+    const auto& audio_focus_stack =
+        AudioFocusManager::GetInstance()->audio_focus_stack_;
+    for (auto iter = audio_focus_stack.rbegin();
+         iter != audio_focus_stack.rend(); ++iter) {
+      if ((*iter)->type() == AudioFocusManager::AudioFocusType::Gain)
+        return (*iter)->web_contents();
+    }
+    return nullptr;
   }
 
   int GetTransientMaybeDuckCount() const {
-    return AudioFocusManager::GetInstance()->TransientMayDuckEntriesCount();
+    int count = 0;
+    const auto& audio_focus_stack =
+        AudioFocusManager::GetInstance()->audio_focus_stack_;
+    for (auto iter = audio_focus_stack.rbegin();
+         iter != audio_focus_stack.rend(); ++iter) {
+      if ((*iter)->type() ==
+          AudioFocusManager::AudioFocusType::GainTransientMayDuck) {
+          ++count;
+      } else {
+        break;
+      }
+    }
+
+    return count;
   }
 
   double IsSessionDucking(MediaSession* session) {
     return session->is_ducking_;  // Quack! Quack!
   }
 
+  void SetAudioFocusState(MediaSession* session,
+                          MediaSession::State audio_focus_state) {
+    session->audio_focus_state_ = audio_focus_state;
+  }
+
   WebContents* CreateWebContents() {
     return TestWebContents::Create(browser_context_.get(),
         SiteInstance::SiteInstance::Create(browser_context_.get()));
   }
 
  private:
   base::MessageLoopForUI message_loop_;
   TestBrowserThread ui_thread_;
 
   std::unique_ptr<MockRenderProcessHostFactory> rph_factory_;
@@ skipping 191 matching lines @@
 
   AudioFocusManager::GetInstance()->RequestAudioFocus(
       media_session_1, AudioFocusManager::AudioFocusType::Gain);
   ASSERT_FALSE(IsSessionDucking(media_session_1));
 
   AudioFocusManager::GetInstance()->RequestAudioFocus(
       media_session_2, AudioFocusManager::AudioFocusType::GainTransientMayDuck);
   ASSERT_TRUE(IsSessionDucking(media_session_1));
 }
 
-TEST_F(AudioFocusManagerTest, DuckWhenStarting) {
+TEST_F(AudioFocusManagerTest, GainSuspendsTransient) {
   std::unique_ptr<WebContents> web_contents_1(CreateWebContents());
   MediaSession* media_session_1 = MediaSession::Get(web_contents_1.get());
 
   std::unique_ptr<WebContents> web_contents_2(CreateWebContents());
   MediaSession* media_session_2 = MediaSession::Get(web_contents_2.get());
 
+  SetAudioFocusState(media_session_2, MediaSession::State::ACTIVE);
+
   AudioFocusManager::GetInstance()->RequestAudioFocus(
       media_session_2, AudioFocusManager::AudioFocusType::GainTransientMayDuck);
 
   AudioFocusManager::GetInstance()->RequestAudioFocus(
       media_session_1, AudioFocusManager::AudioFocusType::Gain);
-  ASSERT_TRUE(IsSessionDucking(media_session_1));
+  ASSERT_TRUE(media_session_2->IsSuspended());
+
+  // Clean up
+  SetAudioFocusState(media_session_2, MediaSession::State::INACTIVE);
 }
 
 TEST_F(AudioFocusManagerTest, DuckWithMultipleTransients) {
   std::unique_ptr<WebContents> web_contents_1(CreateWebContents());
   MediaSession* media_session_1 = MediaSession::Get(web_contents_1.get());
 
   std::unique_ptr<WebContents> web_contents_2(CreateWebContents());
   MediaSession* media_session_2 = MediaSession::Get(web_contents_2.get());
 
   std::unique_ptr<WebContents> web_contents_3(CreateWebContents());
@@ skipping 55 matching lines @@
 
   AudioFocusManager::GetInstance()->RequestAudioFocus(
       media_session_2, AudioFocusManager::AudioFocusType::GainTransientMayDuck);
   ASSERT_TRUE(IsSessionDucking(media_session_1));
 
   web_contents_2.reset();
   ASSERT_FALSE(IsSessionDucking(media_session_1));
 }
 
 }  // namespace content