diff --git a/tests/core_tests/chaingen.cpp b/tests/core_tests/chaingen.cpp index f0406702..fbffae9f 100644 --- a/tests/core_tests/chaingen.cpp +++ b/tests/core_tests/chaingen.cpp @@ -21,6 +21,7 @@ #include "currency_core/bc_offers_service.h" #include "wallet/wallet2.h" #include "wallet_test_core_proxy.h" +#include "wallet_tests_basic.h" #include "pos_block_builder.h" using namespace epee; @@ -1002,10 +1003,7 @@ bool test_generator::init_test_wallet(const currency::account_base& account, con w->assign_account(account); w->set_genesis(genesis_hash); w->set_core_proxy(m_wallet_test_core_proxy); - w->set_disable_tor_relay(true); - w->set_concise_mode(true); - w->set_concise_mode_reorg_max_reorg_blocks(TESTS_CONCISE_MODE_REORG_MAX_REORG_BLOCK); - w->set_use_assets_whitelisting(false); + set_playtime_test_wallet_options(w); result = w; return true; diff --git a/tests/core_tests/wallet_tests_basic.h b/tests/core_tests/wallet_tests_basic.h index 5df3c6ce..8b71cc84 100644 --- a/tests/core_tests/wallet_tests_basic.h +++ b/tests/core_tests/wallet_tests_basic.h @@ -11,6 +11,15 @@ #include "currency_core/bc_offers_service.h" #include "rpc/core_rpc_server.h" +template +void set_playtime_test_wallet_options(std::shared_ptr w) +{ + w->set_disable_tor_relay(true); + w->set_concise_mode(true); + w->set_concise_mode_reorg_max_reorg_blocks(TESTS_CONCISE_MODE_REORG_MAX_REORG_BLOCK); + w->set_use_assets_whitelisting(false); +} + 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 @@ -59,10 +68,7 @@ struct wallet_test : virtual public test_chain_unit_enchanced w->assign_account(acc); w->set_genesis(genesis_hash); w->set_core_proxy(m_core_proxy); - w->set_disable_tor_relay(true); - w->set_concise_mode(true); - w->set_concise_mode_reorg_max_reorg_blocks(TESTS_CONCISE_MODE_REORG_MAX_REORG_BLOCK); - w->set_use_assets_whitelisting(false); + set_playtime_test_wallet_options(w); return w; #undef LOCAL_HOST_CSTR