| OLD | NEW |
| 1 // Copyright (c) 2011 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2011 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 "content/renderer/p2p/p2p_transport_impl.h" | 5 #include "content/renderer/p2p/p2p_transport_impl.h" |
| 6 | 6 |
| 7 #include "base/bind.h" |
| 7 #include "base/compiler_specific.h" | 8 #include "base/compiler_specific.h" |
| 8 #include "base/memory/ref_counted.h" | 9 #include "base/memory/ref_counted.h" |
| 9 #include "base/message_loop.h" | 10 #include "base/message_loop.h" |
| 10 #include "base/test/test_timeouts.h" | 11 #include "base/test/test_timeouts.h" |
| 11 #include "jingle/glue/fake_network_manager.h" | 12 #include "jingle/glue/fake_network_manager.h" |
| 12 #include "jingle/glue/fake_socket_factory.h" | 13 #include "jingle/glue/fake_socket_factory.h" |
| 13 #include "jingle/glue/thread_wrapper.h" | 14 #include "jingle/glue/thread_wrapper.h" |
| 14 #include "net/base/completion_callback.h" | 15 #include "net/base/completion_callback.h" |
| 15 #include "net/base/io_buffer.h" | 16 #include "net/base/io_buffer.h" |
| 16 #include "net/base/net_errors.h" | 17 #include "net/base/net_errors.h" |
| (...skipping 41 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 58 read_errors_(0), | 59 read_errors_(0), |
| 59 packets_sent_(0), | 60 packets_sent_(0), |
| 60 packets_received_(0), | 61 packets_received_(0), |
| 61 broken_packets_(0) { | 62 broken_packets_(0) { |
| 62 } | 63 } |
| 63 | 64 |
| 64 virtual ~UdpChannelTester() { } | 65 virtual ~UdpChannelTester() { } |
| 65 | 66 |
| 66 void Start() { | 67 void Start() { |
| 67 message_loop_->PostTask( | 68 message_loop_->PostTask( |
| 68 FROM_HERE, NewRunnableMethod(this, &UdpChannelTester::DoStart)); | 69 FROM_HERE, base::Bind(&UdpChannelTester::DoStart, this)); |
| 69 } | 70 } |
| 70 | 71 |
| 71 void CheckResults() { | 72 void CheckResults() { |
| 72 EXPECT_EQ(0, write_errors_); | 73 EXPECT_EQ(0, write_errors_); |
| 73 EXPECT_EQ(0, read_errors_); | 74 EXPECT_EQ(0, read_errors_); |
| 74 | 75 |
| 75 EXPECT_EQ(0, broken_packets_); | 76 EXPECT_EQ(0, broken_packets_); |
| 76 | 77 |
| 77 // Verify that we've received at least one packet. | 78 // Verify that we've received at least one packet. |
| 78 EXPECT_GT(packets_received_, 0); | 79 EXPECT_GT(packets_received_, 0); |
| (...skipping 34 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 113 | 114 |
| 114 void HandleWriteResult(int result) { | 115 void HandleWriteResult(int result) { |
| 115 if (result <= 0 && result != net::ERR_IO_PENDING) { | 116 if (result <= 0 && result != net::ERR_IO_PENDING) { |
| 116 LOG(ERROR) << "Received error " << result << " when trying to write"; | 117 LOG(ERROR) << "Received error " << result << " when trying to write"; |
| 117 write_errors_++; | 118 write_errors_++; |
| 118 Done(); | 119 Done(); |
| 119 } else if (result > 0) { | 120 } else if (result > 0) { |
| 120 EXPECT_EQ(kMessageSize, result); | 121 EXPECT_EQ(kMessageSize, result); |
| 121 packets_sent_++; | 122 packets_sent_++; |
| 122 message_loop_->PostDelayedTask( | 123 message_loop_->PostDelayedTask( |
| 123 FROM_HERE, NewRunnableMethod(this, &UdpChannelTester::DoWrite), | 124 FROM_HERE, base::Bind(&UdpChannelTester::DoWrite, this), |
| 124 kUdpWriteDelayMs); | 125 kUdpWriteDelayMs); |
| 125 } | 126 } |
| 126 } | 127 } |
| 127 | 128 |
| 128 void DoRead() { | 129 void DoRead() { |
| 129 int result = 1; | 130 int result = 1; |
| 130 while (result > 0) { | 131 while (result > 0) { |
| 131 int kReadSize = kMessageSize * 2; | 132 int kReadSize = kMessageSize * 2; |
| 132 read_buffer_ = new net::IOBuffer(kReadSize); | 133 read_buffer_ = new net::IOBuffer(kReadSize); |
| 133 | 134 |
| (...skipping 76 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 210 // Initialize |send_buffer_|. | 211 // Initialize |send_buffer_|. |
| 211 send_buffer_ = new net::DrainableIOBuffer(new net::IOBuffer(kTcpDataSize), | 212 send_buffer_ = new net::DrainableIOBuffer(new net::IOBuffer(kTcpDataSize), |
| 212 kTcpDataSize); | 213 kTcpDataSize); |
| 213 for (int i = 0; i < kTcpDataSize; ++i) { | 214 for (int i = 0; i < kTcpDataSize; ++i) { |
| 214 send_buffer_->data()[i] = rand() % 256; | 215 send_buffer_->data()[i] = rand() % 256; |
| 215 } | 216 } |
| 216 } | 217 } |
| 217 | 218 |
| 218 void StartRead() { | 219 void StartRead() { |
| 219 message_loop_->PostTask( | 220 message_loop_->PostTask( |
| 220 FROM_HERE, NewRunnableMethod(this, &TcpChannelTester::DoRead)); | 221 FROM_HERE, base::Bind(&TcpChannelTester::DoRead, this)); |
| 221 } | 222 } |
| 222 | 223 |
| 223 void StartWrite() { | 224 void StartWrite() { |
| 224 message_loop_->PostTask( | 225 message_loop_->PostTask( |
| 225 FROM_HERE, NewRunnableMethod(this, &TcpChannelTester::DoWrite)); | 226 FROM_HERE, base::Bind(&TcpChannelTester::DoWrite, this)); |
| 226 } | 227 } |
| 227 | 228 |
| 228 void CheckResults() { | 229 void CheckResults() { |
| 229 EXPECT_EQ(0, write_errors_); | 230 EXPECT_EQ(0, write_errors_); |
| 230 EXPECT_EQ(0, read_errors_); | 231 EXPECT_EQ(0, read_errors_); |
| 231 | 232 |
| 232 EXPECT_EQ(0, send_buffer_->BytesRemaining()); | 233 EXPECT_EQ(0, send_buffer_->BytesRemaining()); |
| 233 | 234 |
| 234 send_buffer_->SetOffset(0); | 235 send_buffer_->SetOffset(0); |
| 235 EXPECT_EQ(kTcpDataSize, static_cast<int>(received_data_.size())); | 236 EXPECT_EQ(kTcpDataSize, static_cast<int>(received_data_.size())); |
| (...skipping 22 matching lines...) Expand all Loading... |
| 258 } | 259 } |
| 259 | 260 |
| 260 void HandleWriteResult(int result) { | 261 void HandleWriteResult(int result) { |
| 261 if (result <= 0 && result != net::ERR_IO_PENDING) { | 262 if (result <= 0 && result != net::ERR_IO_PENDING) { |
| 262 LOG(ERROR) << "Received error " << result << " when trying to write"; | 263 LOG(ERROR) << "Received error " << result << " when trying to write"; |
| 263 write_errors_++; | 264 write_errors_++; |
| 264 Done(); | 265 Done(); |
| 265 } else if (result > 0) { | 266 } else if (result > 0) { |
| 266 send_buffer_->DidConsume(result); | 267 send_buffer_->DidConsume(result); |
| 267 message_loop_->PostDelayedTask( | 268 message_loop_->PostDelayedTask( |
| 268 FROM_HERE, NewRunnableMethod(this, &TcpChannelTester::DoWrite), | 269 FROM_HERE, base::Bind(&TcpChannelTester::DoWrite, this), |
| 269 kTcpWriteDelayMs); | 270 kTcpWriteDelayMs); |
| 270 } | 271 } |
| 271 } | 272 } |
| 272 | 273 |
| 273 void DoRead() { | 274 void DoRead() { |
| 274 int result = 1; | 275 int result = 1; |
| 275 while (result > 0) { | 276 while (result > 0) { |
| 276 int kReadSize = kMessageSize * 2; | 277 int kReadSize = kMessageSize * 2; |
| 277 read_buffer_ = new net::IOBuffer(kReadSize); | 278 read_buffer_ = new net::IOBuffer(kReadSize); |
| 278 | 279 |
| (...skipping 201 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 480 | 481 |
| 481 message_loop_.PostDelayedTask(FROM_HERE, new MessageLoop::QuitTask(), | 482 message_loop_.PostDelayedTask(FROM_HERE, new MessageLoop::QuitTask(), |
| 482 TestTimeouts::action_max_timeout_ms()); | 483 TestTimeouts::action_max_timeout_ms()); |
| 483 | 484 |
| 484 channel_tester->Init(); | 485 channel_tester->Init(); |
| 485 message_loop_.Run(); | 486 message_loop_.Run(); |
| 486 channel_tester->CheckResults(); | 487 channel_tester->CheckResults(); |
| 487 } | 488 } |
| 488 | 489 |
| 489 } // namespace content | 490 } // namespace content |
| OLD | NEW |