Index: chrome/browser/chromeos/file_system_provider/fileapi/file_stream_reader_unittest.cc |
diff --git a/chrome/browser/chromeos/file_system_provider/fileapi/file_stream_reader_unittest.cc b/chrome/browser/chromeos/file_system_provider/fileapi/file_stream_reader_unittest.cc |
new file mode 100644 |
index 0000000000000000000000000000000000000000..dfd7e671830d619eb11e3397707800234c6dfc3e |
--- /dev/null |
+++ b/chrome/browser/chromeos/file_system_provider/fileapi/file_stream_reader_unittest.cc |
@@ -0,0 +1,294 @@ |
+// Copyright 2014 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 <string> |
+#include <vector> |
+ |
+#include "base/files/file.h" |
+#include "base/files/file_path.h" |
+#include "base/files/scoped_temp_dir.h" |
+#include "base/memory/scoped_ptr.h" |
+#include "base/memory/weak_ptr.h" |
+#include "base/platform_file.h" |
+#include "base/run_loop.h" |
+#include "chrome/browser/chromeos/file_system_provider/fake_provided_file_system.h" |
+#include "chrome/browser/chromeos/file_system_provider/fileapi/file_stream_reader.h" |
+#include "chrome/browser/chromeos/file_system_provider/service.h" |
+#include "chrome/browser/chromeos/file_system_provider/service_factory.h" |
+#include "chrome/test/base/testing_browser_process.h" |
+#include "chrome/test/base/testing_profile.h" |
+#include "chrome/test/base/testing_profile_manager.h" |
+#include "content/public/test/test_browser_thread_bundle.h" |
+#include "content/public/test/test_file_system_context.h" |
+#include "extensions/browser/extension_registry.h" |
+#include "net/base/io_buffer.h" |
+#include "net/base/net_errors.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+#include "webkit/browser/fileapi/async_file_util.h" |
+#include "webkit/browser/fileapi/external_mount_points.h" |
+#include "webkit/browser/fileapi/file_system_url.h" |
+ |
+namespace chromeos { |
+namespace file_system_provider { |
+namespace { |
+ |
+const char kExtensionId[] = "mbflcebpggnecokmikipoihdbecnjfoj"; |
+ |
+// Logs callbacks invocations on the file stream reader. |
+class EventLogger { |
+ public: |
+ EventLogger() : weak_ptr_factory_(this) {} |
+ virtual ~EventLogger() {} |
+ |
+ void OnRead(int result) { results_.push_back(result); } |
+ void OnGetLength(int64 result) { results_.push_back(result); } |
+ |
+ base::WeakPtr<EventLogger> GetWeakPtr() { |
+ return weak_ptr_factory_.GetWeakPtr(); |
+ } |
+ |
+ const std::vector<int64>& results() const { return results_; } |
+ |
+ private: |
+ std::vector<int64> results_; |
+ base::WeakPtrFactory<EventLogger> weak_ptr_factory_; |
+ |
+ DISALLOW_COPY_AND_ASSIGN(EventLogger); |
+}; |
+ |
+// Creates a cracked FileSystemURL for tests. |
+fileapi::FileSystemURL CreateFileSystemURL(const std::string& mount_point_name, |
+ const base::FilePath& file_path) { |
+ const std::string origin = std::string("chrome-extension://") + kExtensionId; |
+ const fileapi::ExternalMountPoints* const mount_points = |
+ fileapi::ExternalMountPoints::GetSystemInstance(); |
+ return mount_points->CreateCrackedFileSystemURL( |
+ GURL(origin), |
+ fileapi::kFileSystemTypeExternal, |
+ base::FilePath::FromUTF8Unsafe(mount_point_name).Append(file_path)); |
+} |
+ |
+// Creates a Service instance. Used to be able to destroy the service in |
+// TearDown(). |
+KeyedService* CreateService(content::BrowserContext* context) { |
+ return new Service(Profile::FromBrowserContext(context), |
+ extensions::ExtensionRegistry::Get(context)); |
+} |
+ |
+} // namespace |
+ |
+class FileSystemProviderFileStreamReader : public testing::Test { |
+ protected: |
+ FileSystemProviderFileStreamReader() {} |
+ virtual ~FileSystemProviderFileStreamReader() {} |
+ |
+ virtual void SetUp() OVERRIDE { |
+ ASSERT_TRUE(data_dir_.CreateUniqueTempDir()); |
+ profile_manager_.reset( |
+ new TestingProfileManager(TestingBrowserProcess::GetGlobal())); |
+ ASSERT_TRUE(profile_manager_->SetUp()); |
+ profile_ = profile_manager_->CreateTestingProfile("testing-profile"); |
+ |
+ ServiceFactory::GetInstance()->SetTestingFactory(profile_, &CreateService); |
+ Service* service = Service::Get(profile_); // Owned by its factory. |
+ service->SetFileSystemFactoryForTests( |
+ base::Bind(&FakeProvidedFileSystem::Create)); |
+ |
+ const int file_system_id = |
+ service->MountFileSystem(kExtensionId, "testing-file-system"); |
+ ASSERT_LT(0, file_system_id); |
+ const ProvidedFileSystemInfo& file_system_info = |
+ service->GetProvidedFileSystem(kExtensionId, file_system_id) |
+ ->GetFileSystemInfo(); |
+ const std::string mount_point_name = |
+ file_system_info.mount_path().BaseName().AsUTF8Unsafe(); |
+ |
+ file_url_ = CreateFileSystemURL( |
+ mount_point_name, base::FilePath::FromUTF8Unsafe(kFakeFilePath + 1)); |
+ ASSERT_TRUE(file_url_.is_valid()); |
+ wrong_file_url_ = CreateFileSystemURL( |
+ mount_point_name, base::FilePath::FromUTF8Unsafe("im-not-here.txt")); |
+ ASSERT_TRUE(wrong_file_url_.is_valid()); |
+ } |
+ |
+ virtual void TearDown() OVERRIDE { |
+ // Setting the testing factory to NULL will destroy the created service |
+ // associated with the testing profile. |
+ ServiceFactory::GetInstance()->SetTestingFactory(profile_, NULL); |
+ } |
+ |
+ content::TestBrowserThreadBundle thread_bundle_; |
+ base::ScopedTempDir data_dir_; |
+ scoped_ptr<TestingProfileManager> profile_manager_; |
+ TestingProfile* profile_; // Owned by TestingProfileManager. |
+ fileapi::FileSystemURL file_url_; |
+ fileapi::FileSystemURL wrong_file_url_; |
+}; |
+ |
+TEST_F(FileSystemProviderFileStreamReader, Read_AllAtOnce) { |
+ EventLogger logger; |
+ |
+ const int64 initial_offset = 0; |
+ FileStreamReader reader(NULL, |
+ file_url_, |
+ initial_offset, |
+ base::Time::Now()); // Not used yet. |
+ scoped_refptr<net::IOBuffer> io_buffer(new net::IOBuffer(kFakeFileSize)); |
+ |
+ const int result = |
+ reader.Read(io_buffer.get(), |
+ kFakeFileSize, |
+ base::Bind(&EventLogger::OnRead, logger.GetWeakPtr())); |
+ EXPECT_EQ(net::ERR_IO_PENDING, result); |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ ASSERT_EQ(1u, logger.results().size()); |
+ EXPECT_LT(0, logger.results()[0]); |
+ EXPECT_EQ(kFakeFileSize, static_cast<size_t>(logger.results()[0])); |
+ |
+ std::string buffer_as_string(io_buffer->data(), kFakeFileSize); |
+ EXPECT_EQ(kFakeFileText, buffer_as_string); |
+} |
+ |
+TEST_F(FileSystemProviderFileStreamReader, Read_WrongFile) { |
+ EventLogger logger; |
+ |
+ const int64 initial_offset = 0; |
+ FileStreamReader reader(NULL, |
+ wrong_file_url_, |
+ initial_offset, |
+ base::Time::Now()); // Not used yet. |
+ scoped_refptr<net::IOBuffer> io_buffer(new net::IOBuffer(kFakeFileSize)); |
+ |
+ const int result = |
+ reader.Read(io_buffer.get(), |
+ kFakeFileSize, |
+ base::Bind(&EventLogger::OnRead, logger.GetWeakPtr())); |
+ EXPECT_EQ(net::ERR_IO_PENDING, result); |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ ASSERT_EQ(1u, logger.results().size()); |
+ EXPECT_EQ(net::ERR_FAILED, logger.results()[0]); |
+} |
+ |
+TEST_F(FileSystemProviderFileStreamReader, Read_InChunks) { |
+ EventLogger logger; |
+ |
+ const int64 initial_offset = 0; |
+ FileStreamReader reader(NULL, |
+ file_url_, |
+ initial_offset, |
+ base::Time::Now()); // Not used yet. |
+ |
+ for (size_t offset = 0; offset < kFakeFileSize; ++offset) { |
+ scoped_refptr<net::IOBuffer> io_buffer(new net::IOBuffer(1)); |
+ const int result = |
+ reader.Read(io_buffer.get(), |
+ 1, |
+ base::Bind(&EventLogger::OnRead, logger.GetWeakPtr())); |
+ EXPECT_EQ(net::ERR_IO_PENDING, result); |
+ base::RunLoop().RunUntilIdle(); |
+ ASSERT_EQ(offset + 1u, logger.results().size()); |
+ EXPECT_EQ(1, logger.results()[offset]); |
+ EXPECT_EQ(kFakeFileText[offset], io_buffer->data()[0]); |
+ } |
+} |
+ |
+TEST_F(FileSystemProviderFileStreamReader, Read_Slice) { |
+ EventLogger logger; |
+ |
+ // Trim first 3 and last 3 characters. |
+ const int64 initial_offset = 3; |
+ const int length = static_cast<int>(kFakeFileSize) - initial_offset - 3; |
+ ASSERT_GT(kFakeFileSize, static_cast<size_t>(initial_offset)); |
+ ASSERT_LT(0, length); |
+ |
+ FileStreamReader reader(NULL, |
+ file_url_, |
+ initial_offset, |
+ base::Time::Now()); // Not used yet. |
+ scoped_refptr<net::IOBuffer> io_buffer(new net::IOBuffer(length)); |
+ |
+ const int result = |
+ reader.Read(io_buffer.get(), |
+ length, |
+ base::Bind(&EventLogger::OnRead, logger.GetWeakPtr())); |
+ EXPECT_EQ(net::ERR_IO_PENDING, result); |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ ASSERT_EQ(1u, logger.results().size()); |
+ EXPECT_EQ(length, logger.results()[0]); |
+ |
+ std::string buffer_as_string(io_buffer->data(), length); |
+ std::string expected_buffer(kFakeFileText + initial_offset, length); |
+ EXPECT_EQ(expected_buffer, buffer_as_string); |
+} |
+ |
+TEST_F(FileSystemProviderFileStreamReader, Read_Beyond) { |
+ EventLogger logger; |
+ |
+ // Request reading 1KB more than available. |
+ const int64 initial_offset = 0; |
+ const int length = static_cast<int>(kFakeFileSize) + 1024; |
+ |
+ FileStreamReader reader(NULL, |
+ file_url_, |
+ initial_offset, |
+ base::Time::Now()); // Not used yet. |
+ scoped_refptr<net::IOBuffer> io_buffer(new net::IOBuffer(length)); |
+ |
+ const int result = |
+ reader.Read(io_buffer.get(), |
+ length, |
+ base::Bind(&EventLogger::OnRead, logger.GetWeakPtr())); |
+ EXPECT_EQ(net::ERR_IO_PENDING, result); |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ ASSERT_EQ(1u, logger.results().size()); |
+ EXPECT_LT(0, logger.results()[0]); |
+ EXPECT_EQ(kFakeFileSize, static_cast<size_t>(logger.results()[0])); |
+ |
+ std::string buffer_as_string(io_buffer->data(), kFakeFileSize); |
+ EXPECT_EQ(kFakeFileText, buffer_as_string); |
+} |
+ |
+TEST_F(FileSystemProviderFileStreamReader, GetLength) { |
+ EventLogger logger; |
+ |
+ const int64 initial_offset = 0; |
+ FileStreamReader reader(NULL, |
+ file_url_, |
+ initial_offset, |
+ base::Time::Now()); // Not used yet. |
+ |
+ const int result = reader.GetLength( |
+ base::Bind(&EventLogger::OnGetLength, logger.GetWeakPtr())); |
+ EXPECT_EQ(net::ERR_IO_PENDING, result); |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ ASSERT_EQ(1u, logger.results().size()); |
+ EXPECT_LT(0, logger.results()[0]); |
+ EXPECT_EQ(kFakeFileSize, static_cast<size_t>(logger.results()[0])); |
+} |
+ |
+TEST_F(FileSystemProviderFileStreamReader, GetLength_WrongFile) { |
+ EventLogger logger; |
+ |
+ const int64 initial_offset = 0; |
+ FileStreamReader reader(NULL, |
+ wrong_file_url_, |
+ initial_offset, |
+ base::Time::Now()); // Not used yet. |
+ |
+ const int result = reader.GetLength( |
+ base::Bind(&EventLogger::OnGetLength, logger.GetWeakPtr())); |
+ EXPECT_EQ(net::ERR_IO_PENDING, result); |
+ base::RunLoop().RunUntilIdle(); |
+ |
+ ASSERT_EQ(1u, logger.results().size()); |
+ EXPECT_EQ(net::ERR_FAILED, logger.results()[0]); |
+} |
+ |
+} // namespace file_system_provider |
+} // namespace chromeos |