| OLD | NEW |
| 1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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/dns/dns_session.h" | 5 #include "net/dns/dns_session.h" |
| 6 | 6 |
| 7 #include <list> | 7 #include <list> |
| 8 | 8 |
| 9 #include "base/bind.h" | 9 #include "base/bind.h" |
| 10 #include "base/memory/scoped_ptr.h" | 10 #include "base/memory/scoped_ptr.h" |
| (...skipping 16 matching lines...) Expand all Loading... |
| 27 virtual ~TestClientSocketFactory(); | 27 virtual ~TestClientSocketFactory(); |
| 28 | 28 |
| 29 virtual scoped_ptr<DatagramClientSocket> CreateDatagramClientSocket( | 29 virtual scoped_ptr<DatagramClientSocket> CreateDatagramClientSocket( |
| 30 DatagramSocket::BindType bind_type, | 30 DatagramSocket::BindType bind_type, |
| 31 const RandIntCallback& rand_int_cb, | 31 const RandIntCallback& rand_int_cb, |
| 32 net::NetLog* net_log, | 32 net::NetLog* net_log, |
| 33 const net::NetLog::Source& source) OVERRIDE; | 33 const net::NetLog::Source& source) OVERRIDE; |
| 34 | 34 |
| 35 virtual scoped_ptr<StreamSocket> CreateTransportClientSocket( | 35 virtual scoped_ptr<StreamSocket> CreateTransportClientSocket( |
| 36 const AddressList& addresses, | 36 const AddressList& addresses, |
| 37 NetLog*, const NetLog::Source&) OVERRIDE { | 37 NetLog*, |
| 38 const NetLog::Source&) OVERRIDE { |
| 38 NOTIMPLEMENTED(); | 39 NOTIMPLEMENTED(); |
| 39 return scoped_ptr<StreamSocket>(); | 40 return scoped_ptr<StreamSocket>(); |
| 40 } | 41 } |
| 41 | 42 |
| 42 virtual scoped_ptr<SSLClientSocket> CreateSSLClientSocket( | 43 virtual scoped_ptr<SSLClientSocket> CreateSSLClientSocket( |
| 43 scoped_ptr<ClientSocketHandle> transport_socket, | 44 scoped_ptr<ClientSocketHandle> transport_socket, |
| 44 const HostPortPair& host_and_port, | 45 const HostPortPair& host_and_port, |
| 45 const SSLConfig& ssl_config, | 46 const SSLConfig& ssl_config, |
| 46 const SSLClientSocketContext& context) OVERRIDE { | 47 const SSLClientSocketContext& context) OVERRIDE { |
| 47 NOTIMPLEMENTED(); | 48 NOTIMPLEMENTED(); |
| 48 return scoped_ptr<SSLClientSocket>(); | 49 return scoped_ptr<SSLClientSocket>(); |
| 49 } | 50 } |
| 50 | 51 |
| 51 virtual void ClearSSLSessionCache() OVERRIDE { | 52 virtual void ClearSSLSessionCache() OVERRIDE { NOTIMPLEMENTED(); } |
| 52 NOTIMPLEMENTED(); | |
| 53 } | |
| 54 | 53 |
| 55 private: | 54 private: |
| 56 std::list<SocketDataProvider*> data_providers_; | 55 std::list<SocketDataProvider*> data_providers_; |
| 57 }; | 56 }; |
| 58 | 57 |
| 59 struct PoolEvent { | 58 struct PoolEvent { |
| 60 enum { ALLOCATE, FREE } action; | 59 enum { ALLOCATE, FREE } action; |
| 61 unsigned server_index; | 60 unsigned server_index; |
| 62 }; | 61 }; |
| 63 | 62 |
| (...skipping 15 matching lines...) Expand all Loading... |
| 79 NetLog::Source source_; | 78 NetLog::Source source_; |
| 80 | 79 |
| 81 private: | 80 private: |
| 82 bool ExpectEvent(const PoolEvent& event); | 81 bool ExpectEvent(const PoolEvent& event); |
| 83 std::list<PoolEvent> events_; | 82 std::list<PoolEvent> events_; |
| 84 }; | 83 }; |
| 85 | 84 |
| 86 class MockDnsSocketPool : public DnsSocketPool { | 85 class MockDnsSocketPool : public DnsSocketPool { |
| 87 public: | 86 public: |
| 88 MockDnsSocketPool(ClientSocketFactory* factory, DnsSessionTest* test) | 87 MockDnsSocketPool(ClientSocketFactory* factory, DnsSessionTest* test) |
| 89 : DnsSocketPool(factory), test_(test) { } | 88 : DnsSocketPool(factory), test_(test) {} |
| 90 | 89 |
| 91 virtual ~MockDnsSocketPool() { } | 90 virtual ~MockDnsSocketPool() {} |
| 92 | 91 |
| 93 virtual void Initialize( | 92 virtual void Initialize(const std::vector<IPEndPoint>* nameservers, |
| 94 const std::vector<IPEndPoint>* nameservers, | 93 NetLog* net_log) OVERRIDE { |
| 95 NetLog* net_log) OVERRIDE { | |
| 96 InitializeInternal(nameservers, net_log); | 94 InitializeInternal(nameservers, net_log); |
| 97 } | 95 } |
| 98 | 96 |
| 99 virtual scoped_ptr<DatagramClientSocket> AllocateSocket( | 97 virtual scoped_ptr<DatagramClientSocket> AllocateSocket( |
| 100 unsigned server_index) OVERRIDE { | 98 unsigned server_index) OVERRIDE { |
| 101 test_->OnSocketAllocated(server_index); | 99 test_->OnSocketAllocated(server_index); |
| 102 return CreateConnectedSocket(server_index); | 100 return CreateConnectedSocket(server_index); |
| 103 } | 101 } |
| 104 | 102 |
| 105 virtual void FreeSocket( | 103 virtual void FreeSocket(unsigned server_index, |
| 106 unsigned server_index, | 104 scoped_ptr<DatagramClientSocket> socket) OVERRIDE { |
| 107 scoped_ptr<DatagramClientSocket> socket) OVERRIDE { | |
| 108 test_->OnSocketFreed(server_index); | 105 test_->OnSocketFreed(server_index); |
| 109 } | 106 } |
| 110 | 107 |
| 111 private: | 108 private: |
| 112 DnsSessionTest* test_; | 109 DnsSessionTest* test_; |
| 113 }; | 110 }; |
| 114 | 111 |
| 115 void DnsSessionTest::Initialize(unsigned num_servers) { | 112 void DnsSessionTest::Initialize(unsigned num_servers) { |
| 116 CHECK(num_servers < 256u); | 113 CHECK(num_servers < 256u); |
| 117 config_.nameservers.clear(); | 114 config_.nameservers.clear(); |
| (...skipping 18 matching lines...) Expand all Loading... |
| 136 | 133 |
| 137 events_.clear(); | 134 events_.clear(); |
| 138 } | 135 } |
| 139 | 136 |
| 140 scoped_ptr<DnsSession::SocketLease> DnsSessionTest::Allocate( | 137 scoped_ptr<DnsSession::SocketLease> DnsSessionTest::Allocate( |
| 141 unsigned server_index) { | 138 unsigned server_index) { |
| 142 return session_->AllocateSocket(server_index, source_); | 139 return session_->AllocateSocket(server_index, source_); |
| 143 } | 140 } |
| 144 | 141 |
| 145 bool DnsSessionTest::DidAllocate(unsigned server_index) { | 142 bool DnsSessionTest::DidAllocate(unsigned server_index) { |
| 146 PoolEvent expected_event = { PoolEvent::ALLOCATE, server_index }; | 143 PoolEvent expected_event = {PoolEvent::ALLOCATE, server_index}; |
| 147 return ExpectEvent(expected_event); | 144 return ExpectEvent(expected_event); |
| 148 } | 145 } |
| 149 | 146 |
| 150 bool DnsSessionTest::DidFree(unsigned server_index) { | 147 bool DnsSessionTest::DidFree(unsigned server_index) { |
| 151 PoolEvent expected_event = { PoolEvent::FREE, server_index }; | 148 PoolEvent expected_event = {PoolEvent::FREE, server_index}; |
| 152 return ExpectEvent(expected_event); | 149 return ExpectEvent(expected_event); |
| 153 } | 150 } |
| 154 | 151 |
| 155 bool DnsSessionTest::NoMoreEvents() { | 152 bool DnsSessionTest::NoMoreEvents() { |
| 156 return events_.empty(); | 153 return events_.empty(); |
| 157 } | 154 } |
| 158 | 155 |
| 159 void DnsSessionTest::OnSocketAllocated(unsigned server_index) { | 156 void DnsSessionTest::OnSocketAllocated(unsigned server_index) { |
| 160 PoolEvent event = { PoolEvent::ALLOCATE, server_index }; | 157 PoolEvent event = {PoolEvent::ALLOCATE, server_index}; |
| 161 events_.push_back(event); | 158 events_.push_back(event); |
| 162 } | 159 } |
| 163 | 160 |
| 164 void DnsSessionTest::OnSocketFreed(unsigned server_index) { | 161 void DnsSessionTest::OnSocketFreed(unsigned server_index) { |
| 165 PoolEvent event = { PoolEvent::FREE, server_index }; | 162 PoolEvent event = {PoolEvent::FREE, server_index}; |
| 166 events_.push_back(event); | 163 events_.push_back(event); |
| 167 } | 164 } |
| 168 | 165 |
| 169 bool DnsSessionTest::ExpectEvent(const PoolEvent& expected) { | 166 bool DnsSessionTest::ExpectEvent(const PoolEvent& expected) { |
| 170 if (events_.empty()) { | 167 if (events_.empty()) { |
| 171 return false; | 168 return false; |
| 172 } | 169 } |
| 173 | 170 |
| 174 const PoolEvent actual = events_.front(); | 171 const PoolEvent actual = events_.front(); |
| 175 if ((expected.action != actual.action) | 172 if ((expected.action != actual.action) || |
| 176 || (expected.server_index != actual.server_index)) { | 173 (expected.server_index != actual.server_index)) { |
| 177 return false; | 174 return false; |
| 178 } | 175 } |
| 179 events_.pop_front(); | 176 events_.pop_front(); |
| 180 | 177 |
| 181 return true; | 178 return true; |
| 182 } | 179 } |
| 183 | 180 |
| 184 scoped_ptr<DatagramClientSocket> | 181 scoped_ptr<DatagramClientSocket> |
| 185 TestClientSocketFactory::CreateDatagramClientSocket( | 182 TestClientSocketFactory::CreateDatagramClientSocket( |
| 186 DatagramSocket::BindType bind_type, | 183 DatagramSocket::BindType bind_type, |
| (...skipping 55 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 242 // Expect long calculated timeout to be equal to one in DnsConfig. | 239 // Expect long calculated timeout to be equal to one in DnsConfig. |
| 243 TEST_F(DnsSessionTest, HistogramTimeoutLong) { | 240 TEST_F(DnsSessionTest, HistogramTimeoutLong) { |
| 244 config_.timeout = base::TimeDelta::FromSeconds(15); | 241 config_.timeout = base::TimeDelta::FromSeconds(15); |
| 245 Initialize(2); | 242 Initialize(2); |
| 246 base::TimeDelta timeout = session_->NextTimeout(0, 0); | 243 base::TimeDelta timeout = session_->NextTimeout(0, 0); |
| 247 EXPECT_EQ(config_.timeout.InMilliseconds(), timeout.InMilliseconds()); | 244 EXPECT_EQ(config_.timeout.InMilliseconds(), timeout.InMilliseconds()); |
| 248 } | 245 } |
| 249 | 246 |
| 250 } // namespace | 247 } // namespace |
| 251 | 248 |
| 252 } // namespace net | 249 } // namespace net |
| OLD | NEW |