| // Copyright 2018 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 THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_ADAPTERS_TEST_MOCK_P2P_QUIC_TRANSPORT_FACTORY_H_ |
| #define THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_ADAPTERS_TEST_MOCK_P2P_QUIC_TRANSPORT_FACTORY_H_ |
| |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "third_party/blink/renderer/modules/peerconnection/adapters/p2p_quic_transport_factory.h" |
| #include "third_party/blink/renderer/modules/peerconnection/adapters/test/mock_p2p_quic_transport.h" |
| |
| namespace blink { |
| |
| class MockP2PQuicTransportFactory |
| : public testing::NiceMock<P2PQuicTransportFactory> { |
| public: |
| MockP2PQuicTransportFactory() = default; |
| MockP2PQuicTransportFactory( |
| std::unique_ptr<MockP2PQuicTransport> mock_transport, |
| P2PQuicTransport::Delegate** delegate_out = nullptr) |
| : mock_transport_(std::move(mock_transport)), |
| delegate_out_(delegate_out) { |
| if (delegate_out) { |
| // Ensure the caller has not left the delegate_out value floating. |
| DCHECK_EQ(nullptr, *delegate_out); |
| } |
| ON_CALL(*this, CreateQuicTransport(testing::_, testing::_, testing::_)) |
| .WillByDefault( |
| testing::Invoke([this](P2PQuicTransport::Delegate* delegate, |
| P2PQuicPacketTransport* packet_transport, |
| const P2PQuicTransportConfig& config) { |
| DCHECK(mock_transport_); |
| if (delegate_out_) { |
| *delegate_out_ = delegate; |
| } |
| return std::move(mock_transport_); |
| })); |
| } |
| |
| // P2PQuicTransportFactory overrides. |
| MOCK_METHOD3(CreateQuicTransport, |
| std::unique_ptr<P2PQuicTransport>( |
| P2PQuicTransport::Delegate* delegate, |
| P2PQuicPacketTransport* packet_transport, |
| const P2PQuicTransportConfig& config)); |
| |
| private: |
| std::unique_ptr<MockP2PQuicTransport> mock_transport_; |
| P2PQuicTransport::Delegate** delegate_out_ = nullptr; |
| }; |
| |
| } // namespace blink |
| |
| #endif // THIRD_PARTY_BLINK_RENDERER_MODULES_PEERCONNECTION_ADAPTERS_TEST_MOCK_P2P_QUIC_TRANSPORT_FACTORY_H_ |