emo/thin-provisioning-tools
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

BTreeDamageVisitorTests

Browse Source
This commit is contained in:
Joe Thornber 2013-05-14 11:27:55 +01:00
parent 7e8e483626
commit 7d49496c3b
1 changed files with 28 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

56
unit-tests/btree_damage_visitor_t.cc
View File

@ -227,11 +227,20 @@ namespace {
return *nodes_; return *nodes_;
} }
unsigned nth_node(node_array const &nodes, unsigned target, bool leaf) const { static bool is_leaf(node_info::ptr n) {
return n->leaf;
}
static bool is_internal(node_info::ptr n) {
return !n->leaf;
}
template <typename Predicate>
unsigned nth_node(node_array const &nodes, unsigned target, Predicate const &pred) const {
unsigned i; unsigned i;
for (i = 0; i < nodes.size(); i++) for (i = 0; i < nodes.size(); i++)
if (nodes[i]->leaf == leaf) { if (pred(nodes[i])) {
if (!target) if (!target)
break; break;
else else
@ -244,42 +253,33 @@ namespace {
return i; return i;
} }
unsigned get_nr_leaf_nodes() { template <typename Predicate>
unsigned get_nr_nodes(Predicate const &pred) {
node_array const &nodes = get_nodes(); node_array const &nodes = get_nodes();
unsigned nr_leaf = 0; unsigned nr = 0;
for (unsigned i = 0; i < nodes.size(); i++) for (unsigned i = 0; i < nodes.size(); i++)
if (nodes[i]->leaf) if (pred(nodes[i]))
nr_leaf++; nr++;
return nr_leaf; return nr;
}
// FIXME: remove duplication
unsigned get_nr_internal_nodes() {
node_array const &nodes = get_nodes();
unsigned nr_internal = 0;
for (unsigned i = 0; i < nodes.size(); i++)
if (!nodes[i]->leaf)
nr_internal++;
return nr_internal;
} }
node_info::ptr get_leaf_node(unsigned index) { node_info::ptr get_leaf_node(unsigned index) {
node_array const &nodes = get_nodes(); node_array const &nodes = get_nodes();
unsigned ni = nth_node(nodes, index, true);
unsigned ni = nth_node(nodes, index, is_leaf);
return nodes[ni]; return nodes[ni];
} }
node_info::ptr random_leaf_node() { node_info::ptr random_leaf_node() {
node_array const &nodes = get_nodes(); node_array const &nodes = get_nodes();
unsigned nr_leaf = get_nr_leaf_nodes(); unsigned nr_leaf = get_nr_nodes(is_leaf);
unsigned target = random() % nr_leaf; unsigned target = random() % nr_leaf;
unsigned i = nth_node(nodes, target, true);
unsigned i = nth_node(nodes, target, is_leaf);
return nodes[i]; return nodes[i];
} }
@ -287,9 +287,9 @@ namespace {
node_info::ptr random_internal_node() { node_info::ptr random_internal_node() {
node_array const &nodes = get_nodes(); node_array const &nodes = get_nodes();
unsigned nr_internal = get_nr_internal_nodes(); unsigned nr_internal = get_nr_nodes(is_internal);
unsigned target = random() % nr_internal; unsigned target = random() % nr_internal;
unsigned i = nth_node(nodes, target, false); unsigned i = nth_node(nodes, target, is_internal);
return nodes[i]; return nodes[i];
} }
@ -297,9 +297,9 @@ namespace {
node_array get_random_leaf_nodes(unsigned count) { node_array get_random_leaf_nodes(unsigned count) {
node_array const &nodes = get_nodes(); node_array const &nodes = get_nodes();
unsigned nr_leaf = get_nr_leaf_nodes(); unsigned nr_leaf = get_nr_nodes(is_leaf);
unsigned target = random() % (nr_leaf - count); unsigned target = random() % (nr_leaf - count);
unsigned i = nth_node(nodes, target, true); unsigned i = nth_node(nodes, target, is_leaf);
node_array v; node_array v;
@ -432,7 +432,7 @@ TEST_F(BTreeDamageVisitorTests, damaged_last_leaf)
insert_values(10000); insert_values(10000);
commit(); commit();
node_info::ptr n = get_leaf_node(get_nr_leaf_nodes() - 1); node_info::ptr n = get_leaf_node(get_nr_nodes(is_leaf) - 1);
block_address begin = *n->keys.begin_; block_address begin = *n->keys.begin_;
trash_block(n->b); trash_block(n->b);