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
|
#include "llvm/CAS/BuiltinUnifiedCASDatabases.h"
#include "llvm/CAS/ActionCache.h"
#include "llvm/CAS/ObjectStore.h"
#include "llvm/Testing/Support/Error.h"
#include "llvm/Testing/Support/SupportHelpers.h"
#include "gtest/gtest.h"
using namespace llvm;
using namespace llvm::cas;
TEST(BuiltinUnifiedCASDatabases,
MaterializationCheckPreventsGarbageCollection) {
unittest::TempDir Temp("on-disk-unified-cas", /*Unique=*/true);
auto WithCAS = [&](llvm::function_ref<void(ObjectStore &)> Action) {
std::pair<std::unique_ptr<ObjectStore>, std::unique_ptr<ActionCache>> DBs;
ASSERT_THAT_ERROR(
createOnDiskUnifiedCASDatabases(Temp.path()).moveInto(DBs),
Succeeded());
ObjectStore &CAS = *DBs.first;
ASSERT_THAT_ERROR(CAS.setSizeLimit(1), Succeeded());
Action(CAS);
};
std::optional<CASID> ID;
// Create an object in the CAS.
WithCAS([&ID](ObjectStore &CAS) {
std::optional<ObjectRef> Ref;
ASSERT_THAT_ERROR(CAS.store({}, "blah").moveInto(Ref), Succeeded());
ASSERT_TRUE(Ref.has_value());
ID = CAS.getID(*Ref);
});
// Check materialization and prune the storage.
WithCAS([&ID](ObjectStore &CAS) {
std::optional<ObjectRef> Ref = CAS.getReference(*ID);
ASSERT_TRUE(Ref.has_value());
std::optional<bool> IsMaterialized;
ASSERT_THAT_ERROR(CAS.isMaterialized(*Ref).moveInto(IsMaterialized),
Succeeded());
ASSERT_TRUE(IsMaterialized);
ASSERT_THAT_ERROR(CAS.pruneStorageData(), Succeeded());
});
// Verify that the previous materialization check kept the object in the CAS.
WithCAS([&ID](ObjectStore &CAS) {
std::optional<ObjectRef> Ref = CAS.getReference(*ID);
ASSERT_TRUE(Ref.has_value());
std::optional<bool> IsMaterialized;
ASSERT_THAT_ERROR(CAS.isMaterialized(*Ref).moveInto(IsMaterialized),
Succeeded());
ASSERT_TRUE(IsMaterialized);
});
}
|
