247 lines
6.6 KiB
C++
247 lines
6.6 KiB
C++
// Copyright (C) 2013 Red Hat, Inc. All rights reserved.
|
|
//
|
|
// This file is part of the thin-provisioning-tools source.
|
|
//
|
|
// thin-provisioning-tools is free software: you can redistribute it
|
|
// and/or modify it under the terms of the GNU General Public License
|
|
// as published by the Free Software Foundation, either version 3 of
|
|
// the License, or (at your option) any later version.
|
|
//
|
|
// thin-provisioning-tools is distributed in the hope that it will be
|
|
// useful, but WITHOUT ANY WARRANTY; without even the implied warranty
|
|
// of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
// GNU General Public License for more details.
|
|
//
|
|
// You should have received a copy of the GNU General Public License along
|
|
// with thin-provisioning-tools. If not, see
|
|
// <http://www.gnu.org/licenses/>.
|
|
|
|
#include "persistent-data/transaction_manager.h"
|
|
#include "persistent-data/space-maps/core.h"
|
|
#include "persistent-data/data-structures/array_block.h"
|
|
#include "test_utils.h"
|
|
|
|
#include <utility>
|
|
#include <vector>
|
|
|
|
#define BOOST_TEST_MODULE ArrayBlockTests
|
|
#include <boost/test/included/unit_test.hpp>
|
|
|
|
using namespace boost;
|
|
using namespace persistent_data;
|
|
using namespace std;
|
|
using namespace test;
|
|
|
|
//----------------------------------------------------------------
|
|
|
|
namespace {
|
|
uint64_t MAX_VALUE = 1000ull;
|
|
block_address const NR_BLOCKS = 1024;
|
|
typedef typename block_manager<>::noop_validator noop_validator;
|
|
typedef typename block_manager<>::read_ref read_ref;
|
|
typedef typename block_manager<>::write_ref write_ref;
|
|
|
|
// FIXME: lift to utils?
|
|
class simple_ref_counter {
|
|
public:
|
|
simple_ref_counter(uint64_t nr_counts)
|
|
: counts_(nr_counts, 0u) {
|
|
}
|
|
|
|
void inc(uint64_t n) {
|
|
counts_.at(n)++;
|
|
}
|
|
|
|
void dec(uint64_t n) {
|
|
counts_.at(n)--;
|
|
}
|
|
|
|
unsigned get(uint64_t n) const {
|
|
return counts_.at(n);
|
|
}
|
|
|
|
private:
|
|
vector<unsigned> counts_;
|
|
};
|
|
|
|
struct uint64_traits {
|
|
typedef base::__le64 disk_type;
|
|
typedef uint64_t value_type;
|
|
typedef simple_ref_counter ref_counter;
|
|
|
|
static void unpack(disk_type const &disk, value_type &value) {
|
|
value = base::to_cpu<uint64_t>(disk);
|
|
}
|
|
|
|
static void pack(value_type const &value, disk_type &disk) {
|
|
disk = base::to_disk<base::__le64>(value);
|
|
}
|
|
};
|
|
|
|
typedef array_block<uint64_traits, write_ref> ablock64;
|
|
typedef array_block<uint64_traits, read_ref> ablock64_r;
|
|
|
|
block_manager<>::validator::ptr
|
|
validator() {
|
|
return block_manager<>::validator::ptr(new block_manager<>::noop_validator);
|
|
}
|
|
|
|
transaction_manager::ptr
|
|
create_tm() {
|
|
block_manager<>::ptr bm = create_bm<4096>(NR_BLOCKS);
|
|
space_map::ptr sm(new core_map(NR_BLOCKS));
|
|
transaction_manager::ptr tm(new transaction_manager(bm, sm));
|
|
return tm;
|
|
}
|
|
|
|
pair<ablock64, block_address>
|
|
new_array_block(transaction_manager::ptr tm) {
|
|
uint64_traits::ref_counter rc(MAX_VALUE);
|
|
write_ref wr = tm->new_block(validator());
|
|
return make_pair(ablock64(wr, rc, sizeof(uint64_t)), wr.get_location());
|
|
}
|
|
|
|
ablock64
|
|
open_array_block(transaction_manager::ptr tm, block_address loc) {
|
|
uint64_traits::ref_counter rc(MAX_VALUE);
|
|
pair<write_ref, bool> p = tm->shadow(loc, validator());
|
|
BOOST_CHECK(!p.second);
|
|
return ablock64(p.first, rc);
|
|
}
|
|
|
|
ablock64_r
|
|
read_array_block(transaction_manager::ptr tm, block_address loc) {
|
|
uint64_traits::ref_counter rc(MAX_VALUE);
|
|
read_ref rr = tm->read_lock(loc, validator());
|
|
return ablock64_r(rr, rc);
|
|
}
|
|
}
|
|
|
|
//----------------------------------------------------------------
|
|
|
|
BOOST_AUTO_TEST_CASE(can_create_an_empty_array)
|
|
{
|
|
block_address loc;
|
|
transaction_manager::ptr tm = create_tm();
|
|
|
|
{
|
|
pair<ablock64, block_address> p = new_array_block(tm);
|
|
ablock64 &b = p.first;
|
|
loc = p.second;
|
|
|
|
BOOST_CHECK_EQUAL(b.nr_entries(), 0);
|
|
BOOST_CHECK_EQUAL(b.value_size(), sizeof(uint64_t));
|
|
BOOST_CHECK_EQUAL(b.max_entries(), (4096 - 24) / 8);
|
|
|
|
BOOST_CHECK_THROW(b.get(0), runtime_error);
|
|
BOOST_CHECK_THROW(b.set(0, 12345LL), runtime_error);
|
|
}
|
|
|
|
{
|
|
ablock64 b = open_array_block(tm, loc);
|
|
|
|
BOOST_CHECK_EQUAL(b.nr_entries(), 0);
|
|
BOOST_CHECK_EQUAL(b.value_size(), sizeof(uint64_t));
|
|
BOOST_CHECK_EQUAL(b.max_entries(), (4096 - 24) / 8);
|
|
|
|
BOOST_CHECK_THROW(b.get(0), runtime_error);
|
|
BOOST_CHECK_THROW(b.set(0, 12345LL), runtime_error);
|
|
}
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(read_only_array_blocks_are_possible)
|
|
{
|
|
block_address loc;
|
|
transaction_manager::ptr tm = create_tm();
|
|
|
|
{
|
|
pair<ablock64, block_address> p = new_array_block(tm);
|
|
ablock64 &b = p.first;
|
|
loc = p.second;
|
|
|
|
BOOST_CHECK_EQUAL(b.nr_entries(), 0);
|
|
BOOST_CHECK_EQUAL(b.value_size(), sizeof(uint64_t));
|
|
BOOST_CHECK_EQUAL(b.max_entries(), (4096 - 24) / 8);
|
|
|
|
BOOST_CHECK_THROW(b.get(0), runtime_error);
|
|
BOOST_CHECK_THROW(b.set(0, 12345LL), runtime_error);
|
|
}
|
|
|
|
{
|
|
ablock64_r b = read_array_block(tm, loc);
|
|
|
|
BOOST_CHECK_EQUAL(b.nr_entries(), 0);
|
|
BOOST_CHECK_EQUAL(b.value_size(), sizeof(uint64_t));
|
|
BOOST_CHECK_EQUAL(b.max_entries(), (4096 - 24) / 8);
|
|
|
|
BOOST_CHECK_THROW(b.get(0), runtime_error);
|
|
|
|
// Compile time error as expected
|
|
// BOOST_CHECK_THROW(b.set(0, 12345LL), runtime_error);
|
|
}
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(growing)
|
|
{
|
|
uint64_t default_value = 123, new_value = 234;
|
|
transaction_manager::ptr tm = create_tm();
|
|
pair<ablock64, block_address> p = new_array_block(tm);
|
|
ablock64 &b = p.first;
|
|
|
|
for (unsigned i = 1; i < b.max_entries(); i++) {
|
|
BOOST_CHECK_THROW(b.get(i - 1), runtime_error);
|
|
|
|
b.grow(i, default_value);
|
|
BOOST_CHECK_EQUAL(b.get(i - 1), default_value);
|
|
|
|
b.set(i - 1, new_value);
|
|
BOOST_CHECK_EQUAL(b.get(i - 1), new_value);
|
|
|
|
BOOST_CHECK_THROW(b.grow(i - 1, default_value), runtime_error);
|
|
}
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(shrinking)
|
|
{
|
|
uint64_t default_value = 123;
|
|
transaction_manager::ptr tm = create_tm();
|
|
pair<ablock64, block_address> p = new_array_block(tm);
|
|
ablock64 &b = p.first;
|
|
|
|
b.grow(b.max_entries() - 1, default_value);
|
|
for (unsigned i = b.max_entries() - 2; i; i--) {
|
|
BOOST_CHECK_EQUAL(b.get(i - 1), default_value);
|
|
b.shrink(i);
|
|
BOOST_CHECK_THROW(b.get(i), runtime_error);
|
|
BOOST_CHECK_THROW(b.shrink(i), runtime_error);
|
|
}
|
|
}
|
|
|
|
BOOST_AUTO_TEST_CASE(ref_counting)
|
|
{
|
|
transaction_manager::ptr tm = create_tm();
|
|
pair<ablock64, block_address> p = new_array_block(tm);
|
|
ablock64 &b = p.first;
|
|
simple_ref_counter const &rc = b.get_ref_counter();
|
|
|
|
BOOST_CHECK_EQUAL(rc.get(123), 0);
|
|
b.grow(b.max_entries() - 1, 123);
|
|
BOOST_CHECK_EQUAL(rc.get(123), b.max_entries() - 1);
|
|
|
|
b.shrink(100);
|
|
BOOST_CHECK_EQUAL(rc.get(123), 100);
|
|
|
|
b.set(1, 0);
|
|
b.set(2, 2);
|
|
BOOST_CHECK_EQUAL(rc.get(123), 98);
|
|
BOOST_CHECK_EQUAL(rc.get(0), 1);
|
|
|
|
b.set(2, 2);
|
|
BOOST_CHECK_EQUAL(rc.get(2), 1);
|
|
b.set(10, 2);
|
|
BOOST_CHECK_EQUAL(rc.get(2), 2);
|
|
BOOST_CHECK_EQUAL(rc.get(123), 97);
|
|
}
|
|
|
|
//----------------------------------------------------------------
|