| Index: trunk/src/content/browser/loader/temporary_file_stream_unittest.cc
|
| ===================================================================
|
| --- trunk/src/content/browser/loader/temporary_file_stream_unittest.cc (revision 256703)
|
| +++ trunk/src/content/browser/loader/temporary_file_stream_unittest.cc (working copy)
|
| @@ -1,120 +0,0 @@
|
| -// 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 "content/browser/loader/temporary_file_stream.h"
|
| -
|
| -#include <string.h>
|
| -
|
| -#include <string>
|
| -
|
| -#include "base/basictypes.h"
|
| -#include "base/bind.h"
|
| -#include "base/file_util.h"
|
| -#include "base/files/file_path.h"
|
| -#include "base/run_loop.h"
|
| -#include "content/public/test/test_browser_thread_bundle.h"
|
| -#include "net/base/file_stream.h"
|
| -#include "net/base/io_buffer.h"
|
| -#include "net/base/net_errors.h"
|
| -#include "net/base/test_completion_callback.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -#include "webkit/common/blob/shareable_file_reference.h"
|
| -
|
| -using webkit_blob::ShareableFileReference;
|
| -
|
| -namespace content {
|
| -
|
| -namespace {
|
| -
|
| -const char kTestData[] = "0123456789";
|
| -const int kTestDataSize = arraysize(kTestData) - 1;
|
| -
|
| -class WaitForFileStream {
|
| - public:
|
| - base::File::Error error() const { return error_; }
|
| - net::FileStream* file_stream() const { return file_stream_.get(); }
|
| - ShareableFileReference* deletable_file() const {
|
| - return deletable_file_.get();
|
| - }
|
| -
|
| - void OnFileStreamCreated(base::File::Error error,
|
| - scoped_ptr<net::FileStream> file_stream,
|
| - ShareableFileReference* deletable_file) {
|
| - error_ = error;
|
| - file_stream_ = file_stream.Pass();
|
| - deletable_file_ = deletable_file;
|
| - loop_.Quit();
|
| - }
|
| -
|
| - void Wait() {
|
| - loop_.Run();
|
| - }
|
| -
|
| - void Release() {
|
| - file_stream_.reset(NULL);
|
| - deletable_file_ = NULL;
|
| - }
|
| - private:
|
| - base::RunLoop loop_;
|
| - base::File::Error error_;
|
| - scoped_ptr<net::FileStream> file_stream_;
|
| - scoped_refptr<ShareableFileReference> deletable_file_;
|
| -};
|
| -
|
| -} // namespace
|
| -
|
| -TEST(TemporaryFileStreamTest, Basic) {
|
| - TestBrowserThreadBundle thread_bundle(TestBrowserThreadBundle::IO_MAINLOOP);
|
| -
|
| - // Create a temporary.
|
| - WaitForFileStream file_stream_waiter;
|
| - CreateTemporaryFileStream(base::Bind(&WaitForFileStream::OnFileStreamCreated,
|
| - base::Unretained(&file_stream_waiter)));
|
| - file_stream_waiter.Wait();
|
| -
|
| - // The temporary should exist.
|
| - EXPECT_EQ(base::File::FILE_OK, file_stream_waiter.error());
|
| - base::FilePath file_path = file_stream_waiter.deletable_file()->path();
|
| - EXPECT_TRUE(base::PathExists(file_path));
|
| -
|
| - // Write some data to the temporary.
|
| - int bytes_written = 0;
|
| - scoped_refptr<net::IOBufferWithSize> buf =
|
| - new net::IOBufferWithSize(kTestDataSize);
|
| - memcpy(buf->data(), kTestData, kTestDataSize);
|
| - scoped_refptr<net::DrainableIOBuffer> drainable =
|
| - new net::DrainableIOBuffer(buf.get(), buf->size());
|
| - while (bytes_written != kTestDataSize) {
|
| - net::TestCompletionCallback write_callback;
|
| - int rv = file_stream_waiter.file_stream()->Write(
|
| - drainable.get(), drainable->BytesRemaining(),
|
| - write_callback.callback());
|
| - if (rv == net::ERR_IO_PENDING)
|
| - rv = write_callback.WaitForResult();
|
| - ASSERT_LT(0, rv);
|
| - drainable->DidConsume(rv);
|
| - bytes_written += rv;
|
| - }
|
| -
|
| - // Verify the data matches.
|
| - std::string contents;
|
| - ASSERT_TRUE(base::ReadFileToString(file_path, &contents));
|
| - EXPECT_EQ(kTestData, contents);
|
| -
|
| - // Close the file.
|
| - net::TestCompletionCallback close_callback;
|
| - int rv = file_stream_waiter.file_stream()->Close(close_callback.callback());
|
| - if (rv == net::ERR_IO_PENDING)
|
| - rv = close_callback.WaitForResult();
|
| - EXPECT_EQ(net::OK, rv);
|
| -
|
| - // Release everything. The file should be gone now.
|
| - file_stream_waiter.Release();
|
| - base::MessageLoop::current()->RunUntilIdle();
|
| -
|
| - // The temporary should be gone now.
|
| - EXPECT_FALSE(base::PathExists(file_path));
|
| -}
|
| -
|
| -} // content
|
|
|