net/base/upload_bytes_element_reader_unittest.cc

   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
   3 // found in the LICENSE file.
   4
   5 #include "net/base/upload_bytes_element_reader.h"
   6
   7 #include "base/basictypes.h"
   8 #include "base/memory/scoped_ptr.h"
   9 #include "net/base/io_buffer.h"
  10 #include "net/base/net_errors.h"
  11 #include "testing/gtest/include/gtest/gtest.h"
  12 #include "testing/platform_test.h"
  13
  14 namespace net {
  15
  16 class UploadBytesElementReaderTest : public PlatformTest {
  17  protected:
  18   virtual void SetUp() OVERRIDE {
  19     const char kData[] = "123abc";
  20     bytes_.assign(kData, kData + arraysize(kData));
  21     reader_.reset(new UploadBytesElementReader(&bytes_[0], bytes_.size()));
  22     ASSERT_EQ(OK, reader_->Init(CompletionCallback()));
  23     EXPECT_EQ(bytes_.size(), reader_->GetContentLength());
  24     EXPECT_EQ(bytes_.size(), reader_->BytesRemaining());
  25     EXPECT_TRUE(reader_->IsInMemory());
  26   }
  27
  28   std::vector<char> bytes_;
  29   scoped_ptr<UploadElementReader> reader_;
  30 };
  31
  32 TEST_F(UploadBytesElementReaderTest, ReadPartially) {
  33   const size_t kHalfSize = bytes_.size() / 2;
  34   std::vector<char> buf(kHalfSize);
  35   scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]);
  36   EXPECT_EQ(static_cast<int>(buf.size()),
  37             reader_->Read(wrapped_buffer, buf.size(), CompletionCallback()));
  38   EXPECT_EQ(bytes_.size() - buf.size(), reader_->BytesRemaining());
  39   bytes_.resize(kHalfSize);  // Resize to compare.
  40   EXPECT_EQ(bytes_, buf);
  41 }
  42
  43 TEST_F(UploadBytesElementReaderTest, ReadAll) {
  44   std::vector<char> buf(bytes_.size());
  45   scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]);
  46   EXPECT_EQ(static_cast<int>(buf.size()),
  47             reader_->Read(wrapped_buffer, buf.size(), CompletionCallback()));
  48   EXPECT_EQ(0U, reader_->BytesRemaining());
  49   EXPECT_EQ(bytes_, buf);
  50   // Try to read again.
  51   EXPECT_EQ(0, reader_->Read(wrapped_buffer, buf.size(), CompletionCallback()));
  52 }
  53
  54 TEST_F(UploadBytesElementReaderTest, ReadTooMuch) {
  55   const size_t kTooLargeSize = bytes_.size() * 2;
  56   std::vector<char> buf(kTooLargeSize);
  57   scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]);
  58   EXPECT_EQ(static_cast<int>(bytes_.size()),
  59             reader_->Read(wrapped_buffer, buf.size(), CompletionCallback()));
  60   EXPECT_EQ(0U, reader_->BytesRemaining());
  61   buf.resize(bytes_.size());  // Resize to compare.
  62   EXPECT_EQ(bytes_, buf);
  63 }
  64
  65 TEST_F(UploadBytesElementReaderTest, MultipleInit) {
  66   std::vector<char> buf(bytes_.size());
  67   scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]);
  68
  69   // Read all.
  70   EXPECT_EQ(static_cast<int>(buf.size()),
  71             reader_->Read(wrapped_buffer, buf.size(), CompletionCallback()));
  72   EXPECT_EQ(0U, reader_->BytesRemaining());
  73   EXPECT_EQ(bytes_, buf);
  74
  75   // Call Init() again to reset the state.
  76   ASSERT_EQ(OK, reader_->Init(CompletionCallback()));
  77   EXPECT_EQ(bytes_.size(), reader_->GetContentLength());
  78   EXPECT_EQ(bytes_.size(), reader_->BytesRemaining());
  79
  80   // Read again.
  81   EXPECT_EQ(static_cast<int>(buf.size()),
  82             reader_->Read(wrapped_buffer, buf.size(), CompletionCallback()));
  83   EXPECT_EQ(0U, reader_->BytesRemaining());
  84   EXPECT_EQ(bytes_, buf);
  85 }
  86
  87 }  // namespace net