1 /*
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
3 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
11 // Tests for the UdpSocketManager interface.
12 // Note: This tests UdpSocketManager together with UdpSocketWrapper,
13 // due to the way the code is full of static-casts to the platform dependent
14 // subtypes.
15 // It also uses the static UdpSocketManager object.
16 // The most important property of these tests is that they do not leak memory.
17
18 #include "testing/gtest/include/gtest/gtest.h"
19 #include "webrtc/system_wrappers/include/trace.h"
20 #include "webrtc/test/channel_transport/udp_socket_manager_wrapper.h"
21 #include "webrtc/test/channel_transport/udp_socket_wrapper.h"
22
23 namespace webrtc {
24 namespace test {
25
TEST(UdpSocketManager,CreateCallsInitAndDoesNotLeakMemory)26 TEST(UdpSocketManager, CreateCallsInitAndDoesNotLeakMemory) {
27 int32_t id = 42;
28 uint8_t threads = 1;
29 UdpSocketManager* mgr = UdpSocketManager::Create(id, threads);
30 // Create is supposed to have called init on the object.
31 EXPECT_FALSE(mgr->Init(id, threads))
32 << "Init should return false since Create is supposed to call it.";
33 UdpSocketManager::Return();
34 }
35
36 // Creates a socket and adds it to the socket manager, and then removes it
37 // before destroying the socket manager.
TEST(UdpSocketManager,AddAndRemoveSocketDoesNotLeakMemory)38 TEST(UdpSocketManager, AddAndRemoveSocketDoesNotLeakMemory) {
39 int32_t id = 42;
40 uint8_t threads = 1;
41 UdpSocketManager* mgr = UdpSocketManager::Create(id, threads);
42 UdpSocketWrapper* socket =
43 UdpSocketWrapper::CreateSocket(id,
44 mgr,
45 NULL, // CallbackObj
46 NULL, // IncomingSocketCallback
47 false, // ipV6Enable
48 false); // disableGQOS
49 // The constructor will do AddSocket on the manager.
50 // RemoveSocket indirectly calls Delete.
51 EXPECT_EQ(true, mgr->RemoveSocket(socket));
52 UdpSocketManager::Return();
53 }
54
55 // Creates a socket and add it to the socket manager, but does not remove it
56 // before destroying the socket manager.
57 // On Posix, this destroys the socket.
58 // On Winsock2 Windows, it enters an infinite wait for all the sockets
59 // to go away.
TEST(UdpSocketManager,UnremovedSocketsGetCollectedAtManagerDeletion)60 TEST(UdpSocketManager, UnremovedSocketsGetCollectedAtManagerDeletion) {
61 #if defined(_WIN32)
62 // It's hard to test an infinite wait, so we don't.
63 #else
64 int32_t id = 42;
65 uint8_t threads = 1;
66 UdpSocketManager* mgr = UdpSocketManager::Create(id, threads);
67 UdpSocketWrapper* unused_socket = UdpSocketWrapper::CreateSocket(
68 id,
69 mgr,
70 NULL, // CallbackObj
71 NULL, // IncomingSocketCallback
72 false, // ipV6Enable
73 false); // disableGQOS
74 // The constructor will do AddSocket on the manager.
75 // Call a member funtion to work around "set but not used" compliation
76 // error on ChromeOS ARM.
77 unused_socket->SetEventToNull();
78 unused_socket = NULL;
79 UdpSocketManager::Return();
80 #endif
81 }
82
83 } // namespace test
84 } // namespace webrtc
85