GlcoinCore/squeaknode
1
0
Fork 0
mirror of https://github.com/yzernik/squeaknode.git synced 2026-08-17 13:07:32 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test for get autoconnect peers (#1662)

Browse source
This commit is contained in:
Jonathan Zernik 2021-10-22 12:18:18 -05:00 • committed by GitHub
parent 1ea1db1d3d
commit 0193647cdf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 26 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

27
tests/db/test_squeak_db.py
View file

@ -277,6 +277,13 @@ def inserted_squeak_peer_ids(squeak_db):
yield ret
@pytest.fixture
def autoconnect_squeak_peer_ids(squeak_db, inserted_squeak_peer_ids):
for peer_id in inserted_squeak_peer_ids:
squeak_db.set_peer_autoconnect(peer_id, True)
yield inserted_squeak_peer_ids
def test_init_with_retries(squeak_db):
with mock.patch.object(squeak_db, 'init', autospec=True) as mock_init, \
mock.patch('squeaknode.db.squeak_db.time.sleep', autospec=True) as mock_sleep:
@ -1011,4 +1018,22 @@ def test_get_squeak_peers(
):
peers = squeak_db.get_peers()
assert len(inserted_squeak_peer_ids) == len(peers)
assert len(peers) == len(inserted_squeak_peer_ids)
def test_get_autoconnect_squeak_peers(
squeak_db,
autoconnect_squeak_peer_ids,
):
peers = squeak_db.get_autoconnect_peers()
assert len(peers) == len(autoconnect_squeak_peer_ids)
def test_get_autoconnect_squeak_peers_none(
squeak_db,
inserted_squeak_peer_ids,
):
peers = squeak_db.get_autoconnect_peers()
assert len(peers) == 0