| Index: chrome/browser/component_updater/test/component_patcher_unittest.cc
|
| diff --git a/chrome/browser/component_updater/test/component_patcher_unittest.cc b/chrome/browser/component_updater/test/component_patcher_unittest.cc
|
| index 8769545f276048c4c13f18637c11280f75b2edaa..3cc4f65610b8cbce7d2f26f08b35ba91a036ca50 100644
|
| --- a/chrome/browser/component_updater/test/component_patcher_unittest.cc
|
| +++ b/chrome/browser/component_updater/test/component_patcher_unittest.cc
|
| @@ -2,54 +2,22 @@
|
| // Use of this source code is governed by a BSD-style license that can be
|
| // found in the LICENSE file.
|
|
|
| -#include "base/bind.h"
|
| #include "base/compiler_specific.h"
|
| #include "base/file_util.h"
|
| #include "base/files/file_path.h"
|
| #include "base/files/scoped_temp_dir.h"
|
| -#include "base/message_loop/message_loop.h"
|
| #include "base/path_service.h"
|
| -#include "base/run_loop.h"
|
| #include "base/values.h"
|
| #include "chrome/browser/component_updater/component_patcher.h"
|
| #include "chrome/browser/component_updater/component_patcher_operation.h"
|
| #include "chrome/browser/component_updater/component_updater_service.h"
|
| +#include "chrome/browser/component_updater/test/component_patcher_mock.h"
|
| #include "chrome/browser/component_updater/test/component_patcher_unittest.h"
|
| #include "chrome/browser/component_updater/test/test_installer.h"
|
| #include "chrome/common/chrome_paths.h"
|
| -#include "content/public/browser/browser_thread.h"
|
| #include "courgette/courgette.h"
|
| #include "courgette/third_party/bsdiff.h"
|
| #include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -namespace {
|
| -
|
| -class TestCallback {
|
| - public:
|
| - TestCallback();
|
| - virtual ~TestCallback() {}
|
| - void Set(component_updater::ComponentUnpacker::Error error, int extra_code);
|
| -
|
| - int error_;
|
| - int extra_code_;
|
| - bool called_;
|
| -
|
| - private:
|
| - DISALLOW_COPY_AND_ASSIGN(TestCallback);
|
| -};
|
| -
|
| -TestCallback::TestCallback()
|
| - : error_(-1), extra_code_(-1), called_(false) {
|
| -}
|
| -
|
| -void TestCallback::Set(component_updater::ComponentUnpacker::Error error,
|
| - int extra_code) {
|
| - error_ = error;
|
| - extra_code_ = extra_code;
|
| - called_ = true;
|
| -}
|
| -
|
| -} // namespace
|
|
|
| namespace component_updater {
|
|
|
| @@ -67,12 +35,37 @@
|
| EXPECT_TRUE(unpack_dir_.CreateUniqueTempDir());
|
| EXPECT_TRUE(input_dir_.CreateUniqueTempDir());
|
| EXPECT_TRUE(installed_dir_.CreateUniqueTempDir());
|
| + patcher_.reset(new MockComponentPatcher());
|
| installer_.reset(new ReadOnlyTestInstaller(installed_dir_.path()));
|
| - task_runner_ = content::BrowserThread::GetMessageLoopProxyForThread(
|
| - content::BrowserThread::FILE);
|
| }
|
|
|
| ComponentPatcherOperationTest::~ComponentPatcherOperationTest() {
|
| +}
|
| +
|
| +ComponentUnpacker::Error MockComponentPatcher::Patch(
|
| + PatchType patch_type,
|
| + const base::FilePath& input_file,
|
| + const base::FilePath& patch_file,
|
| + const base::FilePath& output_file,
|
| + int* error) {
|
| + *error = 0;
|
| + int exit_code;
|
| + if (patch_type == kPatchTypeCourgette) {
|
| + exit_code = courgette::ApplyEnsemblePatch(input_file.value().c_str(),
|
| + patch_file.value().c_str(),
|
| + output_file.value().c_str());
|
| + if (exit_code == courgette::C_OK)
|
| + return ComponentUnpacker::kNone;
|
| + *error = exit_code + kCourgetteErrorOffset;
|
| + } else if (patch_type == kPatchTypeBsdiff) {
|
| + exit_code = courgette::ApplyBinaryPatch(input_file,
|
| + patch_file,
|
| + output_file);
|
| + if (exit_code == courgette::OK)
|
| + return ComponentUnpacker::kNone;
|
| + *error = exit_code + kBsdiffErrorOffset;
|
| + }
|
| + return ComponentUnpacker::kDeltaOperationFailure;
|
| }
|
|
|
| // Verify that a 'create' delta update operation works correctly.
|
| @@ -87,20 +80,17 @@
|
| command_args->SetString("op", "create");
|
| command_args->SetString("patch", "binary_output.bin");
|
|
|
| - TestCallback callback;
|
| - scoped_refptr<DeltaUpdateOp> op = new DeltaUpdateOpCreate();
|
| - op->Run(command_args.get(),
|
| - input_dir_.path(),
|
| - unpack_dir_.path(),
|
| - NULL,
|
| - true,
|
| - base::Bind(&TestCallback::Set, base::Unretained(&callback)),
|
| - task_runner_);
|
| - base::RunLoop().RunUntilIdle();
|
| + int error = 0;
|
| + scoped_ptr<DeltaUpdateOp> op(new DeltaUpdateOpCreate());
|
| + ComponentUnpacker::Error result = op->Run(command_args.get(),
|
| + input_dir_.path(),
|
| + unpack_dir_.path(),
|
| + patcher_.get(),
|
| + NULL,
|
| + &error);
|
|
|
| - EXPECT_EQ(true, callback.called_);
|
| - EXPECT_EQ(ComponentUnpacker::kNone, callback.error_);
|
| - EXPECT_EQ(0, callback.extra_code_);
|
| + EXPECT_EQ(ComponentUnpacker::kNone, result);
|
| + EXPECT_EQ(0, error);
|
| EXPECT_TRUE(base::ContentsEqual(
|
| unpack_dir_.path().Append(FILE_PATH_LITERAL("output.bin")),
|
| test_file("binary_output.bin")));
|
| @@ -118,20 +108,16 @@
|
| command_args->SetString("op", "copy");
|
| command_args->SetString("input", "binary_output.bin");
|
|
|
| - TestCallback callback;
|
| - scoped_refptr<DeltaUpdateOp> op = new DeltaUpdateOpCopy();
|
| - op->Run(command_args.get(),
|
| - input_dir_.path(),
|
| - unpack_dir_.path(),
|
| - installer_.get(),
|
| - true,
|
| - base::Bind(&TestCallback::Set, base::Unretained(&callback)),
|
| - task_runner_);
|
| - base::RunLoop().RunUntilIdle();
|
| -
|
| - EXPECT_EQ(true, callback.called_);
|
| - EXPECT_EQ(ComponentUnpacker::kNone, callback.error_);
|
| - EXPECT_EQ(0, callback.extra_code_);
|
| + int error = 0;
|
| + scoped_ptr<DeltaUpdateOp> op(new DeltaUpdateOpCopy());
|
| + ComponentUnpacker::Error result = op->Run(command_args.get(),
|
| + input_dir_.path(),
|
| + unpack_dir_.path(),
|
| + patcher_.get(),
|
| + installer_.get(),
|
| + &error);
|
| + EXPECT_EQ(ComponentUnpacker::kNone, result);
|
| + EXPECT_EQ(0, error);
|
| EXPECT_TRUE(base::ContentsEqual(
|
| unpack_dir_.path().Append(FILE_PATH_LITERAL("output.bin")),
|
| test_file("binary_output.bin")));
|
| @@ -154,20 +140,16 @@
|
| command_args->SetString("input", "binary_input.bin");
|
| command_args->SetString("patch", "binary_courgette_patch.bin");
|
|
|
| - TestCallback callback;
|
| - scoped_refptr<DeltaUpdateOp> op = CreateDeltaUpdateOp("courgette");
|
| - op->Run(command_args.get(),
|
| - input_dir_.path(),
|
| - unpack_dir_.path(),
|
| - installer_.get(),
|
| - true,
|
| - base::Bind(&TestCallback::Set, base::Unretained(&callback)),
|
| - task_runner_);
|
| - base::RunLoop().RunUntilIdle();
|
| -
|
| - EXPECT_EQ(true, callback.called_);
|
| - EXPECT_EQ(ComponentUnpacker::kNone, callback.error_);
|
| - EXPECT_EQ(0, callback.extra_code_);
|
| + int error = 0;
|
| + scoped_ptr<DeltaUpdateOp> op(new DeltaUpdateOpPatchCourgette());
|
| + ComponentUnpacker::Error result = op->Run(command_args.get(),
|
| + input_dir_.path(),
|
| + unpack_dir_.path(),
|
| + patcher_.get(),
|
| + installer_.get(),
|
| + &error);
|
| + EXPECT_EQ(ComponentUnpacker::kNone, result);
|
| + EXPECT_EQ(0, error);
|
| EXPECT_TRUE(base::ContentsEqual(
|
| unpack_dir_.path().Append(FILE_PATH_LITERAL("output.bin")),
|
| test_file("binary_output.bin")));
|
| @@ -189,20 +171,16 @@
|
| command_args->SetString("input", "binary_input.bin");
|
| command_args->SetString("patch", "binary_bsdiff_patch.bin");
|
|
|
| - TestCallback callback;
|
| - scoped_refptr<DeltaUpdateOp> op = CreateDeltaUpdateOp("bsdiff");
|
| - op->Run(command_args.get(),
|
| - input_dir_.path(),
|
| - unpack_dir_.path(),
|
| - installer_.get(),
|
| - true,
|
| - base::Bind(&TestCallback::Set, base::Unretained(&callback)),
|
| - task_runner_);
|
| - base::RunLoop().RunUntilIdle();
|
| -
|
| - EXPECT_EQ(true, callback.called_);
|
| - EXPECT_EQ(ComponentUnpacker::kNone, callback.error_);
|
| - EXPECT_EQ(0, callback.extra_code_);
|
| + int error = 0;
|
| + scoped_ptr<DeltaUpdateOp> op(new DeltaUpdateOpPatchBsdiff());
|
| + ComponentUnpacker::Error result = op->Run(command_args.get(),
|
| + input_dir_.path(),
|
| + unpack_dir_.path(),
|
| + patcher_.get(),
|
| + installer_.get(),
|
| + &error);
|
| + EXPECT_EQ(ComponentUnpacker::kNone, result);
|
| + EXPECT_EQ(0, error);
|
| EXPECT_TRUE(base::ContentsEqual(
|
| unpack_dir_.path().Append(FILE_PATH_LITERAL("output.bin")),
|
| test_file("binary_output.bin")));
|
|
|