64 lines
2.1 KiB
C++
64 lines
2.1 KiB
C++
/*
|
|
* Copyright 2019 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.
|
|
*/
|
|
|
|
#include "pc/ice_transport.h"
|
|
|
|
#include <memory>
|
|
#include <utility>
|
|
|
|
#include "api/ice_transport_factory.h"
|
|
#include "api/make_ref_counted.h"
|
|
#include "api/scoped_refptr.h"
|
|
#include "p2p/test/fake_ice_transport.h"
|
|
#include "p2p/test/fake_port_allocator.h"
|
|
#include "rtc_base/internal/default_socket_server.h"
|
|
#include "test/gtest.h"
|
|
#include "test/scoped_key_value_config.h"
|
|
|
|
namespace webrtc {
|
|
|
|
class IceTransportTest : public ::testing::Test {
|
|
protected:
|
|
IceTransportTest()
|
|
: socket_server_(rtc::CreateDefaultSocketServer()),
|
|
main_thread_(socket_server_.get()) {}
|
|
|
|
rtc::SocketServer* socket_server() const { return socket_server_.get(); }
|
|
|
|
test::ScopedKeyValueConfig field_trials_;
|
|
|
|
private:
|
|
std::unique_ptr<rtc::SocketServer> socket_server_;
|
|
rtc::AutoSocketServerThread main_thread_;
|
|
};
|
|
|
|
TEST_F(IceTransportTest, CreateNonSelfDeletingTransport) {
|
|
auto cricket_transport =
|
|
std::make_unique<cricket::FakeIceTransport>("name", 0, nullptr);
|
|
auto ice_transport =
|
|
rtc::make_ref_counted<IceTransportWithPointer>(cricket_transport.get());
|
|
EXPECT_EQ(ice_transport->internal(), cricket_transport.get());
|
|
ice_transport->Clear();
|
|
EXPECT_NE(ice_transport->internal(), cricket_transport.get());
|
|
}
|
|
|
|
TEST_F(IceTransportTest, CreateSelfDeletingTransport) {
|
|
std::unique_ptr<cricket::FakePortAllocator> port_allocator(
|
|
std::make_unique<cricket::FakePortAllocator>(
|
|
nullptr,
|
|
std::make_unique<rtc::BasicPacketSocketFactory>(socket_server()),
|
|
&field_trials_));
|
|
IceTransportInit init;
|
|
init.set_port_allocator(port_allocator.get());
|
|
auto ice_transport = CreateIceTransport(std::move(init));
|
|
EXPECT_NE(nullptr, ice_transport->internal());
|
|
}
|
|
|
|
} // namespace webrtc
|