From 3effe81546dc692032a993ecb7bfed7a24714282 Mon Sep 17 00:00:00 2001 From: Jonathan Zernik Date: Sun, 3 Oct 2021 21:11:28 -0700 Subject: [PATCH] Add unit test for get missing squeak entry from db (#1491) --- tests/db/test_squeak_db.py | 9 ++++++++- 1 file changed, 8 insertions(+), 1 deletion(-) diff --git a/tests/db/test_squeak_db.py b/tests/db/test_squeak_db.py index b3e58abc..56cf74c4 100644 --- a/tests/db/test_squeak_db.py +++ b/tests/db/test_squeak_db.py @@ -56,7 +56,7 @@ def test_insert_duplicate_squeak(squeak_db, squeak, genesis_block_info): assert second_squeak_hash is None -def test_get_missing_squeak(squeak_db, squeak, genesis_block_info): +def test_get_missing_squeak(squeak_db, squeak): squeak_hash = get_hash(squeak) retrieved_squeak = squeak_db.get_squeak(squeak_hash) @@ -70,3 +70,10 @@ def test_get_squeak_entry(squeak_db, squeak, genesis_block_info, address): assert retrieved_squeak_entry.squeak_hash == squeak_hash assert retrieved_squeak_entry.address == address + + +def test_get_missing_squeak_entry(squeak_db, squeak, address): + squeak_hash = get_hash(squeak) + retrieved_squeak_entry = squeak_db.get_squeak_entry(squeak_hash) + + assert retrieved_squeak_entry is None