| // Copyright (c) 2012 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/renderer/media/stream/media_stream_device_observer.h" |
| |
| #include <stddef.h> |
| |
| #include <memory> |
| #include <string> |
| #include <utility> |
| |
| #include "base/run_loop.h" |
| #include "base/test/scoped_task_environment.h" |
| #include "content/public/common/media_stream_request.h" |
| #include "content/renderer/media/stream/mock_mojo_media_stream_dispatcher_host.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace content { |
| |
| class MediaStreamDeviceObserverTest : public ::testing::Test { |
| public: |
| MediaStreamDeviceObserverTest() |
| : observer_(std::make_unique<MediaStreamDeviceObserver>(nullptr)) {} |
| |
| void OnDeviceOpened(base::Closure quit_closure, |
| bool success, |
| const std::string& label, |
| const MediaStreamDevice& device) { |
| if (success) { |
| stream_label_ = label; |
| observer_->AddStream(label, device); |
| } |
| |
| std::move(quit_closure).Run(); |
| } |
| |
| protected: |
| std::string stream_label_; |
| base::test::ScopedTaskEnvironment scoped_task_environment_; |
| MockMojoMediaStreamDispatcherHost mock_dispatcher_host_; |
| std::unique_ptr<MediaStreamDeviceObserver> observer_; |
| }; |
| |
| TEST_F(MediaStreamDeviceObserverTest, GetNonScreenCaptureDevices) { |
| const int kRequestId1 = 5; |
| const int kRequestId2 = 7; |
| |
| EXPECT_EQ(observer_->label_stream_map_.size(), 0u); |
| |
| // OpenDevice request 1 |
| base::RunLoop run_loop1; |
| mock_dispatcher_host_.OpenDevice( |
| kRequestId1, "device_path", MEDIA_DEVICE_VIDEO_CAPTURE, |
| base::BindOnce(&MediaStreamDeviceObserverTest::OnDeviceOpened, |
| base::Unretained(this), run_loop1.QuitClosure())); |
| run_loop1.Run(); |
| std::string stream_label1 = stream_label_; |
| |
| // OpenDevice request 2 |
| base::RunLoop run_loop2; |
| mock_dispatcher_host_.OpenDevice( |
| kRequestId2, "screen_capture", MEDIA_GUM_DESKTOP_VIDEO_CAPTURE, |
| base::BindOnce(&MediaStreamDeviceObserverTest::OnDeviceOpened, |
| base::Unretained(this), run_loop2.QuitClosure())); |
| run_loop2.Run(); |
| std::string stream_label2 = stream_label_; |
| |
| EXPECT_EQ(observer_->label_stream_map_.size(), 2u); |
| |
| // Only the device with type MEDIA_DEVICE_VIDEO_CAPTURE will be returned. |
| MediaStreamDevices video_devices = observer_->GetNonScreenCaptureDevices(); |
| EXPECT_EQ(video_devices.size(), 1u); |
| EXPECT_EQ(video_devices[0].type, MEDIA_DEVICE_VIDEO_CAPTURE); |
| |
| // Close the device from request 2. |
| observer_->RemoveStream(stream_label2); |
| EXPECT_EQ(observer_->video_session_id(stream_label2), |
| MediaStreamDevice::kNoId); |
| |
| // Close the device from request 1. |
| observer_->RemoveStream(stream_label1); |
| EXPECT_EQ(observer_->video_session_id(stream_label1), |
| MediaStreamDevice::kNoId); |
| |
| // Verify that the request have been completed. |
| EXPECT_EQ(observer_->label_stream_map_.size(), 0u); |
| } |
| |
| } // namespace content |