diff --git a/src/seeder/test/p2p_messaging_tests.cpp b/src/seeder/test/p2p_messaging_tests.cpp --- a/src/seeder/test/p2p_messaging_tests.cpp +++ b/src/seeder/test/p2p_messaging_tests.cpp @@ -21,38 +21,11 @@ #include -static const unsigned short SERVICE_PORT = 18444; -static const int SEEDER_INIT_VERSION = 0; -// After the 1000th addr, the seeder will only add one more address per addr -// message. -static const int ADDR_SOFT_CAP = 1000; - enum PeerMessagingState : bool { Finished = true, AwaitingMessages = false, }; -static CDataStream -CreateVersionMessage(int64_t now, CAddress addrTo, CAddress addrFrom, - int32_t start_height, uint32_t nVersion, - uint64_t nonce = 0, - std::string user_agent = "/bitcoin-cash-seeder:0.15/") { - CDataStream payload(SER_NETWORK, 0); - payload.SetVersion(nVersion); - ServiceFlags serviceflags = ServiceFlags(NODE_NETWORK); - payload << nVersion << uint64_t(serviceflags) << now << addrTo << addrFrom - << nonce << user_agent << start_height; - return payload; -} - -static CDataStream CreateAddrMessage(std::vector sendAddrs, - uint32_t nVersion = INIT_PROTO_VERSION) { - CDataStream payload(SER_NETWORK, 0); - payload.SetVersion(nVersion); - payload << sendAddrs; - return payload; -} - class TestCSeederNode : public CSeederNode { public: TestCSeederNode(const CService &service, std::vector *vAddrIn) @@ -67,6 +40,8 @@ } }; +static const unsigned short SERVICE_PORT = 18444; + struct SeederTestingSetup { SeederTestingSetup() { CNetAddr ip; @@ -82,6 +57,21 @@ BOOST_FIXTURE_TEST_SUITE(p2p_messaging_tests, SeederTestingSetup) +static CDataStream +CreateVersionMessage(int64_t now, CAddress addrTo, CAddress addrFrom, + int32_t start_height, uint32_t nVersion, + uint64_t nonce = 0, + std::string user_agent = "/bitcoin-cash-seeder:0.15/") { + CDataStream payload(SER_NETWORK, 0); + payload.SetVersion(nVersion); + ServiceFlags serviceflags = ServiceFlags(NODE_NETWORK); + payload << nVersion << uint64_t(serviceflags) << now << addrTo << addrFrom + << nonce << user_agent << start_height; + return payload; +} + +static const int SEEDER_INIT_VERSION = 0; + BOOST_AUTO_TEST_CASE(seeder_node_version_test) { CService serviceFrom; CAddress addrFrom(serviceFrom, @@ -101,6 +91,18 @@ versionMessage.GetVersion()); } +static CDataStream CreateAddrMessage(std::vector sendAddrs, + uint32_t nVersion = INIT_PROTO_VERSION) { + CDataStream payload(SER_NETWORK, 0); + payload.SetVersion(nVersion); + payload << sendAddrs; + return payload; +} + +// After the 1000th addr, the seeder will only add one more address per addr +// message. +static const int ADDR_SOFT_CAP = 1000; + BOOST_AUTO_TEST_CASE(seeder_node_addr_test) { // vAddrs starts with 1 entry. std::vector sendAddrs(ADDR_SOFT_CAP - 1, vAddr[0]);