Add unit test for duplicate squeak (#1489)

* Add unit test for insert duplicate squeak in db

* Add unit test for get missing squeak from db
This commit is contained in:
Jonathan Zernik 2021-10-03 20:52:04 -07:00 committed by GitHub
parent 2e0fc25051
commit 080d6299b5
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -39,10 +39,25 @@ def squeak_db(db_engine):
yield db
def test_insert_retrieve_squeak(squeak_db, squeak, genesis_block_info):
def test_insert_get_squeak(squeak_db, squeak, genesis_block_info):
block_header = parse_block_header(genesis_block_info.block_header)
squeak_db.insert_squeak(squeak, block_header)
squeak_hash = get_hash(squeak)
squeak_hash = squeak_db.insert_squeak(squeak, block_header)
retrieved_squeak = squeak_db.get_squeak(squeak_hash)
assert retrieved_squeak == squeak
def test_insert_duplicate_squeak(squeak_db, squeak, genesis_block_info):
block_header = parse_block_header(genesis_block_info.block_header)
first_squeak_hash = squeak_db.insert_squeak(squeak, block_header)
second_squeak_hash = squeak_db.insert_squeak(squeak, block_header)
assert first_squeak_hash is not None
assert second_squeak_hash is None
def test_get_missing_squeak(squeak_db, squeak, genesis_block_info):
squeak_hash = get_hash(squeak)
retrieved_squeak = squeak_db.get_squeak(squeak_hash)
assert retrieved_squeak is None