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
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "chrome/browser/web_applications/test/test_file_utils.h"
#include <cstdint>
#include <utility>
#include "base/files/file_path.h"
#include "base/memory/scoped_refptr.h"
namespace web_app {
scoped_refptr<TestFileUtils> TestFileUtils::Create(
std::map<base::FilePath, base::FilePath> read_file_rerouting) {
return base::MakeRefCounted<TestFileUtils>(std::move(read_file_rerouting));
}
TestFileUtils::TestFileUtils(
std::map<base::FilePath, base::FilePath> read_file_rerouting)
: read_file_rerouting_(std::move(read_file_rerouting)) {}
TestFileUtils::~TestFileUtils() = default;
void TestFileUtils::SetRemainingDiskSpaceSize(int remaining_disk_space) {
remaining_disk_space_ = remaining_disk_space;
}
void TestFileUtils::SetNextDeleteFileRecursivelyResult(
std::optional<bool> delete_result) {
delete_file_recursively_result_ = delete_result;
}
bool TestFileUtils::WriteFile(const base::FilePath& filename,
base::span<const uint8_t> file_data) {
if (remaining_disk_space_ != kNoLimit) {
int data_size = base::checked_cast<int>(file_data.size());
if (data_size > remaining_disk_space_) {
// Disk full:
const int size_written = remaining_disk_space_;
if (size_written > 0) {
FileUtilsWrapper::WriteFile(filename, file_data);
}
remaining_disk_space_ = 0;
return size_written;
}
remaining_disk_space_ -= file_data.size();
}
return FileUtilsWrapper::WriteFile(filename, file_data);
}
bool TestFileUtils::ReadFileToString(const base::FilePath& path,
std::string* contents) {
for (const std::pair<const base::FilePath, base::FilePath>& route :
read_file_rerouting_) {
if (route.first == path)
return FileUtilsWrapper::ReadFileToString(route.second, contents);
}
return FileUtilsWrapper::ReadFileToString(path, contents);
}
bool TestFileUtils::DeleteFileRecursively(const base::FilePath& path) {
return delete_file_recursively_result_
? *delete_file_recursively_result_
: FileUtilsWrapper::DeleteFileRecursively(path);
}
TestFileUtils* TestFileUtils::AsTestFileUtils() {
return this;
}
} // namespace web_app
|
