danicoin/tests/CoreTests/TestGenerator.h
2016-01-18 15:33:29 +00:00

110 lines
3.3 KiB
C++

// Copyright (c) 2011-2016 The Cryptonote developers
// Distributed under the MIT/X11 software license, see the accompanying
// file COPYING or http://www.opensource.org/licenses/mit-license.php.
#pragma once
#include "Chaingen.h"
#include "CryptoNoteCore/Currency.h"
#include "TransactionBuilder.h"
#include <Logging/LoggerGroup.h>
class TestGenerator {
public:
TestGenerator(
const CryptoNote::Currency& currency,
std::vector<test_event_entry>& eventsRef) :
generator(currency),
events(eventsRef) {
minerAccount.generate();
generator.constructBlock(genesisBlock, minerAccount, 1338224400);
events.push_back(genesisBlock);
lastBlock = genesisBlock;
}
const CryptoNote::Currency& currency() const { return generator.currency(); }
void makeNextBlock(const std::list<CryptoNote::Transaction>& txs = std::list<CryptoNote::Transaction>()) {
CryptoNote::Block block;
generator.constructBlock(block, lastBlock, minerAccount, txs);
events.push_back(block);
lastBlock = block;
}
void makeNextBlock(const CryptoNote::Transaction& tx) {
std::list<CryptoNote::Transaction> txs;
txs.push_back(tx);
makeNextBlock(txs);
}
void generateBlocks() {
generateBlocks(currency().minedMoneyUnlockWindow());
}
void generateBlocks(size_t count, uint8_t majorVersion = CryptoNote::BLOCK_MAJOR_VERSION_1) {
while (count--) {
CryptoNote::Block next;
generator.constructBlockManually(next, lastBlock, minerAccount, test_generator::bf_major_ver, majorVersion);
lastBlock = next;
events.push_back(next);
}
}
TransactionBuilder createTxBuilder(const CryptoNote::AccountBase& from, const CryptoNote::AccountBase& to, uint64_t amount, uint64_t fee) {
std::vector<CryptoNote::TransactionSourceEntry> sources;
std::vector<CryptoNote::TransactionDestinationEntry> destinations;
fillTxSourcesAndDestinations(sources, destinations, from, to, amount, fee);
TransactionBuilder builder(generator.currency());
builder.setInput(sources, from.getAccountKeys());
builder.setOutput(destinations);
return builder;
}
void fillTxSourcesAndDestinations(
std::vector<CryptoNote::TransactionSourceEntry>& sources,
std::vector<CryptoNote::TransactionDestinationEntry>& destinations,
const CryptoNote::AccountBase& from, const CryptoNote::AccountBase& to, uint64_t amount, uint64_t fee, size_t nmix = 0) {
fill_tx_sources_and_destinations(events, lastBlock, from, to, amount, fee, nmix, sources, destinations);
}
void constructTxToKey(
CryptoNote::Transaction& tx,
const CryptoNote::AccountBase& from,
const CryptoNote::AccountBase& to,
uint64_t amount,
uint64_t fee,
size_t nmix = 0) {
construct_tx_to_key(logger, events, tx, lastBlock, from, to, amount, fee, nmix);
}
void addEvent(const test_event_entry& e) {
events.push_back(e);
}
void addCallback(const std::string& name) {
callback_entry cb;
cb.callback_name = name;
events.push_back(cb);
}
void addCheckAccepted() {
addCallback("check_block_accepted");
}
void addCheckPurged() {
addCallback("check_block_purged");
}
Logging::LoggerGroup logger;
test_generator generator;
CryptoNote::Block genesisBlock;
CryptoNote::Block lastBlock;
CryptoNote::AccountBase minerAccount;
std::vector<test_event_entry>& events;
};