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
128
129
130
131
132
133
134
135
136
137
138
139
140
|
//===----------------------------------------------------------------------===//
//
// DuckDB
//
// test_config.hpp
//
//
//===----------------------------------------------------------------------===//
#pragma once
#include "duckdb/common/common.hpp"
#include "duckdb/common/optional_idx.hpp"
#include "duckdb/common/enums/debug_vector_verification.hpp"
#include "duckdb/common/types/value.hpp"
#include "duckdb/common/atomic.hpp"
#include "duckdb/common/mutex.hpp"
#include "duckdb/common/enums/debug_initialize.hpp"
#include <sys/types.h>
#include <unordered_map>
namespace duckdb {
enum class SortStyle : uint8_t { NO_SORT, ROW_SORT, VALUE_SORT };
struct ConfigSetting {
string name;
Value value;
};
class TestConfiguration {
public:
enum class ExtensionAutoLoadingMode { NONE = 0, AVAILABLE = 1, ALL = 2 };
enum class SelectPolicy : uint8_t {
NONE, // does not match any explicit policy (default: policy=SELECT)
SELECT, // matches explicit select
SKIP // matches explicit skip
};
static TestConfiguration &Get();
void Initialize();
bool ParseArgument(const string &arg, idx_t argc, char **argv, idx_t &i);
bool TryParseOption(const string &name, const Value &value);
void ParseOption(const string &name, const Value &value);
void LoadConfig(const string &config_path);
void UpdateEnvironment();
string GetWorkingDirectory();
bool ChangeWorkingDirectory(const string &dir); // true -> changed
void ProcessPath(string &path, const string &test_name);
string GetDescription();
string GetInitialDBPath();
optional_idx GetMaxThreads();
optional_idx GetBlockAllocSize();
idx_t GetCheckpointWALSize();
bool GetForceRestart();
bool GetCheckpointOnShutdown();
bool GetTestMemoryLeaks();
bool RunStorageFuzzer();
bool GetSummarizeFailures();
bool GetSkipCompiledTests();
DebugVectorVerification GetVectorVerification();
DebugInitialize GetDebugInitialize();
ExtensionAutoLoadingMode GetExtensionAutoLoadingMode();
bool ShouldSkipTest(const string &test_name);
string DataLocation();
string OnInitCommand();
string OnLoadCommand();
string OnConnectionCommand();
string OnCleanupCommand();
SortStyle GetDefaultSortStyle();
vector<string> ExtensionToBeLoadedOnLoad();
vector<string> ErrorMessagesToBeSkipped();
string GetStorageVersion();
string GetTestEnv(const string &key, const string &default_value);
const unordered_map<string, string> &GetTestEnvMap();
vector<unordered_set<string>> GetSelectTagSets();
vector<unordered_set<string>> GetSkipTagSets();
SelectPolicy GetPolicyForTagSet(const vector<string> &tag_set);
vector<ConfigSetting> GetConfigSettings();
static bool TestForceStorage();
static bool TestForceReload();
static bool TestMemoryLeaks();
static bool TestRunStorageFuzzer();
static void LoadBaseConfig(const Value &input);
static void ParseConnectScript(const Value &input);
static void CheckSortStyle(const Value &input);
static bool TryParseSortStyle(const string &sort_style, SortStyle &result);
static void AppendSelectTagSet(const Value &tag_set);
static void AppendSkipTagSet(const Value &tag_set);
private:
//! Give preference to settings from loaded configs
bool test_env_from_config_loaded = false;
case_insensitive_map_t<Value> options;
unordered_set<string> tests_to_be_skipped;
// explicitly take ownership of working_dir here, giving runners an API to chdir,
// and get env updates to match
string working_dir;
string test_uuid;
unordered_map<string, string> test_env;
vector<unordered_set<string>> select_tag_sets;
vector<unordered_set<string>> skip_tag_sets;
private:
template <class T, class VAL_T = T>
T GetOptionOrDefault(const string &name, T default_val);
static string ReadFileToString(const string &path);
};
class FailureSummary {
public:
FailureSummary();
static void Log(string message);
static string GetFailureSummary();
static idx_t GetSummaryCounter();
static bool SkipLoggingSameError(const string &file_name);
private:
static FailureSummary &Instance();
bool SkipLoggingSameErrorInternal(const string &file_name);
private:
mutex failures_lock;
atomic<idx_t> failures_summary_counter;
vector<string> failures_summary;
set<string> reported_files;
};
} // namespace duckdb
|
