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
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
|
// 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 "components/update_client/test_installer.h"
#include <string>
#include <utility>
#include "base/bind.h"
#include "base/callback.h"
#include "base/files/file_path.h"
#include "base/files/file_util.h"
#include "base/task/post_task.h"
#include "base/task/task_traits.h"
#include "base/task/thread_pool.h"
#include "base/threading/sequenced_task_runner_handle.h"
#include "base/values.h"
#include "components/update_client/update_client_errors.h"
#include "components/update_client/utils.h"
#include "testing/gtest/include/gtest/gtest.h"
namespace update_client {
TestInstaller::TestInstaller()
: error_(0),
install_count_(0),
task_runner_(base::SequencedTaskRunnerHandle::Get()) {}
TestInstaller::~TestInstaller() {
// The unpack path is deleted unconditionally by the component state code,
// which is driving this installer. Therefore, the unpack path must not
// exist when this object is destroyed.
if (!unpack_path_.empty())
EXPECT_FALSE(base::DirectoryExists(unpack_path_));
}
void TestInstaller::OnUpdateError(int error) {
error_ = error;
}
void TestInstaller::Install(const base::FilePath& unpack_path,
const std::string& /*public_key*/,
std::unique_ptr<InstallParams> install_params,
ProgressCallback progress_callback,
Callback callback) {
++install_count_;
unpack_path_ = unpack_path;
install_params_ = std::move(install_params);
InstallComplete(std::move(callback), progress_callback,
Result(InstallError::NONE));
}
void TestInstaller::InstallComplete(Callback callback,
ProgressCallback progress_callback,
const Result& result) {
for (auto sample : installer_progress_samples_) {
progress_callback.Run(sample);
}
task_runner_->PostTask(FROM_HERE,
base::BindOnce(std::move(callback), result));
}
bool TestInstaller::GetInstalledFile(const std::string& file,
base::FilePath* installed_file) {
return false;
}
bool TestInstaller::Uninstall() {
return false;
}
ReadOnlyTestInstaller::ReadOnlyTestInstaller(const base::FilePath& install_dir)
: install_directory_(install_dir) {}
ReadOnlyTestInstaller::~ReadOnlyTestInstaller() = default;
bool ReadOnlyTestInstaller::GetInstalledFile(const std::string& file,
base::FilePath* installed_file) {
*installed_file = install_directory_.AppendASCII(file);
return true;
}
VersionedTestInstaller::VersionedTestInstaller() {
base::CreateNewTempDirectory(FILE_PATH_LITERAL("TEST_"), &install_directory_);
}
VersionedTestInstaller::~VersionedTestInstaller() {
base::DeletePathRecursively(install_directory_);
}
void VersionedTestInstaller::Install(
const base::FilePath& unpack_path,
const std::string& public_key,
std::unique_ptr<InstallParams> /*install_params*/,
ProgressCallback progress_callback,
Callback callback) {
const auto manifest = update_client::ReadManifest(unpack_path);
std::string version_string;
manifest->GetStringASCII("version", &version_string);
const base::Version version(version_string);
const base::FilePath path =
install_directory_.AppendASCII(version.GetString());
base::CreateDirectory(path.DirName());
if (!base::Move(unpack_path, path)) {
InstallComplete(std::move(callback), progress_callback,
Result(InstallError::GENERIC_ERROR));
return;
}
current_version_ = version;
++install_count_;
InstallComplete(std::move(callback), progress_callback,
Result(InstallError::NONE));
}
bool VersionedTestInstaller::GetInstalledFile(const std::string& file,
base::FilePath* installed_file) {
const base::FilePath path =
install_directory_.AppendASCII(current_version_.GetString());
*installed_file = path.Append(base::FilePath::FromUTF8Unsafe(file));
return true;
}
} // namespace update_client
|
