| // 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 "chrome/browser/chromeos/file_system_provider/fileapi/file_stream_reader.h" |
| |
| #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/run_loop.h" |
| #include "chrome/browser/chromeos/file_system_provider/fake_provided_file_system.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 "storage/browser/fileapi/async_file_util.h" |
| #include "storage/browser/fileapi/external_mount_points.h" |
| #include "storage/browser/fileapi/file_system_url.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| namespace chromeos { |
| namespace file_system_provider { |
| namespace { |
| |
| const char kExtensionId[] = "mbflcebpggnecokmikipoihdbecnjfoj"; |
| const char kFileSystemId[] = "testing-file-system"; |
| |
| // 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. |
| storage::FileSystemURL CreateFileSystemURL(const std::string& mount_point_name, |
| const base::FilePath& file_path) { |
| const std::string origin = std::string("chrome-extension://") + kExtensionId; |
| const storage::ExternalMountPoints* const mount_points = |
| storage::ExternalMountPoints::GetSystemInstance(); |
| return mount_points->CreateCrackedFileSystemURL( |
| GURL(origin), |
| storage::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() : profile_(NULL), fake_file_(NULL) {} |
| 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->SetFileSystemFactoryForTesting( |
| base::Bind(&FakeProvidedFileSystem::Create)); |
| |
| const bool result = service->MountFileSystem( |
| kExtensionId, MountOptions(kFileSystemId, "Testing File System")); |
| ASSERT_TRUE(result); |
| FakeProvidedFileSystem* provided_file_system = |
| static_cast<FakeProvidedFileSystem*>( |
| service->GetProvidedFileSystem(kExtensionId, kFileSystemId)); |
| ASSERT_TRUE(provided_file_system); |
| fake_file_ = provided_file_system->GetEntry( |
| base::FilePath::FromUTF8Unsafe(kFakeFilePath)); |
| ASSERT_TRUE(fake_file_); |
| const ProvidedFileSystemInfo& file_system_info = |
| service->GetProvidedFileSystem(kExtensionId, kFileSystemId) |
| ->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. |
| const FakeEntry* fake_file_; // Owned by FakePRovidedFileSystem. |
| storage::FileSystemURL file_url_; |
| storage::FileSystemURL wrong_file_url_; |
| }; |
| |
| TEST_F(FileSystemProviderFileStreamReader, Read_AllAtOnce) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader( |
| NULL, file_url_, initial_offset, fake_file_->metadata->modification_time); |
| scoped_refptr<net::IOBuffer> io_buffer( |
| new net::IOBuffer(fake_file_->metadata->size)); |
| |
| const int result = |
| reader.Read(io_buffer.get(), |
| fake_file_->metadata->size, |
| 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(fake_file_->metadata->size, logger.results()[0]); |
| |
| std::string buffer_as_string(io_buffer->data(), fake_file_->metadata->size); |
| EXPECT_EQ(fake_file_->contents, buffer_as_string); |
| } |
| |
| TEST_F(FileSystemProviderFileStreamReader, Read_WrongFile) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader(NULL, |
| wrong_file_url_, |
| initial_offset, |
| fake_file_->metadata->modification_time); |
| scoped_refptr<net::IOBuffer> io_buffer( |
| new net::IOBuffer(fake_file_->metadata->size)); |
| |
| const int result = |
| reader.Read(io_buffer.get(), |
| fake_file_->metadata->size, |
| 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_FILE_NOT_FOUND, logger.results()[0]); |
| } |
| |
| TEST_F(FileSystemProviderFileStreamReader, Read_InChunks) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader( |
| NULL, file_url_, initial_offset, fake_file_->metadata->modification_time); |
| |
| for (int64 offset = 0; offset < fake_file_->metadata->size; ++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 + 1, static_cast<int64>(logger.results().size())); |
| EXPECT_EQ(1, logger.results()[offset]); |
| EXPECT_EQ(fake_file_->contents[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 = fake_file_->metadata->size - initial_offset - 3; |
| ASSERT_GT(fake_file_->metadata->size, initial_offset); |
| ASSERT_LT(0, length); |
| |
| FileStreamReader reader( |
| NULL, file_url_, initial_offset, fake_file_->metadata->modification_time); |
| 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(fake_file_->contents.data() + 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 = fake_file_->metadata->size + 1024; |
| |
| FileStreamReader reader( |
| NULL, file_url_, initial_offset, fake_file_->metadata->modification_time); |
| 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(fake_file_->metadata->size, logger.results()[0]); |
| |
| std::string buffer_as_string(io_buffer->data(), fake_file_->metadata->size); |
| EXPECT_EQ(fake_file_->contents, buffer_as_string); |
| } |
| |
| TEST_F(FileSystemProviderFileStreamReader, Read_ModifiedFile) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader(NULL, file_url_, initial_offset, base::Time::Max()); |
| |
| scoped_refptr<net::IOBuffer> io_buffer( |
| new net::IOBuffer(fake_file_->metadata->size)); |
| const int result = |
| reader.Read(io_buffer.get(), |
| fake_file_->metadata->size, |
| 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_UPLOAD_FILE_CHANGED, logger.results()[0]); |
| } |
| |
| TEST_F(FileSystemProviderFileStreamReader, Read_ExpectedModificationTimeNull) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader(NULL, file_url_, initial_offset, base::Time()); |
| |
| scoped_refptr<net::IOBuffer> io_buffer( |
| new net::IOBuffer(fake_file_->metadata->size)); |
| const int result = |
| reader.Read(io_buffer.get(), |
| fake_file_->metadata->size, |
| base::Bind(&EventLogger::OnRead, logger.GetWeakPtr())); |
| |
| EXPECT_EQ(net::ERR_IO_PENDING, result); |
| base::RunLoop().RunUntilIdle(); |
| |
| ASSERT_EQ(1u, logger.results().size()); |
| EXPECT_EQ(fake_file_->metadata->size, logger.results()[0]); |
| |
| std::string buffer_as_string(io_buffer->data(), fake_file_->metadata->size); |
| EXPECT_EQ(fake_file_->contents, buffer_as_string); |
| } |
| |
| TEST_F(FileSystemProviderFileStreamReader, GetLength) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader( |
| NULL, file_url_, initial_offset, fake_file_->metadata->modification_time); |
| |
| 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(fake_file_->metadata->size, logger.results()[0]); |
| } |
| |
| TEST_F(FileSystemProviderFileStreamReader, GetLength_WrongFile) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader(NULL, |
| wrong_file_url_, |
| initial_offset, |
| fake_file_->metadata->modification_time); |
| |
| 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_FILE_NOT_FOUND, logger.results()[0]); |
| } |
| |
| TEST_F(FileSystemProviderFileStreamReader, GetLength_ModifiedFile) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader(NULL, file_url_, initial_offset, base::Time::Max()); |
| |
| 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_UPLOAD_FILE_CHANGED, logger.results()[0]); |
| } |
| |
| TEST_F(FileSystemProviderFileStreamReader, |
| GetLength_ExpectedModificationTimeNull) { |
| EventLogger logger; |
| |
| const int64 initial_offset = 0; |
| FileStreamReader reader(NULL, file_url_, initial_offset, base::Time()); |
| |
| 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(fake_file_->metadata->size, logger.results()[0]); |
| } |
| |
| } // namespace file_system_provider |
| } // namespace chromeos |