GlcoinCore/squeaknode
1
0
Fork 0
mirror of https://github.com/yzernik/squeaknode.git synced 2026-08-13 12:33:25 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test for get following profiles db method (#1647)

Browse source
This commit is contained in:
Jonathan Zernik 2021-10-19 06:32:14 -05:00 • committed by GitHub
parent 70fbb9e7db
commit 3646fbb742
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

19
tests/db/test_squeak_db.py
View file

@ -207,6 +207,16 @@ def inserted_signing_profile_ids(squeak_db):
yield ret
@pytest.fixture
def followed_contact_profile_ids(squeak_db, inserted_contact_profile_ids):
for profile_id in inserted_contact_profile_ids:
squeak_db.set_profile_following(
profile_id,
True,
)
yield inserted_contact_profile_ids
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:
@ -827,3 +837,12 @@ def test_get_contact_profiles_none(
profiles = squeak_db.get_contact_profiles()
assert len(profiles) == 0
def test_get_following_profiles(
squeak_db,
followed_contact_profile_ids,
):
profiles = squeak_db.get_following_profiles()
assert len(profiles) == len(followed_contact_profile_ids)