Chromium Code Reviews| Index: chrome/browser/chromeos/drive/drive_file_stream_reader_unittest.cc |
| diff --git a/chrome/browser/chromeos/drive/drive_file_stream_reader_unittest.cc b/chrome/browser/chromeos/drive/drive_file_stream_reader_unittest.cc |
| new file mode 100644 |
| index 0000000000000000000000000000000000000000..ff6a03ed01fabc5383280348dad1d2ece4727c2c |
| --- /dev/null |
| +++ b/chrome/browser/chromeos/drive/drive_file_stream_reader_unittest.cc |
| @@ -0,0 +1,79 @@ |
| +// Copyright 2013 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/drive/drive_file_stream_reader.h" |
| + |
| +#include <string> |
| + |
| +#include "base/file_util.h" |
| +#include "base/files/file_path.h" |
| +#include "base/message_loop.h" |
| +#include "chrome/browser/google_apis/task_util.h" |
| +#include "chrome/browser/google_apis/test_util.h" |
| +#include "content/public/test/test_browser_thread.h" |
| +#include "net/base/file_stream.h" |
| +#include "net/base/io_buffer.h" |
| +#include "net/base/net_errors.h" |
| +#include "testing/gtest/include/gtest/gtest.h" |
| + |
| +using content::BrowserThread; |
| + |
| +namespace drive { |
| +namespace internal { |
| + |
| +TEST(LocalReaderProxyTest, Read) { |
| + // Prepare the test content. |
| + const base::FilePath kTestFile( |
| + google_apis::test_util::GetTestFilePath("chromeos/drive/applist.json")); |
| + std::string expected_content; |
| + ASSERT_TRUE(file_util::ReadFileToString(kTestFile, &expected_content)); |
| + |
| + // The LocaReaderProxy should live on IO thread. |
| + MessageLoopForIO io_loop; |
| + content::TestBrowserThread io_thread(BrowserThread::IO, &io_loop); |
| + |
| + // Open the file first. |
| + scoped_ptr<net::FileStream> file_stream(new net::FileStream(NULL)); |
| + int result; |
| + net::CompletionCallback callback = |
|
hashimoto
2013/04/16 08:38:29
How about using net::TestCompletionCallback?
hidehiko
2013/04/16 08:57:11
Good to know. Done.
|
| + google_apis::CreateComposedCallback( |
| + base::Bind(google_apis::test_util::RunAndQuit), |
| + google_apis::test_util::CreateCopyResultCallback(&result)); |
| + result = file_stream->Open( |
| + google_apis::test_util::GetTestFilePath("chromeos/drive/applist.json"), |
| + base::PLATFORM_FILE_OPEN | base::PLATFORM_FILE_READ | |
| + base::PLATFORM_FILE_ASYNC, |
| + callback); |
| + if (result == net::ERR_IO_PENDING) { |
| + io_loop.Run(); |
| + } |
| + ASSERT_EQ(net::OK, result); |
| + |
| + // Test instance. |
| + LocalReaderProxy proxy(file_stream.Pass()); |
| + |
| + // Prepare the buffer, which size is smaller than the whole data size. |
|
hashimoto
2013/04/16 08:38:29
nit: s/which/whose/?
hidehiko
2013/04/16 08:57:11
Oops. Done.
|
| + const int kBufferSize = 10; |
| + ASSERT_LE(static_cast<size_t>(kBufferSize), expected_content.size()); |
| + scoped_refptr<net::IOBuffer> buffer(new net::IOBuffer(kBufferSize)); |
| + |
| + // Read repeatedly, until it is finished. |
| + std::string concatenated_content; |
| + while (concatenated_content.size() < expected_content.size()) { |
| + result = proxy.Read(buffer.get(), kBufferSize, callback); |
| + if (result == net::ERR_IO_PENDING) { |
| + io_loop.Run(); |
| + } |
| + |
| + // The read content size should be smaller than the buffer size. |
| + ASSERT_LE(result, kBufferSize); |
| + concatenated_content.append(buffer->data(), result); |
| + } |
| + |
| + // Make sure the read contant is as same as the file. |
| + EXPECT_EQ(expected_content, concatenated_content); |
| +} |
| + |
| +} // namespace internal |
| +} // namespace drive |