Skip to content

Commit

Permalink
test code cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
@jmjatlanta
jmjatlanta committed Feb 19, 2018
1 parent 3224744 commit 2cc12d9
Showing 1 changed file with 76 additions and 92 deletions.
168 changes: 76 additions & 92 deletions tests/cli/main.cpp
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,9 +24,6 @@
#include <graphene/app/application.hpp>
#include <graphene/app/plugin.hpp>

#include <graphene/chain/balance_object.hpp>
#include <graphene/chain/protocol/transaction.hpp>

#include <graphene/utilities/tempdir.hpp>

#include <graphene/account_history/account_history_plugin.hpp>
Expand All @@ -46,7 +43,9 @@
#define BOOST_TEST_MODULE Test Application
#include <boost/test/included/unit_test.hpp>

using namespace graphene;
/*********************
* Helper Methods
*********************/

// hack: import create_example_genesis() even though it's a way, way
// specific internal detail
Expand All @@ -63,68 +62,83 @@ boost::filesystem::path create_genesis_file(fc::temp_directory& directory) {
return genesis_path;
}

/***
* @brief Start the application, listening on port 8090
* @param app_dir the temporary directory to use
* @returns the application object
*/
std::shared_ptr<graphene::app::application> start_application(fc::temp_directory& app_dir) {
std::shared_ptr<graphene::app::application> app1(new graphene::app::application{});

app1->register_plugin<graphene::account_history::account_history_plugin>();
app1->register_plugin< graphene::market_history::market_history_plugin >();
app1->register_plugin< graphene::witness_plugin::witness_plugin >();
app1->startup_plugins();
boost::program_options::variables_map cfg;
cfg.emplace("rpc-endpoint", boost::program_options::variable_value(string("127.0.0.1:8090"), false));
cfg.emplace("genesis-json", boost::program_options::variable_value(create_genesis_file(app_dir), false));
app1->initialize(app_dir.path(), cfg);

app1->startup();
fc::usleep(fc::milliseconds(500));
return app1;
}

/****
* Send a block to the db
* @param app the application
* @returns true on success
*/
bool generate_block(graphene::app::application& app) {
try {
fc::ecc::private_key committee_key = fc::ecc::private_key::regenerate(fc::sha256::hash(string("nathan")));
auto db = app.chain_database();
auto block_1 = db->generate_block(
db->get_slot_time(1),
db->get_scheduled_witness(1),
committee_key,
database::skip_nothing);
return true;
} catch (exception &e) {
return false;
}
}

/****************************
* Tests
****************************/

/**
* Start a server and connect using the same calls as the CLI
*/
BOOST_AUTO_TEST_CASE( cli_connect )
{
using namespace graphene::chain;
using namespace graphene::app;
graphene::app::application app1;
std::shared_ptr<graphene::app::application> app1;
try {
BOOST_TEST_MESSAGE( "Creating temporary files" );

fc::temp_directory app_dir( graphene::utilities::temp_directory_path() );

BOOST_TEST_MESSAGE( "Creating and initializing app1" );
fc::temp_directory app_dir ( graphene::utilities::temp_directory_path() );

app1.register_plugin<graphene::account_history::account_history_plugin>();
app1.register_plugin< graphene::market_history::market_history_plugin >();
app1.register_plugin< graphene::witness_plugin::witness_plugin >();
app1.startup_plugins();
boost::program_options::variables_map cfg;
cfg.emplace("rpc-endpoint", boost::program_options::variable_value(string("127.0.0.1:8090"), false));
cfg.emplace("genesis-json", boost::program_options::variable_value(create_genesis_file(app_dir), false));
app1.initialize(app_dir.path(), cfg);

BOOST_TEST_MESSAGE( "Starting app1 and waiting 500 ms" );
app1.startup();
fc::usleep(fc::milliseconds(500));
app1 = start_application(app_dir);

// connect to the server
graphene::wallet::wallet_data wdata;
wdata.chain_id = app1.chain_database()->get_chain_id();
wdata.chain_id = app1->chain_database()->get_chain_id();
wdata.ws_server = "ws://127.0.0.1:8090";
wdata.ws_user = "";
wdata.ws_password = "";
fc::http::websocket_client client;
auto con = client.connect( wdata.ws_server );

auto apic = std::make_shared<fc::rpc::websocket_api_connection>(*con);
auto apic = std::make_shared<fc::rpc::websocket_api_connection>(*(client.connect(wdata.ws_server)));

auto remote_api = apic->get_remote_api< login_api >(1);
BOOST_CHECK(remote_api->login( wdata.ws_user, wdata.ws_password ) );

/*
fc::mutable_variant_object settings = remote_api->get_server_information();
std::cout << "Server Version: " << settings["server_version"].as<std::string>() << std::endl;
std::cout << "Server SHA Version: " << settings["server_sha_version"].as<std::string>() << std::endl;
std::cout << "Server Version Timestamp: " << settings["server_version_timestamp"].as<std::string>() << std::endl;
std::cout << "SSL Version: " << settings["ssl_version"].as<std::string>() << std::endl;
std::cout << "Boost Version: " << settings["boost_version"].as<std::string>() << std::endl;
std::cout << "Websocket++ Version: " << settings["websocket_version"].as<std::string>() << std::endl;
BOOST_CHECK(settings["server_version"].as<std::string>() != "");
BOOST_CHECK(settings["server_sha_version"].as<std::string>() != "");
BOOST_CHECK(settings["server_version_timestamp"].as<std::string>() != "");
BOOST_CHECK(settings["ssl_version"].as<std::string>() != "");
BOOST_CHECK(settings["boost_version"].as<std::string>() != "");
BOOST_CHECK(settings["websocket_version"].as<std::string>() != "");
*/
} catch( fc::exception& e ) {
edump((e.to_detail_string()));
throw;
}
app1.shutdown();
app1->shutdown();
}

/**
Expand All @@ -134,30 +148,16 @@ BOOST_AUTO_TEST_CASE( cli_vote_for_2_witnesses )
{
using namespace graphene::chain;
using namespace graphene::app;
graphene::app::application app1;
std::shared_ptr<graphene::app::application> app1;
try {
BOOST_TEST_MESSAGE( "Creating temporary files" );

fc::temp_directory app_dir( graphene::utilities::temp_directory_path() );

BOOST_TEST_MESSAGE( "Creating and initializing app1" );

app1.register_plugin<graphene::account_history::account_history_plugin>();
app1.register_plugin< graphene::market_history::market_history_plugin >();
app1.register_plugin< graphene::witness_plugin::witness_plugin >();
app1.startup_plugins();
boost::program_options::variables_map cfg;
cfg.emplace("rpc-endpoint", boost::program_options::variable_value(string("127.0.0.1:8090"), false));
cfg.emplace("genesis-json", boost::program_options::variable_value(create_genesis_file(app_dir), false));
app1.initialize(app_dir.path(), cfg);

BOOST_TEST_MESSAGE( "Starting app1 and waiting 500 ms" );
app1.startup();
fc::usleep(fc::milliseconds(500));
app1 = start_application(app_dir);

// connect to the server
graphene::wallet::wallet_data wdata;
wdata.chain_id = app1.chain_database()->get_chain_id();
wdata.chain_id = app1->chain_database()->get_chain_id();
wdata.ws_server = "ws://127.0.0.1:8090";
wdata.ws_user = "";
wdata.ws_password = "";
Expand Down Expand Up @@ -190,97 +190,81 @@ BOOST_AUTO_TEST_CASE( cli_vote_for_2_witnesses )
wapiptr->set_password("supersecret");
wapiptr->unlock("supersecret");

BOOST_TEST_MESSAGE("Importing nathan key");
// import Nathan account
BOOST_TEST_MESSAGE("Importing nathan key");
std::vector<std::string> nathan_keys{"5KQwrPbwdL6PhXujxW37FSSQZ1JiwsST4cqQzDeyXtP79zkvFD3"};
BOOST_CHECK_EQUAL(nathan_keys[0], "5KQwrPbwdL6PhXujxW37FSSQZ1JiwsST4cqQzDeyXtP79zkvFD3");
BOOST_CHECK(wapiptr->import_key("nathan", nathan_keys[0]));

BOOST_TEST_MESSAGE("Importing nathan's balance");
std::vector<signed_transaction> import_txs = wapiptr->import_balance("nathan", nathan_keys, true);

account_object nathan_acct_before_upgrade = wapiptr->get_account("nathan");

// upgrade nathan
BOOST_TEST_MESSAGE("Upgrading Nathan to LTM");
signed_transaction upgrade_tx = wapiptr->upgrade_account("nathan", true);

account_object nathan_acct_after_upgrade = wapiptr->get_account("nathan");

// verify that the upgrade was successful
BOOST_CHECK_PREDICATE( std::not_equal_to<uint32_t>(), (nathan_acct_before_upgrade.membership_expiration_date.sec_since_epoch())(nathan_acct_after_upgrade.membership_expiration_date.sec_since_epoch()) );
BOOST_CHECK(nathan_acct_after_upgrade.is_lifetime_member());

// create a new account
graphene::wallet::brain_key_info bki = wapiptr->suggest_brain_key();
BOOST_CHECK(!bki.brain_priv_key.empty());
signed_transaction create_acct_tx = wapiptr->create_account_with_brain_key(bki.brain_priv_key, "jmjatlanta", "nathan", "nathan", true);

// save the private key for this new account in the wallet file
BOOST_CHECK(wapiptr->import_key("jmjatlanta", bki.wif_priv_key));
wapiptr->save_wallet_file(wallet_filename.str());

// attempt to give jmjatlanta some bitsahres
BOOST_TEST_MESSAGE("Transferring bitshares from Nathan to jmjatlanta");
signed_transaction transfer_tx = wapiptr->transfer("nathan", "jmjatlanta", "10000", "BTS", "Here are some BTS for your new account", true);

// get the details for init1
witness_object init1_obj = wapiptr->get_witness("init1");
int init1_start_votes = init1_obj.total_votes;
// Vote for a witness
signed_transaction vote_witness1_tx = wapiptr->vote_for_witness("jmjatlanta", "init1", true, true);
std::cout << "Displaying transaction for initial vote: " << std::endl;
//display_account_update_tx(vote_witness1_tx, wapiptr->get_result_formatters());
std::cout << fc::json::to_pretty_string( vote_witness1_tx ) << std::endl;

// wait for a maintenance interval
// NOTE: For this to work consistently, your maintenance interval must be less than 20 seconds
fc::usleep(fc::seconds(20));
// send a block to trigger maintenance interval
fc::ecc::private_key committee_key = fc::ecc::private_key::regenerate(fc::sha256::hash(string("nathan")));
auto db = app1.chain_database();
auto block_1 = db->generate_block(
db->get_slot_time(1),
db->get_scheduled_witness(1),
committee_key,
database::skip_nothing);
BOOST_CHECK(generate_block(*app1.get()));

// Verify that the vote is there
init1_obj = wapiptr->get_witness("init1");
witness_object init2_obj = wapiptr->get_witness("init2");
std::cout << "After first maintenance, here is the init1 witness: " << std::endl;
std::cout << fc::json::to_pretty_string( init1_obj) << std::endl;
std::cout << "After first maintenance, here is the init2 witness: " << std::endl;
std::cout << fc::json::to_pretty_string( init2_obj) << std::endl;
int init1_middle_votes = init1_obj.total_votes;
BOOST_CHECK(init1_middle_votes > init1_start_votes);

// Vote for a 2nd witness
int init2_start_votes = init2_obj.total_votes;
signed_transaction vote_witness2_tx = wapiptr->vote_for_witness("jmjatlanta", "init2", true, true);
std::cout << "Displaying transaction for init2 vote: " << std::endl;
//display_account_update_tx(vote_witness2_tx, wapiptr->get_result_formatters());
std::cout << fc::json::to_pretty_string( vote_witness2_tx ) << std::endl;

// send another block to trigger maintenance interval
fc::usleep(fc::seconds(20));
// send anoter block to trigger maintenance interval
block_1 = db->generate_block(
db->get_slot_time(1),
db->get_scheduled_witness(1),
committee_key,
database::skip_nothing);
BOOST_CHECK(generate_block(*app1.get()));

// Verify that both the first vote and the 2nd are there
init2_obj = wapiptr->get_witness("init2");
init1_obj = wapiptr->get_witness("init1");
std::cout << "After second maintenance, here is the init1 witness: " << std::endl;
std::cout << fc::json::to_pretty_string( init1_obj) << std::endl;
std::cout << "After second maintenance, here is the init2 witness: " << std::endl;
std::cout << fc::json::to_pretty_string( init2_obj) << std::endl;

int init2_middle_votes = init2_obj.total_votes;
BOOST_CHECK(init2_middle_votes > init2_start_votes);
int init1_last_votes = init1_obj.total_votes;
BOOST_CHECK(init1_middle_votes == init1_last_votes);
BOOST_CHECK(init1_last_votes > init1_start_votes);

// set the voting proxy to nathan
BOOST_TEST_MESSAGE("About to set voting proxy.");
signed_transaction voting_tx = wapiptr->set_voting_proxy("jmjatlanta", "nathan", true);

fc::usleep(fc::milliseconds(1000));
// wait for everything to finish up
fc::usleep(fc::seconds(1));
} catch( fc::exception& e ) {
edump((e.to_detail_string()));
throw;
}
app1.shutdown();
app1->shutdown();
}

0 comments on commit 2cc12d9

Please sign in to comment.