pc/test/mock_peer_connection_observers.h - src - Git at Google

 /*
  *  Copyright 2012 The WebRTC project authors. All Rights Reserved.
  *
  *  Use of this source code is governed by a BSD-style license
  *  that can be found in the LICENSE file in the root of the source
  *  tree. An additional intellectual property rights grant can be found
  *  in the file PATENTS.  All contributing project authors may
  *  be found in the AUTHORS file in the root of the source tree.
  */

 // This file contains mock implementations of observers used in PeerConnection.
 // TODO(steveanton): These aren't really mocks and should be renamed.

 #ifndef PC_TEST_MOCK_PEER_CONNECTION_OBSERVERS_H_
 #define PC_TEST_MOCK_PEER_CONNECTION_OBSERVERS_H_

 #include <map>
 #include <memory>
 #include <string>
 #include <utility>
 #include <vector>

 #include "api/data_channel_interface.h"
 #include "api/jsep_ice_candidate.h"
 #include "pc/stream_collection.h"
 #include "rtc_base/checks.h"

 namespace webrtc {

 class MockPeerConnectionObserver : public PeerConnectionObserver {
  public:
   struct AddTrackEvent {
     explicit AddTrackEvent(
         rtc::scoped_refptr<RtpReceiverInterface> event_receiver,
         std::vector<rtc::scoped_refptr<MediaStreamInterface>> event_streams)
         : receiver(std::move(event_receiver)),
           streams(std::move(event_streams)) {
       for (auto stream : streams) {
         std::vector<rtc::scoped_refptr<MediaStreamTrackInterface>> tracks;
         for (auto audio_track : stream->GetAudioTracks()) {
           tracks.push_back(audio_track);
         }
         for (auto video_track : stream->GetVideoTracks()) {
           tracks.push_back(video_track);
         }
         snapshotted_stream_tracks[stream] = tracks;
       }
     }

     rtc::scoped_refptr<RtpReceiverInterface> receiver;
     std::vector<rtc::scoped_refptr<MediaStreamInterface>> streams;
     // This map records the tracks present in each stream at the time the
     // OnAddTrack callback was issued.
     std::map<rtc::scoped_refptr<MediaStreamInterface>,
              std::vector<rtc::scoped_refptr<MediaStreamTrackInterface>>>
         snapshotted_stream_tracks;
   };

   MockPeerConnectionObserver() : remote_streams_(StreamCollection::Create()) {}
   virtual ~MockPeerConnectionObserver() {}
   void SetPeerConnectionInterface(PeerConnectionInterface* pc) {
     pc_ = pc;
     if (pc) {
       state_ = pc_->signaling_state();
     }
   }
   void OnSignalingChange(
       PeerConnectionInterface::SignalingState new_state) override {
     RTC_DCHECK(pc_);
     RTC_DCHECK(pc_->signaling_state() == new_state);
     state_ = new_state;
   }

   MediaStreamInterface* RemoteStream(const std::string& label) {
     return remote_streams_->find(label);
   }
   StreamCollectionInterface* remote_streams() const { return remote_streams_; }
   void OnAddStream(rtc::scoped_refptr<MediaStreamInterface> stream) override {
     last_added_stream_ = stream;
     remote_streams_->AddStream(stream);
   }
   void OnRemoveStream(
       rtc::scoped_refptr<MediaStreamInterface> stream) override {
     last_removed_stream_ = stream;
     remote_streams_->RemoveStream(stream);
   }
   void OnRenegotiationNeeded() override { renegotiation_needed_ = true; }
   void OnDataChannel(
       rtc::scoped_refptr<DataChannelInterface> data_channel) override {
     last_datachannel_ = data_channel;
   }

   void OnIceConnectionChange(
       PeerConnectionInterface::IceConnectionState new_state) override {
     RTC_DCHECK(pc_);
     RTC_DCHECK(pc_->ice_connection_state() == new_state);
     // When ICE is finished, the caller will get to a kIceConnectionCompleted
     // state, because it has the ICE controlling role, while the callee
     // will get to a kIceConnectionConnected state. This means that both ICE
     // and DTLS are connected.
     ice_connected_ =
         (new_state == PeerConnectionInterface::kIceConnectionConnected) ||
         (new_state == PeerConnectionInterface::kIceConnectionCompleted);
     callback_triggered_ = true;
   }
   void OnIceGatheringChange(
       PeerConnectionInterface::IceGatheringState new_state) override {
     RTC_DCHECK(pc_);
     RTC_DCHECK(pc_->ice_gathering_state() == new_state);
     ice_gathering_complete_ =
         new_state == PeerConnectionInterface::kIceGatheringComplete;
     callback_triggered_ = true;
   }
   void OnIceCandidate(const IceCandidateInterface* candidate) override {
     RTC_DCHECK(pc_);
     RTC_DCHECK(PeerConnectionInterface::kIceGatheringNew !=
                pc_->ice_gathering_state());
     candidates_.push_back(std::make_unique<JsepIceCandidate>(
         candidate->sdp_mid(), candidate->sdp_mline_index(),
         candidate->candidate()));
     callback_triggered_ = true;
   }

   void OnIceCandidatesRemoved(
       const std::vector<cricket::Candidate>& candidates) override {
     num_candidates_removed_++;
     callback_triggered_ = true;
   }

   void OnIceConnectionReceivingChange(bool receiving) override {
     callback_triggered_ = true;
   }

   void OnAddTrack(rtc::scoped_refptr<RtpReceiverInterface> receiver,
                   const std::vector<rtc::scoped_refptr<MediaStreamInterface>>&
                       streams) override {
     RTC_DCHECK(receiver);
     num_added_tracks_++;
     last_added_track_label_ = receiver->id();
     add_track_events_.push_back(AddTrackEvent(receiver, streams));
   }

   void OnTrack(
       rtc::scoped_refptr<RtpTransceiverInterface> transceiver) override {
     on_track_transceivers_.push_back(transceiver);
   }

   void OnRemoveTrack(
       rtc::scoped_refptr<RtpReceiverInterface> receiver) override {
     remove_track_events_.push_back(receiver);
   }

   std::vector<rtc::scoped_refptr<RtpReceiverInterface>> GetAddTrackReceivers() {
     std::vector<rtc::scoped_refptr<RtpReceiverInterface>> receivers;
     for (const AddTrackEvent& event : add_track_events_) {
       receivers.push_back(event.receiver);
     }
     return receivers;
   }

   int CountAddTrackEventsForStream(const std::string& stream_id) {
     int found_tracks = 0;
     for (const AddTrackEvent& event : add_track_events_) {
       bool has_stream_id = false;
       for (auto stream : event.streams) {
         if (stream->id() == stream_id) {
           has_stream_id = true;
           break;
         }
       }
       if (has_stream_id) {
         ++found_tracks;
       }
     }
     return found_tracks;
   }

   // Returns the id of the last added stream.
   // Empty string if no stream have been added.
   std::string GetLastAddedStreamId() {
     if (last_added_stream_.get())
       return last_added_stream_->id();
     return "";
   }
   std::string GetLastRemovedStreamId() {
     if (last_removed_stream_.get())
       return last_removed_stream_->id();
     return "";
   }

   IceCandidateInterface* last_candidate() {
     if (candidates_.empty()) {
       return nullptr;
     } else {
       return candidates_.back().get();
     }
   }

   std::vector<const IceCandidateInterface*> GetAllCandidates() {
     std::vector<const IceCandidateInterface*> candidates;
     for (const auto& candidate : candidates_) {
       candidates.push_back(candidate.get());
     }
     return candidates;
   }

   std::vector<IceCandidateInterface*> GetCandidatesByMline(int mline_index) {
     std::vector<IceCandidateInterface*> candidates;
     for (const auto& candidate : candidates_) {
       if (candidate->sdp_mline_index() == mline_index) {
         candidates.push_back(candidate.get());
       }
     }
     return candidates;
   }

   bool negotiation_needed() const { return renegotiation_needed_; }
   void clear_negotiation_needed() { renegotiation_needed_ = false; }

   rtc::scoped_refptr<PeerConnectionInterface> pc_;
   PeerConnectionInterface::SignalingState state_;
   std::vector<std::unique_ptr<IceCandidateInterface>> candidates_;
   rtc::scoped_refptr<DataChannelInterface> last_datachannel_;
   rtc::scoped_refptr<StreamCollection> remote_streams_;
   bool renegotiation_needed_ = false;
   bool ice_gathering_complete_ = false;
   bool ice_connected_ = false;
   bool callback_triggered_ = false;
   int num_added_tracks_ = 0;
   std::string last_added_track_label_;
   std::vector<AddTrackEvent> add_track_events_;
   std::vector<rtc::scoped_refptr<RtpReceiverInterface>> remove_track_events_;
   std::vector<rtc::scoped_refptr<RtpTransceiverInterface>>
       on_track_transceivers_;
   int num_candidates_removed_ = 0;

  private:
   rtc::scoped_refptr<MediaStreamInterface> last_added_stream_;
   rtc::scoped_refptr<MediaStreamInterface> last_removed_stream_;
 };

 class MockCreateSessionDescriptionObserver
     : public webrtc::CreateSessionDescriptionObserver {
  public:
   MockCreateSessionDescriptionObserver()
       : called_(false),
         error_("MockCreateSessionDescriptionObserver not called") {}
   virtual ~MockCreateSessionDescriptionObserver() {}
   void OnSuccess(SessionDescriptionInterface* desc) override {
     called_ = true;
     error_ = "";
     desc_.reset(desc);
   }
   void OnFailure(webrtc::RTCError error) override {
     called_ = true;
     error_ = error.message();
   }
   bool called() const { return called_; }
   bool result() const { return error_.empty(); }
   const std::string& error() const { return error_; }
   std::unique_ptr<SessionDescriptionInterface> MoveDescription() {
     return std::move(desc_);
   }

  private:
   bool called_;
   std::string error_;
   std::unique_ptr<SessionDescriptionInterface> desc_;
 };

 class MockSetSessionDescriptionObserver
     : public webrtc::SetSessionDescriptionObserver {
  public:
   static rtc::scoped_refptr<MockSetSessionDescriptionObserver> Create() {
     return new rtc::RefCountedObject<MockSetSessionDescriptionObserver>();
   }

   MockSetSessionDescriptionObserver()
       : called_(false),
         error_("MockSetSessionDescriptionObserver not called") {}
   ~MockSetSessionDescriptionObserver() override {}
   void OnSuccess() override {
     called_ = true;
     error_ = "";
   }
   void OnFailure(webrtc::RTCError error) override {
     called_ = true;
     error_ = error.message();
   }

   bool called() const { return called_; }
   bool result() const { return error_.empty(); }
   const std::string& error() const { return error_; }

  private:
   bool called_;
   std::string error_;
 };

 class MockSetRemoteDescriptionObserver
     : public rtc::RefCountedObject<SetRemoteDescriptionObserverInterface> {
  public:
   bool called() const { return error_.has_value(); }
   RTCError& error() {
     RTC_DCHECK(error_.has_value());
     return *error_;
   }

   // SetRemoteDescriptionObserverInterface implementation.
   void OnSetRemoteDescriptionComplete(RTCError error) override {
     error_ = std::move(error);
   }

  private:
   // Set on complete, on success this is set to an RTCError::OK() error.
   absl::optional<RTCError> error_;
 };

 class MockDataChannelObserver : public webrtc::DataChannelObserver {
  public:
   explicit MockDataChannelObserver(webrtc::DataChannelInterface* channel)
       : channel_(channel) {
     channel_->RegisterObserver(this);
     state_ = channel_->state();
   }
   virtual ~MockDataChannelObserver() { channel_->UnregisterObserver(); }

   void OnBufferedAmountChange(uint64_t previous_amount) override {}

   void OnStateChange() override { state_ = channel_->state(); }
   void OnMessage(const DataBuffer& buffer) override {
     messages_.push_back(
         std::string(buffer.data.data<char>(), buffer.data.size()));
   }

   bool IsOpen() const { return state_ == DataChannelInterface::kOpen; }
   std::vector<std::string> messages() const { return messages_; }
   std::string last_message() const {
     return messages_.empty() ? std::string() : messages_.back();
   }
   size_t received_message_count() const { return messages_.size(); }

  private:
   rtc::scoped_refptr<webrtc::DataChannelInterface> channel_;
   DataChannelInterface::DataState state_;
   std::vector<std::string> messages_;
 };

 class MockStatsObserver : public webrtc::StatsObserver {
  public:
   MockStatsObserver() : called_(false), stats_() {}
   virtual ~MockStatsObserver() {}

   virtual void OnComplete(const StatsReports& reports) {
     RTC_CHECK(!called_);
     called_ = true;
     stats_.Clear();
     stats_.number_of_reports = reports.size();
     for (const auto* r : reports) {
       if (r->type() == StatsReport::kStatsReportTypeSsrc) {
         stats_.timestamp = r->timestamp();
         GetIntValue(r, StatsReport::kStatsValueNameAudioOutputLevel,
                     &stats_.audio_output_level);
         GetIntValue(r, StatsReport::kStatsValueNameAudioInputLevel,
                     &stats_.audio_input_level);
         GetIntValue(r, StatsReport::kStatsValueNameBytesReceived,
                     &stats_.bytes_received);
         GetIntValue(r, StatsReport::kStatsValueNameBytesSent,
                     &stats_.bytes_sent);
         GetInt64Value(r, StatsReport::kStatsValueNameCaptureStartNtpTimeMs,
                       &stats_.capture_start_ntp_time);
         stats_.track_ids.emplace_back();
         GetStringValue(r, StatsReport::kStatsValueNameTrackId,
                        &stats_.track_ids.back());
       } else if (r->type() == StatsReport::kStatsReportTypeBwe) {
         stats_.timestamp = r->timestamp();
         GetIntValue(r, StatsReport::kStatsValueNameAvailableReceiveBandwidth,
                     &stats_.available_receive_bandwidth);
       } else if (r->type() == StatsReport::kStatsReportTypeComponent) {
         stats_.timestamp = r->timestamp();
         GetStringValue(r, StatsReport::kStatsValueNameDtlsCipher,
                        &stats_.dtls_cipher);
         GetStringValue(r, StatsReport::kStatsValueNameSrtpCipher,
                        &stats_.srtp_cipher);
       }
     }
   }

   bool called() const { return called_; }
   size_t number_of_reports() const { return stats_.number_of_reports; }
   double timestamp() const { return stats_.timestamp; }

   int AudioOutputLevel() const {
     RTC_CHECK(called_);
     return stats_.audio_output_level;
   }

   int AudioInputLevel() const {
     RTC_CHECK(called_);
     return stats_.audio_input_level;
   }

   int BytesReceived() const {
     RTC_CHECK(called_);
     return stats_.bytes_received;
   }

   int BytesSent() const {
     RTC_CHECK(called_);
     return stats_.bytes_sent;
   }

   int64_t CaptureStartNtpTime() const {
     RTC_CHECK(called_);
     return stats_.capture_start_ntp_time;
   }

   int AvailableReceiveBandwidth() const {
     RTC_CHECK(called_);
     return stats_.available_receive_bandwidth;
   }

   std::string DtlsCipher() const {
     RTC_CHECK(called_);
     return stats_.dtls_cipher;
   }

   std::string SrtpCipher() const {
     RTC_CHECK(called_);
     return stats_.srtp_cipher;
   }

   std::vector<std::string> TrackIds() const {
     RTC_CHECK(called_);
     return stats_.track_ids;
   }

  private:
   bool GetIntValue(const StatsReport* report,
                    StatsReport::StatsValueName name,
                    int* value) {
     const StatsReport::Value* v = report->FindValue(name);
     if (v) {
       // TODO(tommi): We should really just be using an int here :-/
       *value = rtc::FromString<int>(v->ToString());
     }
     return v != nullptr;
   }

   bool GetInt64Value(const StatsReport* report,
                      StatsReport::StatsValueName name,
                      int64_t* value) {
     const StatsReport::Value* v = report->FindValue(name);
     if (v) {
       // TODO(tommi): We should really just be using an int here :-/
       *value = rtc::FromString<int64_t>(v->ToString());
     }
     return v != nullptr;
   }

   bool GetStringValue(const StatsReport* report,
                       StatsReport::StatsValueName name,
                       std::string* value) {
     const StatsReport::Value* v = report->FindValue(name);
     if (v)
       *value = v->ToString();
     return v != nullptr;
   }

   bool called_;
   struct {
     void Clear() {
       number_of_reports = 0;
       timestamp = 0;
       audio_output_level = 0;
       audio_input_level = 0;
       bytes_received = 0;
       bytes_sent = 0;
       capture_start_ntp_time = 0;
       available_receive_bandwidth = 0;
       dtls_cipher.clear();
       srtp_cipher.clear();
       track_ids.clear();
     }

     size_t number_of_reports;
     double timestamp;
     int audio_output_level;
     int audio_input_level;
     int bytes_received;
     int bytes_sent;
     int64_t capture_start_ntp_time;
     int available_receive_bandwidth;
     std::string dtls_cipher;
     std::string srtp_cipher;
     std::vector<std::string> track_ids;
   } stats_;
 };

 // Helper class that just stores the report from the callback.
 class MockRTCStatsCollectorCallback : public webrtc::RTCStatsCollectorCallback {
  public:
   rtc::scoped_refptr<const RTCStatsReport> report() { return report_; }

   bool called() const { return called_; }

  protected:
   void OnStatsDelivered(
       const rtc::scoped_refptr<const RTCStatsReport>& report) override {
     report_ = report;
     called_ = true;
   }

  private:
   bool called_ = false;
   rtc::scoped_refptr<const RTCStatsReport> report_;
 };

 }  // namespace webrtc

 #endif  // PC_TEST_MOCK_PEER_CONNECTION_OBSERVERS_H_
	/*
	* Copyright 2012 The WebRTC project authors. All Rights Reserved.
	*
	* Use of this source code is governed by a BSD-style license
	* that can be found in the LICENSE file in the root of the source
	* tree. An additional intellectual property rights grant can be found
	* in the file PATENTS. All contributing project authors may
	* be found in the AUTHORS file in the root of the source tree.
	*/

	// This file contains mock implementations of observers used in PeerConnection.
	// TODO(steveanton): These aren't really mocks and should be renamed.

	#ifndef PC_TEST_MOCK_PEER_CONNECTION_OBSERVERS_H_
	#define PC_TEST_MOCK_PEER_CONNECTION_OBSERVERS_H_

	#include <map>
	#include <memory>
	#include <string>
	#include <utility>
	#include <vector>

	#include "api/data_channel_interface.h"
	#include "api/jsep_ice_candidate.h"
	#include "pc/stream_collection.h"
	#include "rtc_base/checks.h"

	namespace webrtc {

	class MockPeerConnectionObserver : public PeerConnectionObserver {
	public:
	struct AddTrackEvent {
	explicit AddTrackEvent(
	rtc::scoped_refptr<RtpReceiverInterface> event_receiver,
	std::vector<rtc::scoped_refptr<MediaStreamInterface>> event_streams)
	: receiver(std::move(event_receiver)),
	streams(std::move(event_streams)) {
	for (auto stream : streams) {
	std::vector<rtc::scoped_refptr<MediaStreamTrackInterface>> tracks;
	for (auto audio_track : stream->GetAudioTracks()) {
	tracks.push_back(audio_track);
	}
	for (auto video_track : stream->GetVideoTracks()) {
	tracks.push_back(video_track);
	}
	snapshotted_stream_tracks[stream] = tracks;
	}
	}

	rtc::scoped_refptr<RtpReceiverInterface> receiver;
	std::vector<rtc::scoped_refptr<MediaStreamInterface>> streams;
	// This map records the tracks present in each stream at the time the
	// OnAddTrack callback was issued.
	std::map<rtc::scoped_refptr<MediaStreamInterface>,
	std::vector<rtc::scoped_refptr<MediaStreamTrackInterface>>>
	snapshotted_stream_tracks;
	};

	MockPeerConnectionObserver() : remote_streams_(StreamCollection::Create()) {}
	virtual ~MockPeerConnectionObserver() {}
	void SetPeerConnectionInterface(PeerConnectionInterface* pc) {
	pc_ = pc;
	if (pc) {
	state_ = pc_->signaling_state();
	}
	}
	void OnSignalingChange(
	PeerConnectionInterface::SignalingState new_state) override {
	RTC_DCHECK(pc_);
	RTC_DCHECK(pc_->signaling_state() == new_state);
	state_ = new_state;
	}

	MediaStreamInterface* RemoteStream(const std::string& label) {
	return remote_streams_->find(label);
	}
	StreamCollectionInterface* remote_streams() const { return remote_streams_; }
	void OnAddStream(rtc::scoped_refptr<MediaStreamInterface> stream) override {
	last_added_stream_ = stream;
	remote_streams_->AddStream(stream);
	}
	void OnRemoveStream(
	rtc::scoped_refptr<MediaStreamInterface> stream) override {
	last_removed_stream_ = stream;
	remote_streams_->RemoveStream(stream);
	}
	void OnRenegotiationNeeded() override { renegotiation_needed_ = true; }
	void OnDataChannel(
	rtc::scoped_refptr<DataChannelInterface> data_channel) override {
	last_datachannel_ = data_channel;
	}

	void OnIceConnectionChange(
	PeerConnectionInterface::IceConnectionState new_state) override {
	RTC_DCHECK(pc_);
	RTC_DCHECK(pc_->ice_connection_state() == new_state);
	// When ICE is finished, the caller will get to a kIceConnectionCompleted
	// state, because it has the ICE controlling role, while the callee
	// will get to a kIceConnectionConnected state. This means that both ICE
	// and DTLS are connected.
	ice_connected_ =
	(new_state == PeerConnectionInterface::kIceConnectionConnected) \|\|
	(new_state == PeerConnectionInterface::kIceConnectionCompleted);
	callback_triggered_ = true;
	}
	void OnIceGatheringChange(
	PeerConnectionInterface::IceGatheringState new_state) override {
	RTC_DCHECK(pc_);
	RTC_DCHECK(pc_->ice_gathering_state() == new_state);
	ice_gathering_complete_ =
	new_state == PeerConnectionInterface::kIceGatheringComplete;
	callback_triggered_ = true;
	}
	void OnIceCandidate(const IceCandidateInterface* candidate) override {
	RTC_DCHECK(pc_);
	RTC_DCHECK(PeerConnectionInterface::kIceGatheringNew !=
	pc_->ice_gathering_state());
	candidates_.push_back(std::make_unique<JsepIceCandidate>(
	candidate->sdp_mid(), candidate->sdp_mline_index(),
	candidate->candidate()));
	callback_triggered_ = true;
	}

	void OnIceCandidatesRemoved(
	const std::vector<cricket::Candidate>& candidates) override {
	num_candidates_removed_++;
	callback_triggered_ = true;
	}

	void OnIceConnectionReceivingChange(bool receiving) override {
	callback_triggered_ = true;
	}

	void OnAddTrack(rtc::scoped_refptr<RtpReceiverInterface> receiver,
	const std::vector<rtc::scoped_refptr<MediaStreamInterface>>&
	streams) override {
	RTC_DCHECK(receiver);
	num_added_tracks_++;
	last_added_track_label_ = receiver->id();
	add_track_events_.push_back(AddTrackEvent(receiver, streams));
	}

	void OnTrack(
	rtc::scoped_refptr<RtpTransceiverInterface> transceiver) override {
	on_track_transceivers_.push_back(transceiver);
	}

	void OnRemoveTrack(
	rtc::scoped_refptr<RtpReceiverInterface> receiver) override {
	remove_track_events_.push_back(receiver);
	}

	std::vector<rtc::scoped_refptr<RtpReceiverInterface>> GetAddTrackReceivers() {
	std::vector<rtc::scoped_refptr<RtpReceiverInterface>> receivers;
	for (const AddTrackEvent& event : add_track_events_) {
	receivers.push_back(event.receiver);
	}
	return receivers;
	}

	int CountAddTrackEventsForStream(const std::string& stream_id) {
	int found_tracks = 0;
	for (const AddTrackEvent& event : add_track_events_) {
	bool has_stream_id = false;
	for (auto stream : event.streams) {
	if (stream->id() == stream_id) {
	has_stream_id = true;
	break;
	}
	}
	if (has_stream_id) {
	++found_tracks;
	}
	}
	return found_tracks;
	}

	// Returns the id of the last added stream.
	// Empty string if no stream have been added.
	std::string GetLastAddedStreamId() {
	if (last_added_stream_.get())
	return last_added_stream_->id();
	return "";
	}
	std::string GetLastRemovedStreamId() {
	if (last_removed_stream_.get())
	return last_removed_stream_->id();
	return "";
	}

	IceCandidateInterface* last_candidate() {
	if (candidates_.empty()) {
	return nullptr;
	} else {
	return candidates_.back().get();
	}
	}

	std::vector<const IceCandidateInterface*> GetAllCandidates() {
	std::vector<const IceCandidateInterface*> candidates;
	for (const auto& candidate : candidates_) {
	candidates.push_back(candidate.get());
	}
	return candidates;
	}

	std::vector<IceCandidateInterface*> GetCandidatesByMline(int mline_index) {
	std::vector<IceCandidateInterface*> candidates;
	for (const auto& candidate : candidates_) {
	if (candidate->sdp_mline_index() == mline_index) {
	candidates.push_back(candidate.get());
	}
	}
	return candidates;
	}

	bool negotiation_needed() const { return renegotiation_needed_; }
	void clear_negotiation_needed() { renegotiation_needed_ = false; }

	rtc::scoped_refptr<PeerConnectionInterface> pc_;
	PeerConnectionInterface::SignalingState state_;
	std::vector<std::unique_ptr<IceCandidateInterface>> candidates_;
	rtc::scoped_refptr<DataChannelInterface> last_datachannel_;
	rtc::scoped_refptr<StreamCollection> remote_streams_;
	bool renegotiation_needed_ = false;
	bool ice_gathering_complete_ = false;
	bool ice_connected_ = false;
	bool callback_triggered_ = false;
	int num_added_tracks_ = 0;
	std::string last_added_track_label_;
	std::vector<AddTrackEvent> add_track_events_;
	std::vector<rtc::scoped_refptr<RtpReceiverInterface>> remove_track_events_;
	std::vector<rtc::scoped_refptr<RtpTransceiverInterface>>
	on_track_transceivers_;
	int num_candidates_removed_ = 0;

	private:
	rtc::scoped_refptr<MediaStreamInterface> last_added_stream_;
	rtc::scoped_refptr<MediaStreamInterface> last_removed_stream_;
	};

	class MockCreateSessionDescriptionObserver
	: public webrtc::CreateSessionDescriptionObserver {
	public:
	MockCreateSessionDescriptionObserver()
	: called_(false),
	error_("MockCreateSessionDescriptionObserver not called") {}
	virtual ~MockCreateSessionDescriptionObserver() {}
	void OnSuccess(SessionDescriptionInterface* desc) override {
	called_ = true;
	error_ = "";
	desc_.reset(desc);
	}
	void OnFailure(webrtc::RTCError error) override {
	called_ = true;
	error_ = error.message();
	}
	bool called() const { return called_; }
	bool result() const { return error_.empty(); }
	const std::string& error() const { return error_; }
	std::unique_ptr<SessionDescriptionInterface> MoveDescription() {
	return std::move(desc_);
	}

	private:
	bool called_;
	std::string error_;
	std::unique_ptr<SessionDescriptionInterface> desc_;
	};

	class MockSetSessionDescriptionObserver
	: public webrtc::SetSessionDescriptionObserver {
	public:
	static rtc::scoped_refptr<MockSetSessionDescriptionObserver> Create() {
	return new rtc::RefCountedObject<MockSetSessionDescriptionObserver>();
	}

	MockSetSessionDescriptionObserver()
	: called_(false),
	error_("MockSetSessionDescriptionObserver not called") {}
	~MockSetSessionDescriptionObserver() override {}
	void OnSuccess() override {
	called_ = true;
	error_ = "";
	}
	void OnFailure(webrtc::RTCError error) override {
	called_ = true;
	error_ = error.message();
	}

	bool called() const { return called_; }
	bool result() const { return error_.empty(); }
	const std::string& error() const { return error_; }

	private:
	bool called_;
	std::string error_;
	};

	class MockSetRemoteDescriptionObserver
	: public rtc::RefCountedObject<SetRemoteDescriptionObserverInterface> {
	public:
	bool called() const { return error_.has_value(); }
	RTCError& error() {
	RTC_DCHECK(error_.has_value());
	return *error_;
	}

	// SetRemoteDescriptionObserverInterface implementation.
	void OnSetRemoteDescriptionComplete(RTCError error) override {
	error_ = std::move(error);
	}

	private:
	// Set on complete, on success this is set to an RTCError::OK() error.
	absl::optional<RTCError> error_;
	};

	class MockDataChannelObserver : public webrtc::DataChannelObserver {
	public:
	explicit MockDataChannelObserver(webrtc::DataChannelInterface* channel)
	: channel_(channel) {
	channel_->RegisterObserver(this);
	state_ = channel_->state();
	}
	virtual ~MockDataChannelObserver() { channel_->UnregisterObserver(); }

	void OnBufferedAmountChange(uint64_t previous_amount) override {}

	void OnStateChange() override { state_ = channel_->state(); }
	void OnMessage(const DataBuffer& buffer) override {
	messages_.push_back(
	std::string(buffer.data.data<char>(), buffer.data.size()));
	}

	bool IsOpen() const { return state_ == DataChannelInterface::kOpen; }
	std::vector<std::string> messages() const { return messages_; }
	std::string last_message() const {
	return messages_.empty() ? std::string() : messages_.back();
	}
	size_t received_message_count() const { return messages_.size(); }

	private:
	rtc::scoped_refptr<webrtc::DataChannelInterface> channel_;
	DataChannelInterface::DataState state_;
	std::vector<std::string> messages_;
	};

	class MockStatsObserver : public webrtc::StatsObserver {
	public:
	MockStatsObserver() : called_(false), stats_() {}
	virtual ~MockStatsObserver() {}

	virtual void OnComplete(const StatsReports& reports) {
	RTC_CHECK(!called_);
	called_ = true;
	stats_.Clear();
	stats_.number_of_reports = reports.size();
	for (const auto* r : reports) {
	if (r->type() == StatsReport::kStatsReportTypeSsrc) {
	stats_.timestamp = r->timestamp();
	GetIntValue(r, StatsReport::kStatsValueNameAudioOutputLevel,
	&stats_.audio_output_level);
	GetIntValue(r, StatsReport::kStatsValueNameAudioInputLevel,
	&stats_.audio_input_level);
	GetIntValue(r, StatsReport::kStatsValueNameBytesReceived,
	&stats_.bytes_received);
	GetIntValue(r, StatsReport::kStatsValueNameBytesSent,
	&stats_.bytes_sent);
	GetInt64Value(r, StatsReport::kStatsValueNameCaptureStartNtpTimeMs,
	&stats_.capture_start_ntp_time);
	stats_.track_ids.emplace_back();
	GetStringValue(r, StatsReport::kStatsValueNameTrackId,
	&stats_.track_ids.back());
	} else if (r->type() == StatsReport::kStatsReportTypeBwe) {
	stats_.timestamp = r->timestamp();
	GetIntValue(r, StatsReport::kStatsValueNameAvailableReceiveBandwidth,
	&stats_.available_receive_bandwidth);
	} else if (r->type() == StatsReport::kStatsReportTypeComponent) {
	stats_.timestamp = r->timestamp();
	GetStringValue(r, StatsReport::kStatsValueNameDtlsCipher,
	&stats_.dtls_cipher);
	GetStringValue(r, StatsReport::kStatsValueNameSrtpCipher,
	&stats_.srtp_cipher);
	}
	}
	}

	bool called() const { return called_; }
	size_t number_of_reports() const { return stats_.number_of_reports; }
	double timestamp() const { return stats_.timestamp; }

	int AudioOutputLevel() const {
	RTC_CHECK(called_);
	return stats_.audio_output_level;
	}

	int AudioInputLevel() const {
	RTC_CHECK(called_);
	return stats_.audio_input_level;
	}

	int BytesReceived() const {
	RTC_CHECK(called_);
	return stats_.bytes_received;
	}

	int BytesSent() const {
	RTC_CHECK(called_);
	return stats_.bytes_sent;
	}

	int64_t CaptureStartNtpTime() const {
	RTC_CHECK(called_);
	return stats_.capture_start_ntp_time;
	}

	int AvailableReceiveBandwidth() const {
	RTC_CHECK(called_);
	return stats_.available_receive_bandwidth;
	}

	std::string DtlsCipher() const {
	RTC_CHECK(called_);
	return stats_.dtls_cipher;
	}

	std::string SrtpCipher() const {
	RTC_CHECK(called_);
	return stats_.srtp_cipher;
	}

	std::vector<std::string> TrackIds() const {
	RTC_CHECK(called_);
	return stats_.track_ids;
	}

	private:
	bool GetIntValue(const StatsReport* report,
	StatsReport::StatsValueName name,
	int* value) {
	const StatsReport::Value* v = report->FindValue(name);
	if (v) {
	// TODO(tommi): We should really just be using an int here :-/
	*value = rtc::FromString<int>(v->ToString());
	}
	return v != nullptr;
	}

	bool GetInt64Value(const StatsReport* report,
	StatsReport::StatsValueName name,
	int64_t* value) {
	const StatsReport::Value* v = report->FindValue(name);
	if (v) {
	// TODO(tommi): We should really just be using an int here :-/
	*value = rtc::FromString<int64_t>(v->ToString());
	}
	return v != nullptr;
	}

	bool GetStringValue(const StatsReport* report,
	StatsReport::StatsValueName name,
	std::string* value) {
	const StatsReport::Value* v = report->FindValue(name);
	if (v)
	*value = v->ToString();
	return v != nullptr;
	}

	bool called_;
	struct {
	void Clear() {
	number_of_reports = 0;
	timestamp = 0;
	audio_output_level = 0;
	audio_input_level = 0;
	bytes_received = 0;
	bytes_sent = 0;
	capture_start_ntp_time = 0;
	available_receive_bandwidth = 0;
	dtls_cipher.clear();
	srtp_cipher.clear();
	track_ids.clear();
	}

	size_t number_of_reports;
	double timestamp;
	int audio_output_level;
	int audio_input_level;
	int bytes_received;
	int bytes_sent;
	int64_t capture_start_ntp_time;
	int available_receive_bandwidth;
	std::string dtls_cipher;
	std::string srtp_cipher;
	std::vector<std::string> track_ids;
	} stats_;
	};

	// Helper class that just stores the report from the callback.
	class MockRTCStatsCollectorCallback : public webrtc::RTCStatsCollectorCallback {
	public:
	rtc::scoped_refptr<const RTCStatsReport> report() { return report_; }

	bool called() const { return called_; }

	protected:
	void OnStatsDelivered(
	const rtc::scoped_refptr<const RTCStatsReport>& report) override {
	report_ = report;
	called_ = true;
	}

	private:
	bool called_ = false;
	rtc::scoped_refptr<const RTCStatsReport> report_;
	};

	} // namespace webrtc

	#endif // PC_TEST_MOCK_PEER_CONNECTION_OBSERVERS_H_