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
|
//===- CASTestConfig.h ----------------------------------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#ifndef LLVM_UNITTESTS_CASTESTCONFIG_H
#define LLVM_UNITTESTS_CASTESTCONFIG_H
#include "llvm/CAS/ActionCache.h"
#include "llvm/CAS/ObjectStore.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Testing/Support/Error.h"
#include "llvm/Testing/Support/SupportHelpers.h"
#include "gtest/gtest.h"
#include <memory>
namespace llvm::unittest::cas {
class MockEnv {
void anchor();
public:
virtual ~MockEnv();
};
} // namespace llvm::unittest::cas
struct TestingAndDir {
std::shared_ptr<llvm::cas::ObjectStore> CAS;
std::unique_ptr<llvm::cas::ActionCache> Cache;
std::unique_ptr<llvm::unittest::cas::MockEnv> Env;
std::optional<llvm::unittest::TempDir> Temp;
};
void setMaxOnDiskCASMappingSize();
class CASTest
: public testing::TestWithParam<std::function<TestingAndDir(int)>> {
protected:
std::optional<int> NextCASIndex;
llvm::SmallVector<llvm::unittest::TempDir> Dirs;
llvm::SmallVector<std::unique_ptr<llvm::unittest::cas::MockEnv>> Envs;
std::shared_ptr<llvm::cas::ObjectStore> createObjectStore() {
auto TD = GetParam()(++(*NextCASIndex));
if (TD.Temp)
Dirs.push_back(std::move(*TD.Temp));
if (TD.Env)
Envs.emplace_back(std::move(TD.Env));
return std::move(TD.CAS);
}
std::unique_ptr<llvm::cas::ActionCache> createActionCache() {
auto TD = GetParam()(++(*NextCASIndex));
if (TD.Temp)
Dirs.push_back(std::move(*TD.Temp));
if (TD.Env)
Envs.emplace_back(std::move(TD.Env));
return std::move(TD.Cache);
}
void SetUp() {
NextCASIndex = 0;
setMaxOnDiskCASMappingSize();
}
void TearDown() {
NextCASIndex = std::nullopt;
Dirs.clear();
Envs.clear();
}
};
#endif
|
