forked from lthn/blockchain
95 lines
4 KiB
C++
95 lines
4 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.
|
|
|
|
#include "chaingen.h"
|
|
#include "wallet_tests.h"
|
|
#include "wallet_test_core_proxy.h"
|
|
#include "../../src/wallet/wallet_public_structs_defs.h"
|
|
#include "offers_helper.h"
|
|
#include "string_coding.h"
|
|
|
|
using namespace epee;
|
|
using namespace crypto;
|
|
using namespace currency;
|
|
|
|
wallet_test::wallet_test()
|
|
: m_core_proxy(nullptr)
|
|
{
|
|
REGISTER_CALLBACK_METHOD(wallet_test, check_balance);
|
|
}
|
|
|
|
bool wallet_test::check_balance_via_build_wallets(currency::core& c, size_t ev_index, const std::vector<test_event_entry>& events)
|
|
{
|
|
// This is old version, it checks balance using test_generator::build_wallets(), which is using blockchain data from test_generator::m_blocks_info
|
|
params_check_balance pcb = AUTO_VAL_INIT(pcb);
|
|
bool r = epee::string_tools::hex_to_pod(boost::get<callback_entry>(events[ev_index]).callback_params, pcb);
|
|
CHECK_AND_ASSERT_MES(r, false, "check_balance: Can't obtain event params");
|
|
CHECK_AND_ASSERT_MES(pcb.account_index < m_accounts.size(), false, "check_balance: invalid account index");
|
|
|
|
const account_base& acc = m_accounts[pcb.account_index];
|
|
|
|
// obtain top block as most recent block
|
|
const block* top_block = 0;
|
|
size_t i = ev_index;
|
|
do
|
|
{
|
|
--i;
|
|
if (events[i].type() == typeid(block))
|
|
{
|
|
top_block = &boost::get<block>(events[i]);
|
|
break;
|
|
}
|
|
} while (i != 0);
|
|
CHECK_AND_ASSERT_MES(top_block != 0, false, "check_balance: failed to obtain top block from events queue");
|
|
|
|
std::list<account_base> accounts(1, acc);
|
|
test_generator::wallets_vector w;
|
|
r = generator.build_wallets(get_block_hash(*top_block), accounts, w, c.get_blockchain_storage().get_core_runtime_config());
|
|
CHECK_AND_ASSERT_MES(r && w.size() == 1 && w[0] != 0, false, "check_balance: failed to build wallets");
|
|
|
|
if (!check_balance_via_wallet(*w[0], epee::string_tools::num_to_string_fast(pcb.account_index).c_str(), pcb.total_balance, pcb.mined_balance, pcb.unlocked_balance, pcb.awaiting_in, pcb.awaiting_out))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
bool wallet_test::check_balance(currency::core& c, size_t ev_index, const std::vector<test_event_entry>& events)
|
|
{
|
|
// Newer version, uses the core with actual data via special proxy to sync wallets and get balance.
|
|
params_check_balance pcb = AUTO_VAL_INIT(pcb);
|
|
bool r = epee::string_tools::hex_to_pod(boost::get<callback_entry>(events[ev_index]).callback_params, pcb);
|
|
CHECK_AND_ASSERT_MES(r, false, "check_balance: Can't obtain event params");
|
|
|
|
std::shared_ptr<tools::wallet2> w = init_playtime_test_wallet(events, c, pcb.account_index);
|
|
w->refresh();
|
|
bool has_aliases = false;
|
|
w->scan_tx_pool(has_aliases);
|
|
|
|
if (!check_balance_via_wallet(*w.get(), epee::string_tools::num_to_string_fast(pcb.account_index).c_str(), pcb.total_balance, pcb.mined_balance, pcb.unlocked_balance, pcb.awaiting_in, pcb.awaiting_out))
|
|
return false;
|
|
|
|
return true;
|
|
}
|
|
|
|
std::shared_ptr<tools::wallet2> wallet_test::init_playtime_test_wallet(const std::vector<test_event_entry>& events, currency::core& c, const account_base& acc) const
|
|
{
|
|
CHECK_AND_ASSERT_THROW_MES(events.size() > 0 && events[0].type() == typeid(currency::block), "Invalid events queue, can't find genesis block at the beginning");
|
|
crypto::hash genesis_hash = get_block_hash(boost::get<block>(events[0]));
|
|
|
|
std::shared_ptr<tools::wallet2> w(new tools::wallet2);
|
|
w->set_core_runtime_config(c.get_blockchain_storage().get_core_runtime_config());
|
|
w->assign_account(acc);
|
|
w->set_genesis(genesis_hash);
|
|
w->set_core_proxy(m_core_proxy);
|
|
w->set_disable_tor_relay(true);
|
|
return w;
|
|
}
|
|
|
|
std::shared_ptr<tools::wallet2> wallet_test::init_playtime_test_wallet(const std::vector<test_event_entry>& events, currency::core& c, size_t account_index) const
|
|
{
|
|
CHECK_AND_ASSERT_THROW_MES(account_index < m_accounts.size(), "Invalid account index");
|
|
return init_playtime_test_wallet(events, c, m_accounts[account_index]);
|
|
}
|
|
|