diff options
author | João Barbosa <joao.paulo.barbosa@gmail.com> | 2017-10-24 09:26:05 +0100 |
---|---|---|
committer | Suhas Daftuar <sdaftuar@gmail.com> | 2017-11-02 12:39:14 -0400 |
commit | 83df25736ebaa23777e965d265ed677a8d79f23f (patch) | |
tree | 781000ee1e9b246db0ff115229e53171b69b7c96 /src/test | |
parent | ac7b37cd2bd612a64a4009ba82f1cd1d57f37434 (diff) |
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 c768446d30..85476b6da2 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 2390aca342..62ded2aaf5 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; |