Add unit test for stale tip checking

1 files changed, 9 insertions, 0 deletions

diff --git a/src/net_processing.cpp b/src/net_processing.cpp
index 3aa13fff43..faa0c7620b 100644
--- a/src/net_processing.cpp
+++ b/src/net_processing.cpp
@@ -550,6 +550,15 @@ void FindNextBlocksToDownload(NodeId nodeid, unsigned int count, std::vector<con
 
 } // namespace
 
+// This function is used for testing the stale tip eviction logic, see
+// DoS_tests.cpp
+void UpdateLastBlockAnnounceTime(NodeId node, int64_t time_in_seconds)
+{
+    LOCK(cs_main);
+    CNodeState *state = State(node);
+    if (state) state->m_last_block_announcement = time_in_seconds;
+}
+
 // Returns true for outbound peers, excluding manual connections, feelers, and
 // one-shots
 bool IsOutboundDisconnectionCandidate(const CNode *node)