Index: chrome/browser/extensions/api/socket/udp_socket_unittest.cc |
diff --git a/chrome/browser/extensions/api/socket/udp_socket_unittest.cc b/chrome/browser/extensions/api/socket/udp_socket_unittest.cc |
index 980e3e8cfd4e5633b68a7ff4ee6e6eb17cc9221f..b8c50a1d8e68ef99621b254d0cd257a9bf431074 100644 |
--- a/chrome/browser/extensions/api/socket/udp_socket_unittest.cc |
+++ b/chrome/browser/extensions/api/socket/udp_socket_unittest.cc |
@@ -7,6 +7,7 @@ |
#include "base/memory/scoped_ptr.h" |
#include "chrome/browser/extensions/api/api_resource_event_notifier.h" |
#include "net/base/completion_callback.h" |
+#include "net/base/io_buffer.h" |
#include "net/base/net_errors.h" |
#include "net/base/rand_callback.h" |
#include "net/udp/udp_client_socket.h" |
@@ -56,7 +57,9 @@ TEST(SocketTest, TestUDPSocketRead) { |
EXPECT_CALL(*udp_client_socket, Read(_, _, _)) |
.Times(1); |
- std::string message = socket->Read(); |
+ scoped_refptr<net::IOBufferWithSize> io_buffer( |
+ new net::IOBufferWithSize(512)); |
+ socket->Read(io_buffer.get(), io_buffer->size()); |
} |
TEST(SocketTest, TestUDPSocketWrite) { |
@@ -69,7 +72,9 @@ TEST(SocketTest, TestUDPSocketWrite) { |
EXPECT_CALL(*udp_client_socket, Write(_, _, _)) |
.Times(1); |
- socket->Write("foo"); |
+ scoped_refptr<net::IOBufferWithSize> io_buffer( |
+ new net::IOBufferWithSize(512)); |
+ socket->Write(io_buffer.get(), io_buffer->size()); |
} |
TEST(SocketTest, TestUDPSocketBlockedWrite) { |
@@ -85,7 +90,9 @@ TEST(SocketTest, TestUDPSocketBlockedWrite) { |
.WillOnce(testing::DoAll(SaveArg<2>(&callback), |
Return(net::ERR_IO_PENDING))); |
- ASSERT_EQ(net::ERR_IO_PENDING, socket->Write("foo")); |
+ scoped_refptr<net::IOBufferWithSize> io_buffer(new net::IOBufferWithSize(1)); |
+ ASSERT_EQ(net::ERR_IO_PENDING, socket->Write(io_buffer.get(), |
+ io_buffer->size())); |
// Good. Original call came back unable to complete. Now pretend the socket |
// finished, and confirm that we passed the error back. |