diff --git a/tests/db/test_squeak_db.py b/tests/db/test_squeak_db.py index 9333defe..2960ed89 100644 --- a/tests/db/test_squeak_db.py +++ b/tests/db/test_squeak_db.py @@ -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