blob: b2f4f05b6beb2edd368f2d22016d8f2435695674 (
plain)
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
|
#include "common.hh"
#include "file_test.hh"
#include "io.hh"
#include "str_buffer.hh"
FileTest::FileTest() = default;
FileTest::~FileTest() {
if (!path_.empty()) {
std::error_code err;
std::filesystem::remove(path_, err);
}
}
std::string const& FileTest::extension() {
static std::string empty;
return empty;
}
void FileTest::SetUp() {
fd_ = create_temp_file(extension(), &path_);
ASSERT_FALSE(path_.empty());
}
void FileTest::write(std::string_view content) {
ASSERT_TRUE(fd_);
auto buffer = make_strbuffer(content);
while (!buffer->empty()) {
ASSERT_TRUE(io::drain(buffer.get(), fd_.get()));
}
close();
}
void FileTest::close() {
if (fd_) {
ASSERT_TRUE(io::close(fd_.release()));
}
}
unique_fd FileTest::create_temp_file(std::string const& extension,
std::filesystem::path* path) {
std::error_code err;
auto tmpdir = std::filesystem::temp_directory_path(err);
if (tmpdir.empty())
return unique_fd();
unique_fd ret;
for (uint8_t i = 0; i < 0xff; ++i) {
char name[50];
snprintf(name, sizeof(name), "test-%u%s", i, extension.c_str());
auto test = tmpdir / name;
ret = io::open(test, io::open_flags::wronly | io::open_flags::create |
io::open_flags::excl,
std::filesystem::perms::owner_read |
std::filesystem::perms::owner_write);
if (ret) {
*path = test;
break;
}
if (errno != EEXIST)
break;
}
return ret;
}
|
