From 695570064b033c839e2344a01daca64075b273a0 Mon Sep 17 00:00:00 2001 From: Jonathan Zernik Date: Fri, 22 Oct 2021 12:35:23 -0500 Subject: [PATCH] Add test for delete peer db method (#1664) --- tests/db/test_squeak_db.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/tests/db/test_squeak_db.py b/tests/db/test_squeak_db.py index 812861a3..89b06d97 100644 --- a/tests/db/test_squeak_db.py +++ b/tests/db/test_squeak_db.py @@ -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