forked from lthn/blockchain
101 lines
4.6 KiB
C++
101 lines
4.6 KiB
C++
// Copyright (c) 2014-2018 Zano Project
|
|
// Copyright (c) 2014-2018 The Louisdor Project
|
|
// 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"
|
|
|
|
struct wallet_test : virtual public test_chain_unit_enchanced
|
|
{
|
|
enum { MINER_ACC_IDX = 0, ALICE_ACC_IDX = 1, BOB_ACC_IDX = 2, CAROL_ACC_IDX = 3, DAN_ACC_IDX = 4, TOTAL_ACCS_COUNT = 5 }; // to be used as index for m_accounts
|
|
|
|
wallet_test();
|
|
|
|
bool need_core_proxy() const { return true; }
|
|
void set_core_proxy(std::shared_ptr<tools::i_core_proxy> p) { m_core_proxy = p; }
|
|
bool check_balance_via_build_wallets(currency::core& c, size_t ev_index, const std::vector<test_event_entry>& events);
|
|
bool check_balance(currency::core& c, size_t ev_index, const std::vector<test_event_entry>& events);
|
|
|
|
protected:
|
|
|
|
struct params_check_balance
|
|
{
|
|
params_check_balance(size_t account_index = 0,
|
|
uint64_t total_balance = 0,
|
|
uint64_t unlocked_balance = std::numeric_limits<uint64_t>::max(),
|
|
uint64_t mined_balance = std::numeric_limits<uint64_t>::max(),
|
|
uint64_t awaiting_in = std::numeric_limits<uint64_t>::max(),
|
|
uint64_t awaiting_out = std::numeric_limits<uint64_t>::max())
|
|
: account_index(account_index), total_balance(total_balance), unlocked_balance(unlocked_balance), mined_balance(mined_balance), awaiting_in(awaiting_in), awaiting_out(awaiting_out) {}
|
|
uint64_t total_balance;
|
|
uint64_t unlocked_balance;
|
|
uint64_t mined_balance;
|
|
uint64_t awaiting_in;
|
|
uint64_t awaiting_out;
|
|
size_t account_index;
|
|
};
|
|
|
|
std::shared_ptr<tools::wallet2> init_playtime_test_wallet(const std::vector<test_event_entry>& events, currency::core& c, size_t account_index) const;
|
|
std::shared_ptr<tools::wallet2> init_playtime_test_wallet(const std::vector<test_event_entry>& events, currency::core& c, const currency::account_base& acc) const;
|
|
|
|
mutable std::vector<currency::account_base> m_accounts;
|
|
mutable test_generator generator;
|
|
std::shared_ptr<tools::i_core_proxy> m_core_proxy;
|
|
};
|
|
|
|
// wallet callback helper to check balance in wallet callbacks
|
|
// see escrow_balance test for usage example
|
|
struct wallet_callback_balance_checker : public tools::i_wallet2_callback
|
|
{
|
|
wallet_callback_balance_checker(const std::string& label) : m_label(label), m_result(true), m_called(false), m_balance(UINT64_MAX), m_unlocked_balance(UINT64_MAX), m_total_mined(UINT64_MAX) {}
|
|
|
|
void expect_balance(uint64_t balance = UINT64_MAX, uint64_t unlocked_balance = UINT64_MAX, uint64_t total_mined = UINT64_MAX)
|
|
{
|
|
m_balance = balance;
|
|
m_unlocked_balance = unlocked_balance;
|
|
m_total_mined = total_mined;
|
|
m_called = false;
|
|
}
|
|
|
|
virtual void on_transfer2(const tools::wallet_public::wallet_transfer_info& wti, uint64_t balance, uint64_t unlocked_balance, uint64_t total_mined) override
|
|
{
|
|
m_called = true;
|
|
m_result = false;
|
|
CHECK_AND_ASSERT_MES(m_balance == UINT64_MAX || balance == m_balance, (void)(0), m_label << " balance is incorrect: " << currency::print_money_brief(balance) << ", expected: " << currency::print_money_brief(m_balance));
|
|
CHECK_AND_ASSERT_MES(m_unlocked_balance == UINT64_MAX || unlocked_balance == m_unlocked_balance, (void)(0), m_label << " unlocked balance is incorrect: " << currency::print_money_brief(unlocked_balance) << ", expected: " << currency::print_money_brief(m_unlocked_balance));
|
|
CHECK_AND_ASSERT_MES(m_total_mined == UINT64_MAX || total_mined == m_total_mined, (void)(0), m_label << " total mined is incorrect: " << currency::print_money_brief(total_mined) << ", expected: " << currency::print_money_brief(m_total_mined));
|
|
m_result = true;
|
|
}
|
|
|
|
bool check()
|
|
{
|
|
bool result = m_result;
|
|
m_result = false; // clear result to avoid errorneous successive calls to check() without calling except_balance()
|
|
return result;
|
|
}
|
|
|
|
bool called()
|
|
{
|
|
return m_called;
|
|
}
|
|
|
|
bool m_result;
|
|
bool m_called;
|
|
std::string m_label;
|
|
uint64_t m_balance;
|
|
uint64_t m_unlocked_balance;
|
|
uint64_t m_total_mined;
|
|
};
|
|
|
|
struct wlt_lambda_on_transfer2_wrapper : public tools::i_wallet2_callback
|
|
{
|
|
typedef std::function<bool(const tools::wallet_public::wallet_transfer_info&, uint64_t, uint64_t, uint64_t)> Func;
|
|
wlt_lambda_on_transfer2_wrapper(Func callback) : m_result(false), m_callback(callback) {}
|
|
virtual void on_transfer2(const tools::wallet_public::wallet_transfer_info& wti, uint64_t balance, uint64_t unlocked_balance, uint64_t total_mined) override
|
|
{
|
|
m_result = m_callback(wti, balance, unlocked_balance, total_mined);
|
|
}
|
|
bool m_result;
|
|
Func m_callback;
|
|
};
|