1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91
|
// Copyright 2025 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <string_view>
#include <utility>
#include "base/base_paths.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "base/path_service.h"
#include "base/run_loop.h"
#include "base/test/bind.h"
#include "base/test/task_environment.h"
#include "components/services/patch/file_patcher_impl.h"
#include "components/services/patch/in_process_file_patcher.h"
#include "components/services/patch/public/cpp/patch.h"
#include "mojo/public/cpp/bindings/remote.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace patch {
namespace {
base::FilePath TestFile(std::string_view basename) {
base::FilePath path;
EXPECT_TRUE(base::PathService::Get(base::DIR_SRC_TEST_DATA_ROOT, &path));
return path.AppendASCII("components")
.AppendASCII("test")
.AppendASCII("data")
.AppendASCII("patch_service")
.AppendASCII(basename);
}
int DoZucchini(const base::FilePath& old_file,
const base::FilePath& patch_file,
const base::FilePath& new_file) {
base::RunLoop run_loop;
zucchini::status::Code result = zucchini::status::kStatusSuccess;
base::File old =
base::File(old_file, base::File::FLAG_OPEN | base::File::FLAG_READ);
base::File patch =
base::File(patch_file, base::File::FLAG_OPEN | base::File::FLAG_READ);
base::File output = base::File(
new_file, base::File::FLAG_CREATE | base::File::FLAG_READ |
base::File::FLAG_WRITE | base::File::FLAG_WIN_SHARE_DELETE |
base::File::FLAG_CAN_DELETE_ON_CLOSE);
EXPECT_TRUE(old.IsValid());
EXPECT_TRUE(patch.IsValid());
EXPECT_TRUE(output.IsValid());
ZucchiniPatch(LaunchInProcessFilePatcher(), std::move(old), std::move(patch),
std::move(output),
base::BindLambdaForTesting([&](zucchini::status::Code code) {
result = code;
run_loop.QuitClosure().Run();
}));
run_loop.Run();
return result;
}
} // namespace
class PatchTest : public testing::Test {
protected:
void SetUp() override {
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
new_file_ = temp_dir_.GetPath().AppendASCII("new_file");
}
base::test::TaskEnvironment task_environment_;
base::ScopedTempDir temp_dir_;
base::FilePath new_file_;
};
TEST_F(PatchTest, ZucchiniPatchSuccess) {
ASSERT_EQ(DoZucchini(TestFile("old_file"), TestFile("patch_file"), new_file_),
0);
EXPECT_EQ(ReadFileToBytes(new_file_), ReadFileToBytes(TestFile("new_file")));
}
TEST_F(PatchTest, ZucchiniPatchFail) {
ASSERT_NE(DoZucchini(TestFile("new_file"), TestFile("patch_file"), new_file_),
0);
}
TEST_F(PatchTest, ZucchiniPatchMalformed) {
ASSERT_NE(DoZucchini(TestFile("old_file"), TestFile("old_file"), new_file_),
0);
}
} // namespace patch
|