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
|
//===----------------------------------------------------------------------===//
//
// The LLVM Compiler Infrastructure
//
// This file is dual licensed under the MIT and the University of Illinois Open
// Source Licenses. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
//
// Copyright 2019-2020, Intel Corporation
//
// Modified to test pmem::obj containers
//
#include "unittest.hpp"
#include <libpmemobj++/container/string.hpp>
namespace nvobj = pmem::obj;
using S = pmem::obj::string;
struct root {
nvobj::persistent_ptr<S> s, s0;
};
template <class S>
void
test(nvobj::pool<struct root> &pop, S &s)
{
auto r = pop.root();
nvobj::transaction::run(pop,
[&] { r->s0 = nvobj::make_persistent<S>(s); });
auto &s0 = *r->s0;
typename S::size_type old_cap = s.capacity();
s.shrink_to_fit();
UT_ASSERT(s == s0);
UT_ASSERT(s.capacity() <= old_cap);
UT_ASSERT(s.capacity() >= s.size());
nvobj::transaction::run(pop,
[&] { nvobj::delete_persistent<S>(r->s0); });
}
static void
test(int argc, char *argv[])
{
if (argc < 2) {
UT_FATAL("usage: %s file-name", argv[0]);
}
auto path = argv[1];
auto pop = nvobj::pool<root>::create(
path, "string_test", PMEMOBJ_MIN_POOL, S_IWUSR | S_IRUSR);
auto r = pop.root();
{
try {
nvobj::transaction::run(pop, [&] {
r->s = nvobj::make_persistent<S>();
});
auto &s = *r->s;
test(pop, s);
s.assign(10, 'a');
s.erase(5);
test(pop, s);
s.assign(50, 'a');
s.erase(5);
test(pop, s);
s.assign(100, 'a');
s.erase(50);
test(pop, s);
s.assign(200, 'a');
s.erase(100);
test(pop, s);
nvobj::transaction::run(pop, [&] {
nvobj::delete_persistent<S>(r->s);
});
} catch (std::exception &e) {
UT_FATALexc(e);
}
}
pop.close();
}
int
main(int argc, char *argv[])
{
return run_test([&] { test(argc, argv); });
}
|
