1 files changed, 205 insertions, 0 deletions
diff --git a/net/base/file_input_stream_unittest.cc b/net/base/file_input_stream_unittest.cc
new file mode 100644
index 0000000..b259de8
--- /dev/null
+++ b/net/base/file_input_stream_unittest.cc
@@ -0,0 +1,205 @@
+// Copyright (c) 2008 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 "base/file_util.h"
+#include "base/path_service.h"
+#include "base/platform_test.h"
+#include "net/base/file_input_stream.h"
+#include "net/base/net_errors.h"
+#include "net/base/test_completion_callback.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+static const char kTestData[] = "0123456789";
+
+class FileInputStreamTest : public PlatformTest {
+ public:
+  virtual void SetUp() {
+    PlatformTest::SetUp();
+
+    file_util::CreateTemporaryFileName(&temp_file_path_);
+    file_util::WriteFile(temp_file_path_, kTestData, arraysize(kTestData)-1);
+  }
+  virtual void TearDown() {
+    file_util::Delete(temp_file_path_, false);
+
+    PlatformTest::TearDown();
+  }
+  const std::wstring temp_file_path() const { return temp_file_path_; }
+ private:
+  std::wstring temp_file_path_;
+};
+
+TEST_F(FileInputStreamTest, BasicOpenClose) {
+  net::FileInputStream stream;
+  int rv = stream.Open(temp_file_path(), false);
+  EXPECT_EQ(net::OK, rv);
+}
+
+TEST_F(FileInputStreamTest, UseClosedStream) {
+  net::FileInputStream stream;
+
+  EXPECT_FALSE(stream.IsOpen());
+
+  // Try seeking...
+  int64 new_offset = stream.Seek(net::FROM_BEGIN, 5);
+  EXPECT_EQ(net::ERR_UNEXPECTED, new_offset);
+
+  // Try available...
+  int64 avail = stream.Available();
+  EXPECT_EQ(net::ERR_UNEXPECTED, avail);
+
+  // Try reading...
+  char buf[10];
+  int rv = stream.Read(buf, sizeof(buf), NULL);
+  EXPECT_EQ(net::ERR_UNEXPECTED, rv);
+}
+
+TEST_F(FileInputStreamTest, BasicRead) {
+  int64 file_size;
+  bool ok = file_util::GetFileSize(temp_file_path(), &file_size);
+  EXPECT_TRUE(ok);
+
+  net::FileInputStream stream;
+  int rv = stream.Open(temp_file_path(), false);
+  EXPECT_EQ(net::OK, rv);
+
+  int64 total_bytes_avail = stream.Available();
+  EXPECT_EQ(file_size, total_bytes_avail);
+
+  int64 total_bytes_read = 0;
+
+  std::string data_read;
+  for (;;) {
+    char buf[4];
+    rv = stream.Read(buf, sizeof(buf), NULL);
+    EXPECT_LE(0, rv);
+    if (rv <= 0)
+      break;
+    total_bytes_read += rv;
+    data_read.append(buf, rv);
+  }
+  EXPECT_EQ(file_size, total_bytes_read);
+  EXPECT_TRUE(data_read == kTestData);
+}
+
+TEST_F(FileInputStreamTest, AsyncRead) {
+  int64 file_size;
+  bool ok = file_util::GetFileSize(temp_file_path(), &file_size);
+  EXPECT_TRUE(ok);
+
+  net::FileInputStream stream;
+  int rv = stream.Open(temp_file_path(), true);
+  EXPECT_EQ(net::OK, rv);
+
+  int64 total_bytes_avail = stream.Available();
+  EXPECT_EQ(file_size, total_bytes_avail);
+
+  TestCompletionCallback callback;
+
+  int64 total_bytes_read = 0;
+
+  std::string data_read;
+  for (;;) {
+    char buf[4];
+    rv = stream.Read(buf, sizeof(buf), &callback);
+    if (rv == net::ERR_IO_PENDING)
+      rv = callback.WaitForResult();
+    EXPECT_LE(0, rv);
+    if (rv <= 0)
+      break;
+    total_bytes_read += rv;
+    data_read.append(buf, rv);
+  }
+  EXPECT_EQ(file_size, total_bytes_read);
+  EXPECT_TRUE(data_read == kTestData);
+}
+
+TEST_F(FileInputStreamTest, BasicRead_FromOffset) {
+  int64 file_size;
+  bool ok = file_util::GetFileSize(temp_file_path(), &file_size);
+  EXPECT_TRUE(ok);
+
+  net::FileInputStream stream;
+  int rv = stream.Open(temp_file_path(), false);
+  EXPECT_EQ(net::OK, rv);
+
+  const int64 kOffset = 3;
+  int64 new_offset = stream.Seek(net::FROM_BEGIN, kOffset);
+  EXPECT_EQ(kOffset, new_offset);
+
+  int64 total_bytes_avail = stream.Available();
+  EXPECT_EQ(file_size - kOffset, total_bytes_avail);
+
+  int64 total_bytes_read = 0;
+
+  std::string data_read;
+  for (;;) {
+    char buf[4];
+    rv = stream.Read(buf, sizeof(buf), NULL);
+    EXPECT_LE(0, rv);
+    if (rv <= 0)
+      break;
+    total_bytes_read += rv;
+    data_read.append(buf, rv);
+  }
+  EXPECT_EQ(file_size - kOffset, total_bytes_read);
+  EXPECT_TRUE(data_read == kTestData + kOffset);
+}
+
+TEST_F(FileInputStreamTest, AsyncRead_FromOffset) {
+  int64 file_size;
+  bool ok = file_util::GetFileSize(temp_file_path(), &file_size);
+  EXPECT_TRUE(ok);
+
+  net::FileInputStream stream;
+  int rv = stream.Open(temp_file_path(), true);
+  EXPECT_EQ(net::OK, rv);
+
+  const int64 kOffset = 3;
+  int64 new_offset = stream.Seek(net::FROM_BEGIN, kOffset);
+  EXPECT_EQ(kOffset, new_offset);
+
+  int64 total_bytes_avail = stream.Available();
+  EXPECT_EQ(file_size - kOffset, total_bytes_avail);
+
+  TestCompletionCallback callback;
+
+  int64 total_bytes_read = 0;
+
+  std::string data_read;
+  for (;;) {
+    char buf[4];
+    rv = stream.Read(buf, sizeof(buf), &callback);
+    if (rv == net::ERR_IO_PENDING)
+      rv = callback.WaitForResult();
+    EXPECT_LE(0, rv);
+    if (rv <= 0)
+      break;
+    total_bytes_read += rv;
+    data_read.append(buf, rv);
+  }
+  EXPECT_EQ(file_size - kOffset, total_bytes_read);
+  EXPECT_TRUE(data_read == kTestData + kOffset);
+}
+
+TEST_F(FileInputStreamTest, SeekAround) {
+  net::FileInputStream stream;
+  int rv = stream.Open(temp_file_path(), true);
+  EXPECT_EQ(net::OK, rv);
+
+  const int64 kOffset = 3;
+  int64 new_offset = stream.Seek(net::FROM_BEGIN, kOffset);
+  EXPECT_EQ(kOffset, new_offset);
+
+  new_offset = stream.Seek(net::FROM_CURRENT, kOffset);
+  EXPECT_EQ(2 * kOffset, new_offset);
+
+  new_offset = stream.Seek(net::FROM_CURRENT, -kOffset);
+  EXPECT_EQ(kOffset, new_offset);
+
+  const int kTestDataLen = arraysize(kTestData) - 1;
+
+  new_offset = stream.Seek(net::FROM_END, -kTestDataLen);
+  EXPECT_EQ(0, new_offset);
+}