| OLD | NEW |
| 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2009 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "net/base/address_list.h" | 5 #include "net/base/address_list.h" |
| 6 | 6 |
| 7 #if defined(OS_WIN) | |
| 8 #include <ws2tcpip.h> | |
| 9 #include <wspiapi.h> // Needed for Win2k compat. | |
| 10 #elif defined(OS_POSIX) | |
| 11 #include <netdb.h> | |
| 12 #include <sys/socket.h> | |
| 13 #endif | |
| 14 | |
| 15 #include "base/string_util.h" | 7 #include "base/string_util.h" |
| 8 #include "net/base/host_resolver_proc.h" |
| 16 #include "net/base/net_util.h" | 9 #include "net/base/net_util.h" |
| 17 #if defined(OS_WIN) | 10 #if defined(OS_WIN) |
| 18 #include "net/base/winsock_init.h" | 11 #include "net/base/winsock_init.h" |
| 19 #endif | 12 #endif |
| 20 #include "testing/gtest/include/gtest/gtest.h" | 13 #include "testing/gtest/include/gtest/gtest.h" |
| 21 | 14 |
| 22 namespace { | 15 namespace { |
| 23 | 16 |
| 24 // Use getaddrinfo() to allocate an addrinfo structure. | 17 // Use getaddrinfo() to allocate an addrinfo structure. |
| 25 void CreateAddressList(net::AddressList* addrlist, int port) { | 18 void CreateAddressList(net::AddressList* addrlist, int port) { |
| 26 #if defined(OS_WIN) | 19 #if defined(OS_WIN) |
| 27 net::EnsureWinsockInit(); | 20 net::EnsureWinsockInit(); |
| 28 #endif | 21 #endif |
| 29 std::string portstr = IntToString(port); | 22 int rv = SystemHostResolverProc("192.168.1.1", addrlist); |
| 30 | 23 EXPECT_EQ(0, rv); |
| 31 struct addrinfo* result = NULL; | 24 addrlist->SetPort(port); |
| 32 struct addrinfo hints = {0}; | |
| 33 hints.ai_family = AF_UNSPEC; | |
| 34 hints.ai_flags = AI_NUMERICHOST; | |
| 35 hints.ai_socktype = SOCK_STREAM; | |
| 36 | |
| 37 int err = getaddrinfo("192.168.1.1", portstr.c_str(), &hints, &result); | |
| 38 EXPECT_EQ(0, err); | |
| 39 addrlist->Adopt(result); | |
| 40 } | 25 } |
| 41 | 26 |
| 42 TEST(AddressListTest, GetPort) { | 27 TEST(AddressListTest, GetPort) { |
| 43 net::AddressList addrlist; | 28 net::AddressList addrlist; |
| 44 CreateAddressList(&addrlist, 81); | 29 CreateAddressList(&addrlist, 81); |
| 45 EXPECT_EQ(81, addrlist.GetPort()); | 30 EXPECT_EQ(81, addrlist.GetPort()); |
| 46 | 31 |
| 47 addrlist.SetPort(83); | 32 addrlist.SetPort(83); |
| 48 EXPECT_EQ(83, addrlist.GetPort()); | 33 EXPECT_EQ(83, addrlist.GetPort()); |
| 49 } | 34 } |
| (...skipping 27 matching lines...) Expand all Loading... |
| 77 | 62 |
| 78 // Changes to addrlist1 are not reflected in addrlist2. | 63 // Changes to addrlist1 are not reflected in addrlist2. |
| 79 addrlist1.SetPort(70); | 64 addrlist1.SetPort(70); |
| 80 addrlist2.SetPort(90); | 65 addrlist2.SetPort(90); |
| 81 | 66 |
| 82 EXPECT_EQ(70, addrlist1.GetPort()); | 67 EXPECT_EQ(70, addrlist1.GetPort()); |
| 83 EXPECT_EQ(90, addrlist2.GetPort()); | 68 EXPECT_EQ(90, addrlist2.GetPort()); |
| 84 } | 69 } |
| 85 | 70 |
| 86 } // namespace | 71 } // namespace |
| OLD | NEW |