Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(326)

Side by Side Diff: net/dns/dns_session_unittest.cc

Issue 266243004: Clang format slam. Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: Created 6 years, 7 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch | Annotate | Revision Log
OLDNEW
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
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
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
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
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
OLDNEW

Powered by Google App Engine
This is Rietveld 408576698