@@ -18,11 +18,12 @@ class TransportFactoryMock : public transport::TransportFactory
1818 size_t expectedOutboundStreamCount,
1919 size_t jobQueueSize,
2020 bool enableUplinkEstimation,
21- bool enableDownlinkEstimation),
21+ bool enableDownlinkEstimation,
22+ bool privatePort),
2223 (override ));
2324 MOCK_METHOD (std::shared_ptr<transport::RtcTransport>,
2425 create,
25- (const ice::IceRole iceRole, const size_t endpointId),
26+ (const ice::IceRole iceRole, const size_t endpointId, bool privatePort ),
2627 (override ));
2728
2829 MOCK_METHOD (std::shared_ptr<transport::RtcTransport>, create, (const size_t endpointIdHash), (override ));
@@ -100,8 +101,8 @@ class TransportFactoryMock : public transport::TransportFactory
100101 using namespace ::testing;
101102
102103 ON_CALL (*this , create (_)).WillByDefault (Return (rtcTransport));
103- ON_CALL (*this , create (_, _)).WillByDefault (Return (rtcTransport));
104- ON_CALL (*this , create (_, _, _, _, _, _, _)).WillByDefault (Return (rtcTransport));
104+ ON_CALL (*this , create (_, _, _ )).WillByDefault (Return (rtcTransport));
105+ ON_CALL (*this , create (_, _, _, _, _, _, _, _ )).WillByDefault (Return (rtcTransport));
105106 ON_CALL (*this , createOnSharedPort (_, _)).WillByDefault (Return (rtcTransport));
106107 ON_CALL (*this , createOnSharedPort (_, _, _, _, _, _, _)).WillByDefault (Return (rtcTransport));
107108 ON_CALL (*this , createOnPrivatePort (_, _)).WillByDefault (Return (rtcTransport));
@@ -112,26 +113,17 @@ class TransportFactoryMock : public transport::TransportFactory
112113 {
113114 using namespace ::testing;
114115
115- const auto callback1Args = [=](const auto &) {
116- return rtcTransport.lock ();
117- };
118-
119- const auto callback2Args = [=](const auto &, const auto &) {
116+ const auto callbackNArgs = [=](const auto &... args) {
120117 return rtcTransport.lock ();
121118 };
122119
123- const auto callback7Args =
124- [=](const auto &, const auto &, const auto &, const auto &, const auto &, const auto &, const auto &) {
125- return rtcTransport.lock ();
126- };
127-
128- ON_CALL (*this , create (_)).WillByDefault (callback1Args);
129- ON_CALL (*this , create (_, _)).WillByDefault (callback2Args);
130- ON_CALL (*this , create (_, _, _, _, _, _, _)).WillByDefault (callback7Args);
131- ON_CALL (*this , createOnSharedPort (_, _)).WillByDefault (callback2Args);
132- ON_CALL (*this , createOnSharedPort (_, _, _, _, _, _, _)).WillByDefault (callback7Args);
133- ON_CALL (*this , createOnPrivatePort (_, _)).WillByDefault (callback2Args);
134- ON_CALL (*this , createOnPrivatePort (_, _, _, _, _, _, _)).WillByDefault (callback7Args);
120+ ON_CALL (*this , create (_)).WillByDefault (callbackNArgs);
121+ ON_CALL (*this , create (_, _, _)).WillByDefault (callbackNArgs);
122+ ON_CALL (*this , create (_, _, _, _, _, _, _, _)).WillByDefault (callbackNArgs);
123+ ON_CALL (*this , createOnSharedPort (_, _)).WillByDefault (callbackNArgs);
124+ ON_CALL (*this , createOnSharedPort (_, _, _, _, _, _, _)).WillByDefault (callbackNArgs);
125+ ON_CALL (*this , createOnPrivatePort (_, _)).WillByDefault (callbackNArgs);
126+ ON_CALL (*this , createOnPrivatePort (_, _, _, _, _, _, _)).WillByDefault (callbackNArgs);
135127 }
136128};
137129
0 commit comments