Improve fixure for liked and unliked squeak hashes (#1528)

This commit is contained in:
Jonathan Zernik 2021-10-07 03:59:52 -07:00 committed by GitHub
parent be5815d402
commit 7aa88efd17
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -134,11 +134,7 @@ def unfollowed_contact_profile(squeak_db, followed_contact_profile):
@pytest.fixture
def followed_squeak_hashes(
squeak_db,
signing_key,
followed_contact_profile,
):
def inserted_squeak_hashes(squeak_db, signing_key):
ret = []
for i in range(100):
squeak, header = gen_squeak_with_block_header(signing_key, i)
@ -147,39 +143,33 @@ def followed_squeak_hashes(
yield ret
@pytest.fixture
def followed_squeak_hashes(
squeak_db,
inserted_squeak_hashes,
followed_contact_profile,
):
yield inserted_squeak_hashes
@pytest.fixture
def unfollowed_squeak_hashes(
squeak_db,
signing_key,
inserted_squeak_hashes,
unfollowed_contact_profile,
):
ret = []
for i in range(100):
squeak, header = gen_squeak_with_block_header(signing_key, i)
squeak_hash = squeak_db.insert_squeak(squeak, header)
ret.append(squeak_hash)
yield ret
yield inserted_squeak_hashes
@pytest.fixture
def liked_squeak_hashes(
squeak_db,
signing_key,
):
ret = []
for i in range(100):
squeak, header = gen_squeak_with_block_header(signing_key, i)
squeak_hash = squeak_db.insert_squeak(squeak, header)
def liked_squeak_hashes(squeak_db, inserted_squeak_hashes):
for squeak_hash in inserted_squeak_hashes:
squeak_db.set_squeak_liked(squeak_hash)
ret.append(squeak_hash)
yield ret
yield inserted_squeak_hashes
@pytest.fixture
def unliked_squeak_hashes(
squeak_db,
liked_squeak_hashes,
):
def unliked_squeak_hashes(squeak_db, liked_squeak_hashes):
for squeak_hash in liked_squeak_hashes:
squeak_db.set_squeak_unliked(squeak_hash)
yield liked_squeak_hashes