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
|
#include "tests.pch.h"
#include <vcpkg/base/files.h>
#include <vcpkg/base/strings.h>
#include <iostream>
#include <random>
#include <windows.h>
using namespace Microsoft::VisualStudio::CppUnitTestFramework;
namespace UnitTest1 {
class FilesTest : public TestClass<FilesTest> {
using uid = std::uniform_int_distribution<std::uint64_t>;
std::string get_random_filename()
{
std::random_device rd;
return vcpkg::Strings::b64url_encode(uid{}(rd));
}
void create_directory_tree(
vcpkg::Files::Filesystem& fs,
std::uint64_t depth,
const fs::path& base)
{
std::random_device rd;
constexpr auto max_depth = std::uint64_t(3);
const auto width = depth ? uid{0, (max_depth - depth) * 3 / 2}(rd) : 5;
std::error_code ec;
if (width == 0) {
fs.write_contents(base, "", ec);
Assert::IsFalse(bool(ec));
return;
}
fs.create_directory(base, ec);
Assert::IsFalse(bool(ec));
for (int i = 0; i < width; ++i) {
create_directory_tree(fs, depth + 1, base / get_random_filename());
}
}
TEST_METHOD(remove_all) {
fs::path temp_dir;
{
wchar_t* tmp = static_cast<wchar_t*>(calloc(32'767, 2));
if (!GetEnvironmentVariableW(L"TEMP", tmp, 32'767)) {
Assert::Fail(L"GetEnvironmentVariable(\"TEMP\") failed");
}
temp_dir = tmp;
std::string dir_name = "vcpkg-tmp-dir-";
dir_name += get_random_filename();
temp_dir /= dir_name;
}
auto& fs = vcpkg::Files::get_real_filesystem();
std::cout << "temp dir is: " << temp_dir << '\n';
std::error_code ec;
create_directory_tree(fs, 0, temp_dir);
fs::path fp;
fs.remove_all(temp_dir, ec, fp);
Assert::IsFalse(bool(ec));
Assert::IsFalse(fs.exists(temp_dir));
}
};
}
|
