diff options
| author | João Barbosa <[email protected]> | 2017-10-24 09:26:05 +0100 |
|---|---|---|
| committer | Suhas Daftuar <[email protected]> | 2017-11-02 12:39:14 -0400 |
| commit | 83df25736ebaa23777e965d265ed677a8d79f23f (patch) | |
| tree | 781000ee1e9b246db0ff115229e53171b69b7c96 /src/test | |
| parent | Connect to an extra outbound peer if our tip is stale (diff) | |
| download | discoin-83df25736ebaa23777e965d265ed677a8d79f23f.tar.xz discoin-83df25736ebaa23777e965d265ed677a8d79f23f.zip | |
Add CConnmanTest to mutate g_connman in tests
Diffstat (limited to 'src/test')
| -rw-r--r-- | src/test/test_bitcoin.cpp | 12 | ||||
| -rw-r--r-- | src/test/test_bitcoin.h | 6 |
2 files changed, 18 insertions, 0 deletions
diff --git a/src/test/test_bitcoin.cpp b/src/test/test_bitcoin.cpp index c768446d3..85476b6da 100644 --- a/src/test/test_bitcoin.cpp +++ b/src/test/test_bitcoin.cpp @@ -25,6 +25,18 @@ #include <memory> +void CConnmanTest::AddNode(CNode& node) +{ + LOCK(g_connman->cs_vNodes); + g_connman->vNodes.push_back(&node); +} + +void CConnmanTest::ClearNodes() +{ + LOCK(g_connman->cs_vNodes); + g_connman->vNodes.clear(); +} + uint256 insecure_rand_seed = GetRandHash(); FastRandomContext insecure_rand_ctx(insecure_rand_seed); diff --git a/src/test/test_bitcoin.h b/src/test/test_bitcoin.h index 2390aca34..62ded2aaf 100644 --- a/src/test/test_bitcoin.h +++ b/src/test/test_bitcoin.h @@ -49,6 +49,12 @@ struct BasicTestingSetup { * Included are data directory, coins database, script check threads setup. */ class CConnman; +class CNode; +struct CConnmanTest { + static void AddNode(CNode& node); + static void ClearNodes(); +}; + class PeerLogicValidation; struct TestingSetup: public BasicTestingSetup { CCoinsViewDB *pcoinsdbview; |