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

Add test for delete peer db method (#1664)

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

14
tests/db/test_squeak_db.py
View file

@ -298,6 +298,14 @@ def peer_with_new_name_id(squeak_db, inserted_peer_id, new_peer_name):
yield inserted_peer_id
@pytest.fixture
def deleted_peer_id(squeak_db, inserted_peer_id):
squeak_db.delete_peer(
inserted_peer_id,
)
yield inserted_peer_id
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:
@ -1057,3 +1065,9 @@ def test_set_peer_name(squeak_db, peer_with_new_name_id, new_peer_name):
peer = squeak_db.get_peer(peer_with_new_name_id)
assert peer.peer_name == new_peer_name
def test_get_deleted_peer(squeak_db, deleted_peer_id):
retrieved_peer = squeak_db.get_peer(deleted_peer_id)
assert retrieved_peer is None