diff options
Diffstat (limited to 'src/test/util/net.h')
-rw-r--r-- | src/test/util/net.h | 16 |
1 files changed, 11 insertions, 5 deletions
diff --git a/src/test/util/net.h b/src/test/util/net.h index 939ec322ed..ad1061a451 100644 --- a/src/test/util/net.h +++ b/src/test/util/net.h @@ -17,18 +17,24 @@ struct ConnmanTestMsg : public CConnman { using CConnman::CConnman; + + void SetPeerConnectTimeout(std::chrono::seconds timeout) + { + m_peer_connect_timeout = timeout; + } + void AddTestNode(CNode& node) { - LOCK(cs_vNodes); - vNodes.push_back(&node); + LOCK(m_nodes_mutex); + m_nodes.push_back(&node); } void ClearTestNodes() { - LOCK(cs_vNodes); - for (CNode* node : vNodes) { + LOCK(m_nodes_mutex); + for (CNode* node : m_nodes) { delete node; } - vNodes.clear(); + m_nodes.clear(); } void ProcessMessagesOnce(CNode& node) { m_msgproc->ProcessMessages(&node, flagInterruptMsgProc); } |