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
|
// -*- mode:C++; tab-width:8; c-basic-offset:2; indent-tabs-mode:t -*-
// vim: ts=8 sw=2 smarttab
#include "cls/rbd/cls_rbd_client.h"
#include "cls/rbd/cls_rbd_types.h"
#include "test/librbd/test_fixture.h"
#include "test/librbd/test_support.h"
#include "librbd/api/Trash.h"
#include <set>
#include <vector>
void register_test_trash() {
}
namespace librbd {
static bool operator==(const trash_image_info_t& lhs,
const trash_image_info_t& rhs) {
return (lhs.id == rhs.id &&
lhs.name == rhs.name &&
lhs.source == rhs.source);
}
static bool operator==(const image_spec_t& lhs,
const image_spec_t& rhs) {
return (lhs.id == rhs.id && lhs.name == rhs.name);
}
class TestTrash : public TestFixture {
public:
TestTrash() {}
};
TEST_F(TestTrash, UserRemovingSource) {
REQUIRE_FORMAT_V2();
auto compare_lambda = [](const trash_image_info_t& lhs,
const trash_image_info_t& rhs) {
if (lhs.id != rhs.id) {
return lhs.id < rhs.id;
} else if (lhs.name != rhs.name) {
return lhs.name < rhs.name;
}
return lhs.source < rhs.source;
};
typedef std::set<trash_image_info_t, decltype(compare_lambda)> TrashEntries;
librbd::RBD rbd;
librbd::Image image;
auto image_name1 = m_image_name;
std::string image_id1;
ASSERT_EQ(0, rbd.open(m_ioctx, image, image_name1.c_str()));
ASSERT_EQ(0, image.get_id(&image_id1));
ASSERT_EQ(0, image.close());
auto image_name2 = get_temp_image_name();
ASSERT_EQ(0, create_image_pp(m_rbd, m_ioctx, image_name2, m_image_size));
std::string image_id2;
ASSERT_EQ(0, rbd.open(m_ioctx, image, image_name2.c_str()));
ASSERT_EQ(0, image.get_id(&image_id2));
ASSERT_EQ(0, image.close());
ASSERT_EQ(0, api::Trash<>::move(m_ioctx, RBD_TRASH_IMAGE_SOURCE_USER,
image_name1, image_id1, 0));
ASSERT_EQ(0, api::Trash<>::move(m_ioctx, RBD_TRASH_IMAGE_SOURCE_REMOVING,
image_name2, image_id2, 0));
TrashEntries trash_entries{compare_lambda};
TrashEntries expected_trash_entries{compare_lambda};
std::vector<trash_image_info_t> entries;
ASSERT_EQ(0, api::Trash<>::list(m_ioctx, entries, true));
trash_entries.insert(entries.begin(), entries.end());
expected_trash_entries = {
{.id = image_id1,
.name = image_name1,
.source = RBD_TRASH_IMAGE_SOURCE_USER},
};
ASSERT_EQ(expected_trash_entries, trash_entries);
std::vector<image_spec_t> expected_images = {
{.id = image_id2, .name = image_name2}
};
std::vector<image_spec_t> images;
ASSERT_EQ(0, rbd.list2(m_ioctx, &images));
ASSERT_EQ(expected_images, images);
}
TEST_F(TestTrash, RestoreMirroringSource) {
REQUIRE_FORMAT_V2();
librbd::RBD rbd;
librbd::Image image;
std::string image_id;
ASSERT_EQ(0, rbd.open(m_ioctx, image, m_image_name.c_str()));
ASSERT_EQ(0, image.get_id(&image_id));
ASSERT_EQ(0, image.close());
ASSERT_EQ(0, api::Trash<>::move(m_ioctx, RBD_TRASH_IMAGE_SOURCE_MIRRORING,
m_image_name, 0));
ASSERT_EQ(0, rbd.trash_restore(m_ioctx, image_id.c_str(),
m_image_name.c_str()));
}
} // namespace librbd
|
