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 |
new file mode 100644 |
index 0000000000000000000000000000000000000000..8744c548eb8bb09b262c9abdb1321a34d1157658 |
--- /dev/null |
+++ b/chrome/browser/component_updater/test/component_patcher_unittest.cc |
@@ -0,0 +1,118 @@ |
+// Copyright (c) 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 "base/compiler_specific.h" |
+#include "base/file_util.h" |
+#include "base/files/file_path.h" |
+#include "base/files/scoped_temp_dir.h" |
+#include "base/path_service.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 "courgette/courgette.h" |
+#include "courgette/third_party/bsdiff.h" |
+#include "testing/gtest/include/gtest/gtest.h" |
+ |
+base::FilePath test_file(const char* file) { |
+ base::FilePath path; |
+ PathService::Get(chrome::DIR_TEST_DATA, &path); |
+ return path.AppendASCII("components").AppendASCII(file); |
+} |
+ |
+ComponentPatcherOperationTest::ComponentPatcherOperationTest() { |
+ 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())); |
+} |
+ |
+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. |
+TEST_F(ComponentPatcherOperationTest, CheckCreateOperation) { |
+ EXPECT_TRUE(file_util::CopyFile( |
+ test_file("binary_output.bin"), |
+ input_dir_.path().Append(FILE_PATH_LITERAL("binary_output.bin")))); |
+ |
+ scoped_ptr<base::DictionaryValue> command_args(new base::DictionaryValue()); |
+ command_args->SetString("output", "output.bin"); |
+ command_args->SetString("sha256", binary_output_hash); |
+ command_args->SetString("op", "create"); |
+ command_args->SetString("patch", "binary_output.bin"); |
+ |
+ 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(ComponentUnpacker::kNone, result); |
+ EXPECT_EQ(0, error); |
+ EXPECT_TRUE(file_util::ContentsEqual( |
+ unpack_dir_.path().Append(FILE_PATH_LITERAL("output.bin")), |
+ test_file("binary_output.bin"))); |
+} |
+ |
+// Verify that a 'copy' delta update operation works correctly. |
+TEST_F(ComponentPatcherOperationTest, CheckCopyOperation) { |
+ EXPECT_TRUE(file_util::CopyFile( |
+ test_file("binary_output.bin"), |
+ installed_dir_.path().Append(FILE_PATH_LITERAL("binary_output.bin")))); |
+ |
+ scoped_ptr<base::DictionaryValue> command_args(new base::DictionaryValue()); |
+ command_args->SetString("output", "output.bin"); |
+ command_args->SetString("sha256", binary_output_hash); |
+ command_args->SetString("op", "copy"); |
+ command_args->SetString("input", "binary_output.bin"); |
+ |
+ 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(file_util::ContentsEqual( |
+ unpack_dir_.path().Append(FILE_PATH_LITERAL("output.bin")), |
+ test_file("binary_output.bin"))); |
+} |