From e217217bf3782dbd3ea27be33d5ebd26ae0500ca Mon Sep 17 00:00:00 2001
From: Harald Alvestrand <hta@webrtc.org>
Date: Fri, 22 Apr 2022 08:37:14 +0000
Subject: [PATCH] Add canonical mock for MediaStreamInterface

Needed to let upstream APIs integrate changes more easily.

Bug: webrtc:13980
Change-Id: I6cd46f75d56597c10e08d0d66e16089516f5129c
Reviewed-on: https://webrtc-review.googlesource.com/c/src/+/259821
Commit-Queue: Harald Alvestrand <hta@webrtc.org>
Reviewed-by: Niels Moller <nisse@webrtc.org>
Cr-Commit-Position: refs/heads/main@{#36613}
---
 api/test/mock_media_stream_interface.h | 46 ++++++++++++++++++++++++++
 1 file changed, 46 insertions(+)

diff --git a/api/test/mock_media_stream_interface.h b/api/test/mock_media_stream_interface.h
index 17a30a877e..209962358d 100644
--- a/api/test/mock_media_stream_interface.h
+++ b/api/test/mock_media_stream_interface.h
@@ -84,6 +84,52 @@ class MockAudioTrack final : public rtc::RefCountedObject<AudioTrackInterface> {
   MockAudioTrack() = default;
 };
 
+class MockMediaStream : public MediaStreamInterface {
+ public:
+  MOCK_METHOD(std::string, id, (), (const override));
+  MOCK_METHOD(AudioTrackVector, GetAudioTracks, (), (override));
+  MOCK_METHOD(VideoTrackVector, GetVideoTracks, (), (override));
+  MOCK_METHOD(rtc::scoped_refptr<AudioTrackInterface>,
+              FindAudioTrack,
+              (const std::string& track_id),
+              (override));
+  MOCK_METHOD(rtc::scoped_refptr<VideoTrackInterface>,
+              FindVideoTrack,
+              (const std::string& track_id),
+              (override));
+  MOCK_METHOD(bool,
+              AddTrack,
+              (rtc::scoped_refptr<AudioTrackInterface> track),
+              (override));
+  MOCK_METHOD(bool,
+              AddTrack,
+              (rtc::scoped_refptr<VideoTrackInterface> track),
+              (override));
+  MOCK_METHOD(bool,
+              RemoveTrack,
+              (rtc::scoped_refptr<AudioTrackInterface> track),
+              (override));
+  MOCK_METHOD(bool,
+              RemoveTrack,
+              (rtc::scoped_refptr<VideoTrackInterface> track),
+              (override));
+  // Old AddTrack/RemoveTrack methods - slated for removal
+  MOCK_METHOD(bool, AddTrack, (AudioTrackInterface * track), (override));
+  MOCK_METHOD(bool, AddTrack, (VideoTrackInterface * track), (override));
+  MOCK_METHOD(bool, RemoveTrack, (AudioTrackInterface * track), (override));
+  MOCK_METHOD(bool, RemoveTrack, (VideoTrackInterface * track), (override));
+  MOCK_METHOD(void,
+              RegisterObserver,
+              (ObserverInterface * observer),
+              (override));
+  MOCK_METHOD(void,
+              UnregisterObserver,
+              (ObserverInterface * observer),
+              (override));
+};
+
+static_assert(!std::is_abstract_v<rtc::RefCountedObject<MockMediaStream>>, "");
+
 }  // namespace webrtc
 
 #endif  // API_TEST_MOCK_MEDIA_STREAM_INTERFACE_H_