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   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(
  37       static_cast<int>(buf.size()),
  38       reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback()));
  39   EXPECT_EQ(bytes_.size() - buf.size(), reader_->BytesRemaining());
  40   bytes_.resize(kHalfSize);  // Resize to compare.
  41   EXPECT_EQ(bytes_, buf);
  42 }
  43
  44 TEST_F(UploadBytesElementReaderTest, ReadAll) {
  45   std::vector<char> buf(bytes_.size());
  46   scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]);
  47   EXPECT_EQ(
  48       static_cast<int>(buf.size()),
  49       reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback()));
  50   EXPECT_EQ(0U, reader_->BytesRemaining());
  51   EXPECT_EQ(bytes_, buf);
  52   // Try to read again.
  53   EXPECT_EQ(
  54       0, reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback()));
  55 }
  56
  57 TEST_F(UploadBytesElementReaderTest, ReadTooMuch) {
  58   const size_t kTooLargeSize = bytes_.size() * 2;
  59   std::vector<char> buf(kTooLargeSize);
  60   scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]);
  61   EXPECT_EQ(
  62       static_cast<int>(bytes_.size()),
  63       reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback()));
  64   EXPECT_EQ(0U, reader_->BytesRemaining());
  65   buf.resize(bytes_.size());  // Resize to compare.
  66   EXPECT_EQ(bytes_, buf);
  67 }
  68
  69 TEST_F(UploadBytesElementReaderTest, MultipleInit) {
  70   std::vector<char> buf(bytes_.size());
  71   scoped_refptr<IOBuffer> wrapped_buffer = new WrappedIOBuffer(&buf[0]);
  72
  73   // Read all.
  74   EXPECT_EQ(
  75       static_cast<int>(buf.size()),
  76       reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback()));
  77   EXPECT_EQ(0U, reader_->BytesRemaining());
  78   EXPECT_EQ(bytes_, buf);
  79
  80   // Call Init() again to reset the state.
  81   ASSERT_EQ(OK, reader_->Init(CompletionCallback()));
  82   EXPECT_EQ(bytes_.size(), reader_->GetContentLength());
  83   EXPECT_EQ(bytes_.size(), reader_->BytesRemaining());
  84
  85   // Read again.
  86   EXPECT_EQ(
  87       static_cast<int>(buf.size()),
  88       reader_->Read(wrapped_buffer.get(), buf.size(), CompletionCallback()));
  89   EXPECT_EQ(0U, reader_->BytesRemaining());
  90   EXPECT_EQ(bytes_, buf);
  91 }
  92
  93 }  // namespace net