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
|
#include "libstore/mock-binary-cache-store.hh"
#include <glog/logging.h>
namespace nix {
MockBinaryCacheStore::MockBinaryCacheStore(const Params& params)
: BinaryCacheStore(params), contents_(), errorInjections_() {}
std::string MockBinaryCacheStore::getUri() { return "mock://1"; }
bool MockBinaryCacheStore::fileExists(const std::string& path) {
ThrowInjectedErrors(path);
return contents_.find(path) != contents_.end();
};
void MockBinaryCacheStore::upsertFile(const std::string& path,
const std::string& data,
const std::string& mimeType) {
ThrowInjectedErrors(path);
contents_[path] = MemoryFile{data, mimeType};
}
void MockBinaryCacheStore::getFile(
const std::string& path,
Callback<std::shared_ptr<std::string>> callback) noexcept {
auto eit = errorInjections_.find(path);
if (eit != errorInjections_.end()) {
try {
eit->second();
LOG(FATAL) << "thrower failed to throw";
} catch (...) {
callback.rethrow();
}
return;
}
auto it = contents_.find(path);
if (it == contents_.end()) {
try {
throw NoSuchBinaryCacheFile(absl::StrCat(
"file '", path, "' was not added to the MockBinaryCache"));
} catch (...) {
callback.rethrow();
}
return;
}
callback(std::make_shared<std::string>(it->second.data));
}
PathSet MockBinaryCacheStore::queryAllValidPaths() {
PathSet paths;
for (auto it : contents_) {
paths.insert(it.first);
}
return paths;
}
void MockBinaryCacheStore::DeleteFile(const std::string& path) {
contents_.erase(path);
}
// Same as upsert, but bypasses injected errors.
void MockBinaryCacheStore::SetFileContentsForTest(const std::string& path,
const std::string& data,
const std::string& mimeType) {
contents_[path] = MemoryFile{data, mimeType};
}
void MockBinaryCacheStore::PrepareErrorInjection(
const std::string& path, std::function<void()> err_factory) {
errorInjections_[path] = err_factory;
}
void MockBinaryCacheStore::CancelErrorInjection(const std::string& path) {
errorInjections_.erase(path);
}
void MockBinaryCacheStore::ThrowInjectedErrors(const std::string& path) {
auto it = errorInjections_.find(path);
if (it != errorInjections_.end()) {
it->second();
LOG(FATAL) << "thrower failed to throw";
}
}
} // namespace nix
|
