Differential D5458 Diff 16905 src/seeder/test/p2p_messaging_tests.cpp

Changeset View

Standalone View

src/seeder/test/p2p_messaging_tests.cpp

	Show All 15 Lines
	#include <memory>			#include <memory>
	#include <string>			#include <string>
	#include <vector>			#include <vector>

	const std::function<std::string(const char *)> G_TRANSLATION_FUN = nullptr;			const std::function<std::string(const char *)> G_TRANSLATION_FUN = nullptr;

	#include <boost/test/unit_test.hpp>			#include <boost/test/unit_test.hpp>

	enum PeerMessagingState : bool {
	Finished = true,
	AwaitingMessages = false,
	};

	class TestCSeederNode : public CSeederNode {			class TestCSeederNode : public CSeederNode {
	public:			public:
	TestCSeederNode(const CService &service, std::vector<CAddress> *vAddrIn)			TestCSeederNode(const CService &service, std::vector<CAddress> *vAddrIn)
	: CSeederNode(service, vAddrIn) {			: CSeederNode(service, vAddrIn) {
	SelectParams(CBaseChainParams::REGTEST);			SelectParams(CBaseChainParams::REGTEST);
	}			}

	void TestProcessMessage(const std::string &strCommand, CDataStream &message,			void TestProcessMessage(const std::string &strCommand, CDataStream &message,
	PeerMessagingState state) {			PeerMessagingState expectedState) {
	bool ret = CSeederNode::ProcessMessage(strCommand, message);			PeerMessagingState ret =
	BOOST_CHECK_EQUAL(ret, bool(state));			CSeederNode::ProcessMessage(strCommand, message);
				BOOST_CHECK_EQUAL(ret, expectedState);
	}			}
	};			};

	static const unsigned short SERVICE_PORT = 18444;			static const unsigned short SERVICE_PORT = 18444;

	struct SeederTestingSetup {			struct SeederTestingSetup {
	SeederTestingSetup() {			SeederTestingSetup() {
	CNetAddr ip;			CNetAddr ip;
	▲ Show 20 Lines • Show All 92 Lines • Show Last 20 Lines