From b2c4700d39fbedaff9bdbee934e1f3f8032bb35b Mon Sep 17 00:00:00 2001
From: Harald Alvestrand <hta@webrtc.org>
Date: Sun, 3 Mar 2019 18:50:22 +0100
Subject: [PATCH] Another mock for GetSctpTransport

Bug: chromium:818643
Change-Id: I4ae7826efa7afa8e7b2ecd8a5928071a1b913ded
Reviewed-on: https://webrtc-review.googlesource.com/c/125340
Reviewed-by: Karl Wiberg <kwiberg@webrtc.org>
Commit-Queue: Harald Alvestrand <hta@webrtc.org>
Cr-Commit-Position: refs/heads/master@{#26941}
---
 api/test/mock_peerconnectioninterface.h | 2 ++
 pc/test/fake_peer_connection_base.h     | 5 +++++
 2 files changed, 7 insertions(+)

diff --git a/api/test/mock_peerconnectioninterface.h b/api/test/mock_peerconnectioninterface.h
index 86abad5eac..bac9f08027 100644
--- a/api/test/mock_peerconnectioninterface.h
+++ b/api/test/mock_peerconnectioninterface.h
@@ -74,6 +74,8 @@ class MockPeerConnectionInterface
                void(rtc::scoped_refptr<RtpReceiverInterface>,
                     rtc::scoped_refptr<RTCStatsCollectorCallback>));
   MOCK_METHOD0(ClearStatsCache, void());
+  MOCK_CONST_METHOD0(GetSctpTransport,
+                     rtc::scoped_refptr<SctpTransportInterface>());
   MOCK_METHOD2(
       CreateDataChannel,
       rtc::scoped_refptr<DataChannelInterface>(const std::string&,
diff --git a/pc/test/fake_peer_connection_base.h b/pc/test/fake_peer_connection_base.h
index 85feebc358..d2fa1fd7d8 100644
--- a/pc/test/fake_peer_connection_base.h
+++ b/pc/test/fake_peer_connection_base.h
@@ -17,6 +17,7 @@
 #include <string>
 #include <vector>
 
+#include "api/sctp_transport_interface.h"
 #include "pc/peer_connection_internal.h"
 
 namespace webrtc {
@@ -108,6 +109,10 @@ class FakePeerConnectionBase : public PeerConnectionInternal {
 
   void ClearStatsCache() override {}
 
+  rtc::scoped_refptr<SctpTransportInterface> GetSctpTransport() const {
+    return nullptr;
+  }
+
   rtc::scoped_refptr<DataChannelInterface> CreateDataChannel(
       const std::string& label,
       const DataChannelInit* config) override {