| // 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 "media/remoting/remote_demuxer_stream_adapter.h" |
| |
| #include <memory> |
| |
| #include "base/callback_helpers.h" |
| #include "base/run_loop.h" |
| #include "media/base/decoder_buffer.h" |
| #include "media/base/demuxer_stream.h" |
| #include "media/remoting/fake_remoting_controller.h" |
| #include "media/remoting/fake_remoting_demuxer_stream_provider.h" |
| #include "media/remoting/rpc/proto_utils.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using testing::_; |
| using testing::Invoke; |
| using testing::Return; |
| |
| namespace media { |
| namespace remoting { |
| |
| class MockRemoteDemuxerStreamAdapter { |
| public: |
| MockRemoteDemuxerStreamAdapter( |
| scoped_refptr<base::SingleThreadTaskRunner> main_task_runner, |
| scoped_refptr<base::SingleThreadTaskRunner> media_task_runner, |
| const std::string& name, |
| ::media::DemuxerStream* demuxer_stream, |
| mojom::RemotingDataStreamSenderPtrInfo stream_sender_info, |
| mojo::ScopedDataPipeProducerHandle producer_handle) |
| : weak_factory_(this) { |
| rpc_broker_.reset(new RpcBroker( |
| base::Bind(&MockRemoteDemuxerStreamAdapter::OnSendMessageToSink, |
| weak_factory_.GetWeakPtr()))); |
| demuxer_stream_adapter_.reset(new RemoteDemuxerStreamAdapter( |
| std::move(main_task_runner), std::move(media_task_runner), name, |
| demuxer_stream, rpc_broker_->GetWeakPtr(), |
| std::move(stream_sender_info), std::move(producer_handle))); |
| } |
| |
| int rpc_handle() const { return demuxer_stream_adapter_->rpc_handle(); } |
| base::WeakPtr<MockRemoteDemuxerStreamAdapter> GetWeakPtr() { |
| return weak_factory_.GetWeakPtr(); |
| } |
| |
| // Fake to signal that it's in reading state. |
| void FakeReadUntil(int read_until_count) { |
| std::unique_ptr<pb::RpcMessage> rpc(new pb::RpcMessage()); |
| rpc->set_handle(rpc_handle()); |
| rpc->set_proc(pb::RpcMessage::RPC_DS_READUNTIL); |
| auto* read_message = rpc->mutable_demuxerstream_readuntil_rpc(); |
| read_message->set_callback_handle(999); // Given an unique callback handle. |
| read_message->set_count(read_until_count); // Request 1 frame |
| |
| demuxer_stream_adapter_->OnReceivedRpc(std::move(rpc)); |
| } |
| void OnNewBuffer(const scoped_refptr<::media::DecoderBuffer>& frame) { |
| demuxer_stream_adapter_->OnNewBuffer(1, DemuxerStream::kOk, frame); |
| } |
| |
| void SignalFlush(bool flush) { demuxer_stream_adapter_->SignalFlush(flush); } |
| |
| private: |
| void OnSendMessageToSink(std::unique_ptr<std::vector<uint8_t>> message) {} |
| |
| std::unique_ptr<RpcBroker> rpc_broker_; |
| std::unique_ptr<RemoteDemuxerStreamAdapter> demuxer_stream_adapter_; |
| base::WeakPtrFactory<MockRemoteDemuxerStreamAdapter> weak_factory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(MockRemoteDemuxerStreamAdapter); |
| }; |
| |
| class RemoteDemuxerStreamAdapterTest : public ::testing::Test { |
| public: |
| RemoteDemuxerStreamAdapterTest() {} |
| ~RemoteDemuxerStreamAdapterTest() override = default; |
| |
| void SetUpDataPipe() { |
| constexpr size_t kDataPipeCapacity = 256; |
| demuxer_stream_.reset(new DummyDemuxerStream(true)); // audio. |
| const MojoCreateDataPipeOptions data_pipe_options{ |
| sizeof(MojoCreateDataPipeOptions), |
| MOJO_CREATE_DATA_PIPE_OPTIONS_FLAG_NONE, 1, kDataPipeCapacity}; |
| mojom::RemotingDataStreamSenderPtr stream_sender; |
| mojo::ScopedDataPipeProducerHandle producer_end; |
| mojo::ScopedDataPipeConsumerHandle consumer_end; |
| CHECK_EQ( |
| MOJO_RESULT_OK, |
| mojo::CreateDataPipe(&data_pipe_options, &producer_end, &consumer_end)); |
| |
| data_stream_sender_.reset(new FakeRemotingDataStreamSender( |
| GetProxy(&stream_sender), std::move(consumer_end))); |
| demuxer_stream_adapter_.reset(new MockRemoteDemuxerStreamAdapter( |
| message_loop_.task_runner(), message_loop_.task_runner(), "test", |
| demuxer_stream_.get(), stream_sender.PassInterface(), |
| std::move(producer_end))); |
| // RemoteDemuxerStreamAdapter constructor posts task to main thread to |
| // register MessageReceiverCallback. Therefore it should call |
| // RunPendingTasks() to make sure task is executed. |
| RunPendingTasks(); |
| } |
| |
| void TearDown() override { base::RunLoop().RunUntilIdle(); } |
| |
| void RunPendingTasks() { base::RunLoop().RunUntilIdle(); } |
| |
| protected: |
| void SetUp() override { SetUpDataPipe(); } |
| |
| std::unique_ptr<DummyDemuxerStream> demuxer_stream_; |
| std::unique_ptr<FakeRemotingDataStreamSender> data_stream_sender_; |
| std::unique_ptr<MockRemoteDemuxerStreamAdapter> demuxer_stream_adapter_; |
| base::MessageLoop message_loop_; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(RemoteDemuxerStreamAdapterTest); |
| }; |
| |
| TEST_F(RemoteDemuxerStreamAdapterTest, SingleReadUntil) { |
| // Read will be called once since it doesn't return frame buffer in the dummy |
| // implementation. |
| EXPECT_CALL(*demuxer_stream_, Read(_)).Times(1); |
| |
| demuxer_stream_adapter_->FakeReadUntil(3); |
| RunPendingTasks(); |
| } |
| |
| TEST_F(RemoteDemuxerStreamAdapterTest, MultiReadUntil) { |
| // Read will be called once since it doesn't return frame buffer in the dummy |
| // implementation, and 2nd one will not proceed when there is ongoing read. |
| EXPECT_CALL(*demuxer_stream_, Read(_)).Times(1); |
| |
| demuxer_stream_adapter_->FakeReadUntil(1); |
| RunPendingTasks(); |
| |
| demuxer_stream_adapter_->FakeReadUntil(2); |
| RunPendingTasks(); |
| } |
| |
| TEST_F(RemoteDemuxerStreamAdapterTest, WriteOneFrameSmallerThanCapacity) { |
| // Sends a frame with size 50 bytes, pts = 1 and key frame. |
| demuxer_stream_->CreateFakeFrame(50, true, 1 /* pts */); |
| demuxer_stream_adapter_->FakeReadUntil(1); |
| RunPendingTasks(); |
| |
| // Checks if it's sent to consumer side and data is correct |
| ASSERT_EQ(data_stream_sender_->send_frame_count(), 1U); |
| ASSERT_EQ(data_stream_sender_->consume_data_chunk_count(), 1U); |
| ASSERT_TRUE(data_stream_sender_->ValidateFrameBuffer(0, 50, true, 1)); |
| data_stream_sender_->ResetHistory(); |
| } |
| |
| TEST_F(RemoteDemuxerStreamAdapterTest, WriteOneFrameLargerThanCapacity) { |
| // Sends a frame with size 800 bytes, pts = 1 and key frame. |
| demuxer_stream_->CreateFakeFrame(800, true, 1 /* pts */); |
| demuxer_stream_adapter_->FakeReadUntil(1); |
| RunPendingTasks(); |
| |
| // Checks if it's sent to consumer side and data is correct |
| ASSERT_EQ(data_stream_sender_->send_frame_count(), 1U); |
| ASSERT_EQ(data_stream_sender_->consume_data_chunk_count(), 4U); |
| ASSERT_TRUE(data_stream_sender_->ValidateFrameBuffer(0, 800, true, 1)); |
| data_stream_sender_->ResetHistory(); |
| } |
| |
| TEST_F(RemoteDemuxerStreamAdapterTest, SendFrameAndSignalFlushMix) { |
| // Sends a frame with size 50 bytes, pts = 1 and key frame. |
| demuxer_stream_->CreateFakeFrame(50, true, 1 /* pts */); |
| // Issues ReadUntil request with frame count up to 1 (fetch #0). |
| demuxer_stream_adapter_->FakeReadUntil(1); |
| RunPendingTasks(); |
| ASSERT_EQ(data_stream_sender_->send_frame_count(), 1U); |
| ASSERT_TRUE(data_stream_sender_->ValidateFrameBuffer(0, 50, true, 1)); |
| data_stream_sender_->ResetHistory(); |
| |
| // Sends two frames with size 100 + 150 bytes |
| demuxer_stream_->CreateFakeFrame(100, false, 2 /* pts */); |
| demuxer_stream_->CreateFakeFrame(150, false, 3 /* pts */); |
| // Issues ReadUntil request with frame count up to 3 (fetch #1 and #2). |
| demuxer_stream_adapter_->FakeReadUntil(3); |
| RunPendingTasks(); |
| ASSERT_EQ(data_stream_sender_->send_frame_count(), 2U); |
| ASSERT_TRUE(data_stream_sender_->ValidateFrameBuffer(0, 100, false, 2)); |
| ASSERT_TRUE(data_stream_sender_->ValidateFrameBuffer(1, 150, false, 3)); |
| data_stream_sender_->ResetHistory(); |
| |
| // Signal flush |
| ASSERT_EQ(data_stream_sender_->cancel_in_flight_count(), 0U); |
| demuxer_stream_adapter_->SignalFlush(true); |
| RunPendingTasks(); |
| ASSERT_EQ(data_stream_sender_->cancel_in_flight_count(), 1U); |
| |
| // ReadUntil request after flush signaling should be ignored. |
| demuxer_stream_->CreateFakeFrame(100, false, 4 /* pts */); |
| demuxer_stream_->CreateFakeFrame(100, false, 5 /* pts */); |
| // Issues ReadUntil request with frame count up to 5 (fetch #3 and #4). |
| demuxer_stream_adapter_->FakeReadUntil(5); |
| RunPendingTasks(); |
| ASSERT_EQ(data_stream_sender_->send_frame_count(), 0U); |
| |
| // Signal flush done |
| demuxer_stream_adapter_->SignalFlush(false); |
| RunPendingTasks(); |
| ASSERT_EQ(data_stream_sender_->cancel_in_flight_count(), 1U); |
| data_stream_sender_->ResetHistory(); |
| |
| // Re-issues ReadUntil request with frame count up to 4 (fetch #3). |
| demuxer_stream_adapter_->FakeReadUntil(4); |
| RunPendingTasks(); |
| ASSERT_EQ(data_stream_sender_->send_frame_count(), 1U); |
| ASSERT_TRUE(data_stream_sender_->ValidateFrameBuffer(0, 100, false, 4)); |
| data_stream_sender_->ResetHistory(); |
| } |
| |
| } // namesapce remoting |
| } // namespace media |