You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.4 KiB
118 lines
3.4 KiB
/*
|
|
This file is part of cpp-ethereum.
|
|
|
|
cpp-ethereum is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
cpp-ethereum is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with cpp-ethereum. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
/** @file TestUtils.cpp
|
|
* @author Marek Kotewicz <marek@ethdev.com>
|
|
* @date 2015
|
|
*/
|
|
|
|
#include <thread>
|
|
#include <boost/test/unit_test.hpp>
|
|
#include <boost/filesystem.hpp>
|
|
#include <libtestutils/Common.h>
|
|
#include <libtestutils/BlockChainLoader.h>
|
|
#include <libtestutils/FixedClient.h>
|
|
#include "TestUtils.h"
|
|
|
|
using namespace std;
|
|
using namespace dev;
|
|
using namespace dev::eth;
|
|
using namespace dev::test;
|
|
|
|
namespace dev
|
|
{
|
|
namespace test
|
|
{
|
|
|
|
bool getCommandLineOption(std::string const& _name);
|
|
std::string getCommandLineArgument(std::string const& _name, bool _require = false);
|
|
|
|
}
|
|
}
|
|
|
|
bool dev::test::getCommandLineOption(string const& _name)
|
|
{
|
|
auto argc = boost::unit_test::framework::master_test_suite().argc;
|
|
auto argv = boost::unit_test::framework::master_test_suite().argv;
|
|
bool result = false;
|
|
for (auto i = 0; !result && i < argc; ++i)
|
|
result = _name == argv[i];
|
|
return result;
|
|
}
|
|
|
|
std::string dev::test::getCommandLineArgument(string const& _name, bool _require)
|
|
{
|
|
auto argc = boost::unit_test::framework::master_test_suite().argc;
|
|
auto argv = boost::unit_test::framework::master_test_suite().argv;
|
|
for (auto i = 1; i < argc; ++i)
|
|
{
|
|
string str = argv[i];
|
|
if (_name == str.substr(0, _name.size()))
|
|
return str.substr(str.find("=") + 1);
|
|
}
|
|
if (_require)
|
|
BOOST_ERROR("Failed getting command line argument: " << _name << " from: " << argv);
|
|
return "";
|
|
}
|
|
|
|
LoadTestFileFixture::LoadTestFileFixture()
|
|
{
|
|
m_json = loadJsonFromFile(toTestFilePath(getCommandLineArgument("--eth_testfile")));
|
|
}
|
|
|
|
void ParallelFixture::enumerateThreads(std::function<void()> callback) const
|
|
{
|
|
size_t threadsCount = std::stoul(getCommandLineArgument("--eth_threads"), nullptr, 10);
|
|
|
|
vector<thread> workers;
|
|
for (size_t i = 0; i < threadsCount; i++)
|
|
workers.emplace_back(callback);
|
|
|
|
for_each(workers.begin(), workers.end(), [](thread &t)
|
|
{
|
|
t.join();
|
|
});
|
|
}
|
|
|
|
void BlockChainFixture::enumerateBlockchains(std::function<void(Json::Value const&, dev::eth::BlockChain const&, State state)> callback) const
|
|
{
|
|
for (string const& name: m_json.getMemberNames())
|
|
{
|
|
BlockChainLoader bcl(m_json[name]);
|
|
callback(m_json[name], bcl.bc(), bcl.state());
|
|
}
|
|
}
|
|
|
|
void ClientBaseFixture::enumerateClients(std::function<void(Json::Value const&, dev::eth::ClientBase&)> callback) const
|
|
{
|
|
enumerateBlockchains([&callback](Json::Value const& _json, BlockChain const& _bc, State _state) -> void
|
|
{
|
|
FixedClient client(_bc, _state);
|
|
callback(_json, client);
|
|
});
|
|
}
|
|
|
|
void ParallelClientBaseFixture::enumerateClients(std::function<void(Json::Value const&, dev::eth::ClientBase&)> callback) const
|
|
{
|
|
ClientBaseFixture::enumerateClients([this, &callback](Json::Value const& _json, dev::eth::ClientBase& _client) -> void
|
|
{
|
|
// json is being copied here
|
|
enumerateThreads([callback, _json, &_client]() -> void
|
|
{
|
|
callback(_json, _client);
|
|
});
|
|
});
|
|
}
|
|
|