android_webview/native/input_stream_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 "android_webview/native/input_stream_impl.h"
   6 #include "base/android/jni_android.h"
   7 #include "base/android/scoped_java_ref.h"
   8 #include "base/memory/scoped_ptr.h"
   9 #include "jni/InputStreamUnittest_jni.h"
  10 #include "net/base/io_buffer.h"
  11 #include "net/base/net_errors.h"
  12 #include "net/http/http_byte_range.h"
  13
  14 #include "testing/gmock/include/gmock/gmock.h"
  15 #include "testing/gtest/include/gtest/gtest.h"
  16
  17 using android_webview::InputStream;
  18 using android_webview::InputStreamImpl;
  19 using base::android::AttachCurrentThread;
  20 using base::android::ScopedJavaLocalRef;
  21 using net::IOBuffer;
  22 using testing::DoAll;
  23 using testing::Ge;
  24 using testing::InSequence;
  25 using testing::Lt;
  26 using testing::Ne;
  27 using testing::NotNull;
  28 using testing::Return;
  29 using testing::SetArgPointee;
  30 using testing::Test;
  31 using testing::_;
  32
  33 class InputStreamTest : public Test {
  34  public:
  35   InputStreamTest() {
  36   }
  37  protected:
  38   virtual void SetUp() {
  39     env_ = AttachCurrentThread();
  40     ASSERT_THAT(env_, NotNull());
  41     ASSERT_TRUE(android_webview::RegisterInputStream(env_));
  42     ASSERT_TRUE(RegisterNativesImpl(env_));
  43   }
  44
  45   scoped_refptr<IOBuffer> DoReadCountedStreamTest(int stream_size,
  46                                                   int bytes_requested,
  47                                                   int* bytes_read) {
  48     ScopedJavaLocalRef<jobject> counting_jstream =
  49         Java_InputStreamUnittest_getCountingStream(env_, stream_size);
  50     EXPECT_FALSE(counting_jstream.is_null());
  51
  52     scoped_ptr<InputStream> input_stream(
  53         new InputStreamImpl(counting_jstream));
  54     scoped_refptr<IOBuffer> buffer = new IOBuffer(bytes_requested);
  55
  56     EXPECT_TRUE(input_stream->Read(buffer.get(), bytes_requested, bytes_read));
  57     return buffer;
  58   }
  59
  60   JNIEnv* env_;
  61 };
  62
  63 TEST_F(InputStreamTest, ReadEmptyStream) {
  64   ScopedJavaLocalRef<jobject> empty_jstream =
  65       Java_InputStreamUnittest_getEmptyStream(env_);
  66   EXPECT_FALSE(empty_jstream.is_null());
  67
  68   scoped_ptr<InputStream> input_stream(new InputStreamImpl(empty_jstream));
  69   const int bytes_requested = 10;
  70   int bytes_read = 0;
  71   scoped_refptr<IOBuffer> buffer = new IOBuffer(bytes_requested);
  72
  73   EXPECT_TRUE(input_stream->Read(buffer.get(), bytes_requested, &bytes_read));
  74   EXPECT_EQ(0, bytes_read);
  75 }
  76
  77 TEST_F(InputStreamTest, ReadStreamPartial) {
  78   const int bytes_requested = 128;
  79   int bytes_read = 0;
  80   DoReadCountedStreamTest(bytes_requested * 2, bytes_requested, &bytes_read);
  81   EXPECT_EQ(bytes_requested, bytes_read);
  82 }
  83
  84 TEST_F(InputStreamTest, ReadStreamCompletely) {
  85   const int bytes_requested = 42;
  86   int bytes_read = 0;
  87   DoReadCountedStreamTest(bytes_requested, bytes_requested, &bytes_read);
  88   EXPECT_EQ(bytes_requested, bytes_read);
  89 }
  90
  91 TEST_F(InputStreamTest, TryReadMoreThanBuffer) {
  92   const int buffer_size = 3 * InputStreamImpl::kBufferSize;
  93   int bytes_read = 0;
  94   DoReadCountedStreamTest(buffer_size, buffer_size * 2, &bytes_read);
  95   EXPECT_EQ(buffer_size, bytes_read);
  96 }
  97
  98 TEST_F(InputStreamTest, CheckContentsReadCorrectly) {
  99   const int bytes_requested = 256;
 100   int bytes_read = 0;
 101   scoped_refptr<IOBuffer> buffer =
 102       DoReadCountedStreamTest(bytes_requested, bytes_requested, &bytes_read);
 103   EXPECT_EQ(bytes_requested, bytes_read);
 104   for (int i = 0; i < bytes_requested; ++i) {
 105     EXPECT_EQ(i, (unsigned char)buffer->data()[i]);
 106   }
 107 }
 108
 109 TEST_F(InputStreamTest, ReadLargeStreamPartial) {
 110   const int bytes_requested = 3 * InputStreamImpl::kBufferSize;
 111   int bytes_read = 0;
 112   DoReadCountedStreamTest(bytes_requested + 32, bytes_requested, &bytes_read);
 113   EXPECT_EQ(bytes_requested, bytes_read);
 114 }
 115
 116 TEST_F(InputStreamTest, ReadLargeStreamCompletely) {
 117   const int bytes_requested = 3 * InputStreamImpl::kBufferSize;
 118   int bytes_read = 0;
 119   DoReadCountedStreamTest(bytes_requested, bytes_requested, &bytes_read);
 120   EXPECT_EQ(bytes_requested, bytes_read);
 121 }
 122
 123 TEST_F(InputStreamTest, DoesNotCrashWhenExceptionThrown) {
 124   ScopedJavaLocalRef<jobject> throw_jstream =
 125       Java_InputStreamUnittest_getThrowingStream(env_);
 126   EXPECT_FALSE(throw_jstream.is_null());
 127
 128   scoped_ptr<InputStream> input_stream(new InputStreamImpl(throw_jstream));
 129
 130   int64_t bytes_skipped;
 131   EXPECT_FALSE(input_stream->Skip(10, &bytes_skipped));
 132
 133   int bytes_available;
 134   EXPECT_FALSE(input_stream->BytesAvailable(&bytes_available));
 135
 136
 137   const int bytes_requested = 10;
 138   int bytes_read = 0;
 139   scoped_refptr<IOBuffer> buffer = new IOBuffer(bytes_requested);
 140   EXPECT_FALSE(input_stream->Read(buffer.get(), bytes_requested, &bytes_read));
 141   EXPECT_EQ(0, bytes_read);
 142
 143   // This closes the stream.
 144   input_stream.reset(NULL);
 145 }