139 lines
3.8 KiB
C++
139 lines
3.8 KiB
C++
// Copyright (C) 2011 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 "gmock/gmock.h"
|
|
#include "persistent-data/transaction_manager.h"
|
|
#include "persistent-data/space-maps/core.h"
|
|
|
|
using namespace std;
|
|
using namespace persistent_data;
|
|
using namespace testing;
|
|
|
|
//----------------------------------------------------------------
|
|
|
|
namespace {
|
|
unsigned const MAX_HELD_LOCKS = 16;
|
|
block_address const NR_BLOCKS = 1024;
|
|
|
|
transaction_manager::ptr
|
|
create_tm() {
|
|
block_manager<>::ptr bm(
|
|
new block_manager<>("./test.data", NR_BLOCKS, MAX_HELD_LOCKS, block_manager<>::READ_WRITE));
|
|
space_map::ptr sm(new core_map(NR_BLOCKS));
|
|
transaction_manager::ptr tm(new transaction_manager(bm, sm));
|
|
tm->get_sm()->inc(0);
|
|
return tm;
|
|
}
|
|
|
|
typedef bcache::validator::ptr validator_ptr;
|
|
|
|
validator_ptr mk_noop_validator() {
|
|
return bcache::validator::ptr(
|
|
new bcache::noop_validator);
|
|
}
|
|
|
|
typedef block_manager<>::write_ref write_ref;
|
|
}
|
|
|
|
//----------------------------------------------------------------
|
|
|
|
TEST(TransactionManagerTests, commit_succeeds)
|
|
{
|
|
transaction_manager::ptr tm = create_tm();
|
|
tm->begin(0, mk_noop_validator());
|
|
}
|
|
|
|
TEST(TransactionManagerTests, shadowing)
|
|
{
|
|
transaction_manager::ptr tm = create_tm();
|
|
block_manager<>::write_ref superblock = tm->begin(0, mk_noop_validator());
|
|
|
|
space_map::ptr sm = tm->get_sm();
|
|
sm->inc(1);
|
|
block_address b;
|
|
|
|
{
|
|
pair<write_ref, bool> p = tm->shadow(1, mk_noop_validator());
|
|
b = p.first.get_location();
|
|
ASSERT_THAT(b, Ne(1u));
|
|
ASSERT_FALSE(p.second);
|
|
ASSERT_THAT(sm->get_count(1), Eq(0u));
|
|
}
|
|
|
|
{
|
|
pair<write_ref, bool> p = tm->shadow(b, mk_noop_validator());
|
|
ASSERT_THAT(p.first.get_location(), Eq(b));
|
|
ASSERT_FALSE(p.second);
|
|
}
|
|
|
|
sm->inc(b);
|
|
|
|
{
|
|
pair<write_ref, bool> p = tm->shadow(b, mk_noop_validator());
|
|
ASSERT_THAT(p.first.get_location(), Ne(b));
|
|
ASSERT_TRUE(p.second);
|
|
}
|
|
}
|
|
|
|
TEST(TransactionManagerTests, multiple_shadowing)
|
|
{
|
|
transaction_manager::ptr tm = create_tm();
|
|
space_map::ptr sm = tm->get_sm();
|
|
sm->set_count(1, 3);
|
|
block_address b, b2;
|
|
|
|
{
|
|
write_ref superblock = tm->begin(0, mk_noop_validator());
|
|
pair<write_ref, bool> p = tm->shadow(1, mk_noop_validator());
|
|
b = p.first.get_location();
|
|
ASSERT_THAT(b, Ne(1u));
|
|
ASSERT_TRUE(p.second);
|
|
sm->commit();
|
|
}
|
|
|
|
{
|
|
write_ref superblock = tm->begin(0, mk_noop_validator());
|
|
pair<write_ref, bool> p = tm->shadow(1, mk_noop_validator());
|
|
b2 = p.first.get_location();
|
|
ASSERT_THAT(b2, Ne(1u));
|
|
ASSERT_THAT(b2, Ne(b));
|
|
ASSERT_TRUE(p.second);
|
|
sm->commit();
|
|
}
|
|
|
|
{
|
|
write_ref superblock = tm->begin(0, mk_noop_validator());
|
|
pair<write_ref, bool> p = tm->shadow(1, mk_noop_validator());
|
|
block_address b3 = p.first.get_location();
|
|
ASSERT_THAT(b3, Ne(b2));
|
|
ASSERT_THAT(b3, Ne(b));
|
|
ASSERT_THAT(b3, Ne(1u));
|
|
ASSERT_FALSE(p.second);
|
|
sm->commit();
|
|
}
|
|
}
|
|
|
|
TEST(TransactionManagerTests, shadow_free_block_fails)
|
|
{
|
|
transaction_manager::ptr tm = create_tm();
|
|
write_ref superblock = tm->begin(0, mk_noop_validator());
|
|
ASSERT_THROW(tm->shadow(1, mk_noop_validator()), runtime_error);
|
|
}
|
|
|
|
//----------------------------------------------------------------
|