blob: 95d7bf5fa7dc8927c6828abd7c697031bee66108 [file] [log] [blame]
// Copyright (c) 2012 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "android_webview/native/input_stream_impl.h"
#include "base/android/jni_android.h"
#include "base/android/scoped_java_ref.h"
#include "base/memory/scoped_ptr.h"
#include "jni/InputStreamUnittest_jni.h"
#include "net/base/io_buffer.h"
#include "net/base/net_errors.h"
#include "net/http/http_byte_range.h"
#include "testing/gmock/include/gmock/gmock.h"
#include "testing/gtest/include/gtest/gtest.h"
using android_webview::InputStream;
using android_webview::InputStreamImpl;
using base::android::AttachCurrentThread;
using base::android::ScopedJavaLocalRef;
using net::IOBuffer;
using testing::DoAll;
using testing::Ge;
using testing::InSequence;
using testing::Lt;
using testing::Ne;
using testing::NotNull;
using testing::Return;
using testing::SetArgPointee;
using testing::Test;
using testing::_;
class InputStreamTest : public Test {
public:
InputStreamTest() {
}
protected:
void SetUp() override {
env_ = AttachCurrentThread();
ASSERT_THAT(env_, NotNull());
ASSERT_TRUE(android_webview::RegisterInputStream(env_));
ASSERT_TRUE(RegisterNativesImpl(env_));
}
scoped_refptr<IOBuffer> DoReadCountedStreamTest(int stream_size,
int bytes_requested,
int* bytes_read) {
ScopedJavaLocalRef<jobject> counting_jstream =
Java_InputStreamUnittest_getCountingStream(env_, stream_size);
EXPECT_FALSE(counting_jstream.is_null());
scoped_ptr<InputStream> input_stream(
new InputStreamImpl(counting_jstream));
scoped_refptr<IOBuffer> buffer = new IOBuffer(bytes_requested);
EXPECT_TRUE(input_stream->Read(buffer.get(), bytes_requested, bytes_read));
return buffer;
}
JNIEnv* env_;
};
TEST_F(InputStreamTest, ReadEmptyStream) {
ScopedJavaLocalRef<jobject> empty_jstream =
Java_InputStreamUnittest_getEmptyStream(env_);
EXPECT_FALSE(empty_jstream.is_null());
scoped_ptr<InputStream> input_stream(new InputStreamImpl(empty_jstream));
const int bytes_requested = 10;
int bytes_read = 0;
scoped_refptr<IOBuffer> buffer = new IOBuffer(bytes_requested);
EXPECT_TRUE(input_stream->Read(buffer.get(), bytes_requested, &bytes_read));
EXPECT_EQ(0, bytes_read);
}
TEST_F(InputStreamTest, ReadStreamPartial) {
const int bytes_requested = 128;
int bytes_read = 0;
DoReadCountedStreamTest(bytes_requested * 2, bytes_requested, &bytes_read);
EXPECT_EQ(bytes_requested, bytes_read);
}
TEST_F(InputStreamTest, ReadStreamCompletely) {
const int bytes_requested = 42;
int bytes_read = 0;
DoReadCountedStreamTest(bytes_requested, bytes_requested, &bytes_read);
EXPECT_EQ(bytes_requested, bytes_read);
}
TEST_F(InputStreamTest, TryReadMoreThanBuffer) {
const int buffer_size = 3 * InputStreamImpl::kBufferSize;
int bytes_read = 0;
DoReadCountedStreamTest(buffer_size, buffer_size * 2, &bytes_read);
EXPECT_EQ(buffer_size, bytes_read);
}
TEST_F(InputStreamTest, CheckContentsReadCorrectly) {
const int bytes_requested = 256;
int bytes_read = 0;
scoped_refptr<IOBuffer> buffer =
DoReadCountedStreamTest(bytes_requested, bytes_requested, &bytes_read);
EXPECT_EQ(bytes_requested, bytes_read);
for (int i = 0; i < bytes_requested; ++i) {
EXPECT_EQ(i, (unsigned char)buffer->data()[i]);
}
}
TEST_F(InputStreamTest, ReadLargeStreamPartial) {
const int bytes_requested = 3 * InputStreamImpl::kBufferSize;
int bytes_read = 0;
DoReadCountedStreamTest(bytes_requested + 32, bytes_requested, &bytes_read);
EXPECT_EQ(bytes_requested, bytes_read);
}
TEST_F(InputStreamTest, ReadLargeStreamCompletely) {
const int bytes_requested = 3 * InputStreamImpl::kBufferSize;
int bytes_read = 0;
DoReadCountedStreamTest(bytes_requested, bytes_requested, &bytes_read);
EXPECT_EQ(bytes_requested, bytes_read);
}
TEST_F(InputStreamTest, DoesNotCrashWhenExceptionThrown) {
ScopedJavaLocalRef<jobject> throw_jstream =
Java_InputStreamUnittest_getThrowingStream(env_);
EXPECT_FALSE(throw_jstream.is_null());
scoped_ptr<InputStream> input_stream(new InputStreamImpl(throw_jstream));
int64_t bytes_skipped;
EXPECT_FALSE(input_stream->Skip(10, &bytes_skipped));
int bytes_available;
EXPECT_FALSE(input_stream->BytesAvailable(&bytes_available));
const int bytes_requested = 10;
int bytes_read = 0;
scoped_refptr<IOBuffer> buffer = new IOBuffer(bytes_requested);
EXPECT_FALSE(input_stream->Read(buffer.get(), bytes_requested, &bytes_read));
EXPECT_EQ(0, bytes_read);
// This closes the stream.
input_stream.reset(NULL);
}