115 lines
3.8 KiB
C++
115 lines
3.8 KiB
C++
#ifndef BITCOIN_TEST_TEST_BITCOIN_H
|
|
#define BITCOIN_TEST_TEST_BITCOIN_H
|
|
|
|
#include "chainparamsbase.h"
|
|
#include "consensus/upgrades.h"
|
|
#include "fs.h"
|
|
#include "key.h"
|
|
#include "pubkey.h"
|
|
#include "random.h"
|
|
#include "txdb.h"
|
|
|
|
#include <boost/thread.hpp>
|
|
|
|
extern uint256 insecure_rand_seed;
|
|
extern FastRandomContext insecure_rand_ctx;
|
|
|
|
static inline void SeedInsecureRand(bool fDeterministic = false)
|
|
{
|
|
if (fDeterministic) {
|
|
insecure_rand_seed = uint256();
|
|
} else {
|
|
insecure_rand_seed = GetRandHash();
|
|
}
|
|
insecure_rand_ctx = FastRandomContext(insecure_rand_seed);
|
|
}
|
|
|
|
static inline uint32_t InsecureRand32() { return insecure_rand_ctx.rand32(); }
|
|
static inline uint256 InsecureRand256() { return insecure_rand_ctx.rand256(); }
|
|
static inline uint64_t InsecureRandBits(int bits) { return insecure_rand_ctx.randbits(bits); }
|
|
static inline uint64_t InsecureRandRange(uint64_t range) { return insecure_rand_ctx.randrange(range); }
|
|
static inline bool InsecureRandBool() { return insecure_rand_ctx.randbool(); }
|
|
static inline std::vector<unsigned char> InsecureRandBytes(size_t len) { return insecure_rand_ctx.randbytes(len); }
|
|
|
|
/** Basic testing setup.
|
|
* This just configures logging and chain parameters.
|
|
*/
|
|
struct BasicTestingSetup {
|
|
ECCVerifyHandle globalVerifyHandle;
|
|
|
|
BasicTestingSetup(const std::string& chainName = CBaseChainParams::MAIN);
|
|
~BasicTestingSetup();
|
|
};
|
|
|
|
/** Testing setup that configures a complete environment.
|
|
* Included are data directory, coins database, script check threads setup.
|
|
*/
|
|
struct TestingSetup: public BasicTestingSetup {
|
|
CCoinsViewDB *pcoinsdbview;
|
|
fs::path orig_current_path;
|
|
fs::path pathTemp;
|
|
boost::thread_group threadGroup;
|
|
|
|
TestingSetup(const std::string& chainName = CBaseChainParams::MAIN);
|
|
~TestingSetup();
|
|
};
|
|
|
|
class CBlock;
|
|
struct CMutableTransaction;
|
|
class CScript;
|
|
|
|
#ifdef ENABLE_MINING
|
|
//
|
|
// Testing fixture that pre-creates a
|
|
// 100-block REGTEST-mode block chain
|
|
//
|
|
struct TestChain100Setup : public TestingSetup {
|
|
TestChain100Setup();
|
|
|
|
// Create a new block with just given transactions, coinbase paying to
|
|
// scriptPubKey, and try to add it to the current chain.
|
|
CBlock CreateAndProcessBlock(const std::vector<CMutableTransaction>& txns,
|
|
const CScript& scriptPubKey);
|
|
|
|
~TestChain100Setup();
|
|
|
|
std::vector<CTransaction> coinbaseTxns; // For convenience, coinbase transactions
|
|
CKey coinbaseKey; // private/public key needed to spend coinbase transactions
|
|
};
|
|
#endif // ENABLE_MINING
|
|
|
|
class CTxMemPoolEntry;
|
|
class CTxMemPool;
|
|
|
|
struct TestMemPoolEntryHelper
|
|
{
|
|
// Default values
|
|
CAmount nFee;
|
|
int64_t nTime;
|
|
unsigned int nHeight;
|
|
bool hadNoDependencies;
|
|
bool spendsCoinbase;
|
|
unsigned int sigOpCount;
|
|
uint32_t nBranchId;
|
|
|
|
TestMemPoolEntryHelper() :
|
|
nFee(0), nTime(0), nHeight(1),
|
|
hadNoDependencies(false), spendsCoinbase(false), sigOpCount(1),
|
|
nBranchId(SPROUT_BRANCH_ID) { }
|
|
|
|
CTxMemPoolEntry FromTx(CMutableTransaction &tx, CTxMemPool *pool = NULL);
|
|
|
|
// Change the default value
|
|
TestMemPoolEntryHelper &Fee(CAmount _fee) { nFee = _fee; return *this; }
|
|
TestMemPoolEntryHelper &Time(int64_t _time) { nTime = _time; return *this; }
|
|
TestMemPoolEntryHelper &Height(unsigned int _height) { nHeight = _height; return *this; }
|
|
TestMemPoolEntryHelper &HadNoDependencies(bool _hnd) { hadNoDependencies = _hnd; return *this; }
|
|
TestMemPoolEntryHelper &SpendsCoinbase(bool _flag) { spendsCoinbase = _flag; return *this; }
|
|
TestMemPoolEntryHelper &SigOps(unsigned int _sigops) { sigOpCount = _sigops; return *this; }
|
|
TestMemPoolEntryHelper &BranchId(uint32_t _branchId) { nBranchId = _branchId; return *this; }
|
|
};
|
|
|
|
void CheckRPCThrows(std::string rpcString, std::string expectedErrorMessage);
|
|
|
|
#endif
|