refactor constants to be determined during compile. added RLPXDatagram. constructors and updates to datagram constructors. docs and logging. WIP tests and host/harness for NodeTable.

This commit is contained in:
subtly 2014-12-23 08:56:56 +01:00
parent 5f9a740fd6
commit 1e0d4c95ce

71
net.cpp
View File

@ -30,6 +30,8 @@ using namespace dev::p2p;
namespace ba = boost::asio; namespace ba = boost::asio;
namespace bi = ba::ip; namespace bi = ba::ip;
BOOST_AUTO_TEST_SUITE(p2p)
/** /**
* Only used for testing. Not useful beyond tests. * Only used for testing. Not useful beyond tests.
*/ */
@ -37,7 +39,7 @@ class TestHost: public Worker
{ {
public: public:
TestHost(): Worker("test",0), m_io() {}; TestHost(): Worker("test",0), m_io() {};
~TestHost() { m_io.stop(); stopWorking(); } virtual ~TestHost() { m_io.stop(); stopWorking(); }
void start() { startWorking(); } void start() { startWorking(); }
void doWork() { m_io.run(); } void doWork() { m_io.run(); }
@ -45,28 +47,62 @@ protected:
ba::io_service m_io; ba::io_service m_io;
}; };
struct TestNodeTable: public NodeTable
{
void generateTestNodes(int _count = 10)
{
asserts(_count < 1000);
static uint16_t s_basePort = 30500;
m_testNodes.clear();
for (auto i = 0; i < _count; i++)
m_testNodes.push_back(make_pair(KeyPair::create(),s_basePort++));
}
std::vector<std::pair<KeyPair,unsigned>> m_testNodes; // keypair and port
/// Constructor
using NodeTable::NodeTable;
void setup()
{
/// Phase 1 test: populate with pings
/// Phase 2 test: pre-populate *expected* ping-responses, send pings
bi::address ourIp = bi::address::from_string("127.0.0.1");
uint16_t ourPort = 30300;
bi::udp::endpoint ourEndpoint(ourIp, ourPort);
generateTestNodes();
for (auto& n: m_testNodes)
ping(bi::udp::endpoint(ourIp, n.second));
// wait 1ms between each send
// send PingNode for each s_bootstrapNodes
// wait until nodecount is s_testNodes.count()
}
void reset()
{
Guard l(x_state);
for (auto& n: m_state) n.nodes.clear();
}
};
/** /**
* Only used for testing. Not useful beyond tests. * Only used for testing. Not useful beyond tests.
*/ */
class TestNodeHost: public TestHost struct TestNodeTableHost: public TestHost
{ {
public: TestNodeTableHost(): nodeTable(new TestNodeTable(m_io)) {};
TestNodeHost(): m_nodes(m_io) {}; shared_ptr<TestNodeTable> nodeTable;
~TestNodeHost() { m_io.stop(); stopWorking(); }
void start() { startWorking(); }
void doWork() { m_io.run(); }
NodeTable m_nodes;
}; };
class TestUDPSocket: UDPSocketEvents, public TestHost class TestUDPSocket: UDPSocketEvents, public TestHost
{ {
public: public:
TestUDPSocket(): m_socket(new UDPSocket<TestUDPSocket, 1024>(m_io, *this, 30300)) {} TestUDPSocket(): m_socket(new UDPSocket<TestUDPSocket, 1024>(m_io, *this, 30300)) {}
~TestUDPSocket() { m_io.stop(); stopWorking(); }
void start() { startWorking(); }
void doWork() { m_io.run(); }
void onDisconnected(UDPSocketFace*) {}; void onDisconnected(UDPSocketFace*) {};
void onReceived(UDPSocketFace*, bi::udp::endpoint const&, bytesConstRef _packet) { if (_packet.toString() == "AAAA") success = true; } void onReceived(UDPSocketFace*, bi::udp::endpoint const&, bytesConstRef _packet) { if (_packet.toString() == "AAAA") success = true; }
@ -75,12 +111,13 @@ public:
bool success = false; bool success = false;
}; };
BOOST_AUTO_TEST_SUITE(p2p)
BOOST_AUTO_TEST_CASE(kademlia) BOOST_AUTO_TEST_CASE(kademlia)
{ {
TestNodeHost nodeHost; // TestNodeTableHost node;
// node.start();
// node.nodeTable->join(); // ideally, joining with empty node table logs warning we can check for
// node.nodeTable->setup();
// sleep(1);
} }
BOOST_AUTO_TEST_CASE(test_txrx_one) BOOST_AUTO_TEST_CASE(test_txrx_one)