blob: 295916f402686421c3eb1358204f1be951321ec1 [file] [log] [blame]
// 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.
#ifndef MEDIA_REMOTING_FAKE_REMOTER_H_
#define MEDIA_REMOTING_FAKE_REMOTER_H_
#include "media/base/decoder_buffer.h"
#include "media/mojo/common/mojo_data_pipe_read_write.h"
#include "media/mojo/mojom/remoting.mojom.h"
#include "mojo/public/cpp/bindings/binding.h"
namespace media {
namespace remoting {
class RendererController;
class FakeRemotingDataStreamSender : public mojom::RemotingDataStreamSender {
public:
FakeRemotingDataStreamSender(
mojo::PendingReceiver<mojom::RemotingDataStreamSender> receiver,
mojo::ScopedDataPipeConsumerHandle consumer_handle);
~FakeRemotingDataStreamSender() override;
uint32_t send_frame_count() const { return send_frame_count_; }
uint32_t cancel_in_flight_count() const { return cancel_in_flight_count_; }
void ResetHistory();
bool ValidateFrameBuffer(size_t index,
size_t size,
bool key_frame,
int pts_ms);
private:
// mojom::RemotingDataStreamSender implementation.
void SendFrame(uint32_t frame_size) final;
void CancelInFlightData() final;
void OnFrameRead(bool success);
mojo::Binding<RemotingDataStreamSender> receiver_;
MojoDataPipeReader data_pipe_reader_;
std::vector<uint8_t> next_frame_data_;
std::vector<std::vector<uint8_t>> received_frame_list;
uint32_t send_frame_count_;
uint32_t cancel_in_flight_count_;
DISALLOW_COPY_AND_ASSIGN(FakeRemotingDataStreamSender);
};
class FakeRemoter final : public mojom::Remoter {
public:
// |start_will_fail| indicates whether starting remoting will fail.
FakeRemoter(mojom::RemotingSourcePtr source, bool start_will_fail);
~FakeRemoter() override;
// mojom::Remoter implementations.
void Start() override;
void StartDataStreams(mojo::ScopedDataPipeConsumerHandle audio_pipe,
mojo::ScopedDataPipeConsumerHandle video_pipe,
mojo::PendingReceiver<mojom::RemotingDataStreamSender>
audio_sender_receiver,
mojo::PendingReceiver<mojom::RemotingDataStreamSender>
video_sender_receiver) override;
void Stop(mojom::RemotingStopReason reason) override;
void SendMessageToSink(const std::vector<uint8_t>& message) override;
void EstimateTransmissionCapacity(
mojom::Remoter::EstimateTransmissionCapacityCallback callback) override;
private:
void Started();
void StartFailed();
void Stopped(mojom::RemotingStopReason reason);
mojom::RemotingSourcePtr source_;
bool start_will_fail_;
std::unique_ptr<FakeRemotingDataStreamSender> audio_stream_sender_;
std::unique_ptr<FakeRemotingDataStreamSender> video_stream_sender_;
base::WeakPtrFactory<FakeRemoter> weak_factory_{this};
DISALLOW_COPY_AND_ASSIGN(FakeRemoter);
};
class FakeRemoterFactory final : public mojom::RemoterFactory {
public:
// |start_will_fail| indicates whether starting remoting will fail.
explicit FakeRemoterFactory(bool start_will_fail);
~FakeRemoterFactory() override;
// mojom::RemoterFactory implementation.
void Create(mojom::RemotingSourcePtr source,
mojom::RemoterRequest request) override;
static std::unique_ptr<RendererController> CreateController(
bool start_will_fail);
private:
bool start_will_fail_;
DISALLOW_COPY_AND_ASSIGN(FakeRemoterFactory);
};
} // namespace remoting
} // namespace media
#endif // MEDIA_REMOTING_FAKE_REMOTER_H_