Index: net/base/upload_bytes_element_reader_unittest.cc |
diff --git a/net/base/upload_bytes_element_reader_unittest.cc b/net/base/upload_bytes_element_reader_unittest.cc |
index 9e68febd493aa58f4157041e2f52241bb1422643..1a3c5d4ebe21c8a4aecaaed9c08737f5925ee68a 100644 |
--- a/net/base/upload_bytes_element_reader_unittest.cc |
+++ b/net/base/upload_bytes_element_reader_unittest.cc |
@@ -6,6 +6,7 @@ |
#include "base/basictypes.h" |
#include "base/memory/scoped_ptr.h" |
+#include "net/base/io_buffer.h" |
#include "net/base/net_errors.h" |
#include "testing/gtest/include/gtest/gtest.h" |
#include "testing/platform_test.h" |
@@ -31,8 +32,9 @@ class UploadBytesElementReaderTest : public PlatformTest { |
TEST_F(UploadBytesElementReaderTest, ReadPartially) { |
const size_t kHalfSize = bytes_.size() / 2; |
std::vector<char> buf(kHalfSize); |
+ scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]); |
EXPECT_EQ(static_cast<int>(buf.size()), |
- reader_->ReadSync(&buf[0], buf.size())); |
+ reader_->ReadSync(wrapped_buffer, buf.size())); |
EXPECT_EQ(bytes_.size() - buf.size(), reader_->BytesRemaining()); |
bytes_.resize(kHalfSize); // Resize to compare. |
EXPECT_EQ(bytes_, buf); |
@@ -40,19 +42,21 @@ TEST_F(UploadBytesElementReaderTest, ReadPartially) { |
TEST_F(UploadBytesElementReaderTest, ReadAll) { |
std::vector<char> buf(bytes_.size()); |
+ scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]); |
EXPECT_EQ(static_cast<int>(buf.size()), |
- reader_->ReadSync(&buf[0], buf.size())); |
+ reader_->ReadSync(wrapped_buffer, buf.size())); |
EXPECT_EQ(0U, reader_->BytesRemaining()); |
EXPECT_EQ(bytes_, buf); |
// Try to read again. |
- EXPECT_EQ(0, reader_->ReadSync(&buf[0], buf.size())); |
+ EXPECT_EQ(0, reader_->ReadSync(wrapped_buffer, buf.size())); |
} |
TEST_F(UploadBytesElementReaderTest, ReadTooMuch) { |
const size_t kTooLargeSize = bytes_.size() * 2; |
std::vector<char> buf(kTooLargeSize); |
+ scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]); |
EXPECT_EQ(static_cast<int>(bytes_.size()), |
- reader_->ReadSync(&buf[0], buf.size())); |
+ reader_->ReadSync(wrapped_buffer, buf.size())); |
EXPECT_EQ(0U, reader_->BytesRemaining()); |
buf.resize(bytes_.size()); // Resize to compare. |
EXPECT_EQ(bytes_, buf); |