| // Copyright 2018 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 "ash/media/media_controller.h" |
| |
| #include <memory> |
| |
| #include "ash/session/session_controller.h" |
| #include "ash/shell.h" |
| #include "ash/test/ash_test_base.h" |
| #include "base/macros.h" |
| #include "services/media_session/public/cpp/test/test_media_controller.h" |
| |
| namespace ash { |
| |
| class MediaControllerTest : public AshTestBase { |
| public: |
| MediaControllerTest() = default; |
| ~MediaControllerTest() override = default; |
| |
| // AshTestBase |
| void SetUp() override { |
| AshTestBase::SetUp(); |
| |
| controller_ = std::make_unique<media_session::test::TestMediaController>(); |
| |
| MediaController* media_controller = Shell::Get()->media_controller(); |
| media_controller->SetMediaSessionControllerForTest( |
| controller_->CreateMediaControllerPtr()); |
| media_controller->FlushForTesting(); |
| |
| { |
| std::vector<media_session::mojom::MediaSessionAction> actions; |
| actions.push_back(media_session::mojom::MediaSessionAction::kPlay); |
| controller_->SimulateMediaSessionActionsChanged(actions); |
| } |
| |
| { |
| media_session::mojom::MediaSessionInfoPtr session_info( |
| media_session::mojom::MediaSessionInfo::New()); |
| |
| session_info->state = |
| media_session::mojom::MediaSessionInfo::SessionState::kActive; |
| session_info->playback_state = |
| media_session::mojom::MediaPlaybackState::kPlaying; |
| controller_->SimulateMediaSessionInfoChanged(std::move(session_info)); |
| } |
| |
| Flush(); |
| } |
| |
| media_session::test::TestMediaController* controller() const { |
| return controller_.get(); |
| } |
| |
| void SimulateSessionLock() { |
| mojom::SessionInfoPtr info_ptr = mojom::SessionInfo::New(); |
| info_ptr->state = session_manager::SessionState::LOCKED; |
| Shell::Get()->session_controller()->SetSessionInfo(std::move(info_ptr)); |
| } |
| |
| void Flush() { |
| controller_->Flush(); |
| Shell::Get()->media_controller()->FlushForTesting(); |
| } |
| |
| private: |
| std::unique_ptr<media_session::test::TestMediaController> controller_; |
| |
| DISALLOW_COPY_AND_ASSIGN(MediaControllerTest); |
| }; |
| |
| TEST_F(MediaControllerTest, DisablePlayPauseWhenLocked) { |
| EXPECT_EQ(0, controller()->suspend_count()); |
| |
| Shell::Get()->media_controller()->HandleMediaPlayPause(); |
| Flush(); |
| |
| EXPECT_EQ(1, controller()->suspend_count()); |
| |
| SimulateSessionLock(); |
| |
| Shell::Get()->media_controller()->HandleMediaPlayPause(); |
| Flush(); |
| |
| EXPECT_EQ(1, controller()->suspend_count()); |
| } |
| |
| TEST_F(MediaControllerTest, DisablePreviousTrackWhenLocked) { |
| EXPECT_EQ(0, controller()->previous_track_count()); |
| |
| Shell::Get()->media_controller()->HandleMediaPrevTrack(); |
| Flush(); |
| |
| EXPECT_EQ(1, controller()->previous_track_count()); |
| |
| SimulateSessionLock(); |
| |
| Shell::Get()->media_controller()->HandleMediaPrevTrack(); |
| Flush(); |
| |
| EXPECT_EQ(1, controller()->previous_track_count()); |
| } |
| |
| TEST_F(MediaControllerTest, DisableNextTrackWhenLocked) { |
| EXPECT_EQ(0, controller()->next_track_count()); |
| |
| Shell::Get()->media_controller()->HandleMediaNextTrack(); |
| Flush(); |
| |
| EXPECT_EQ(1, controller()->next_track_count()); |
| |
| SimulateSessionLock(); |
| |
| Shell::Get()->media_controller()->HandleMediaNextTrack(); |
| Flush(); |
| |
| EXPECT_EQ(1, controller()->next_track_count()); |
| } |
| |
| } // namespace ash |