From e1a9674200f895a8a2cfafcf21167fd86a011ae5 Mon Sep 17 00:00:00 2001 From: Jonathan Zernik Date: Wed, 6 Oct 2021 15:39:42 -0700 Subject: [PATCH] Add unit test for set profile following (#1520) --- tests/db/test_squeak_db.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/db/test_squeak_db.py b/tests/db/test_squeak_db.py index 6e7c938d..59fb5f3d 100644 --- a/tests/db/test_squeak_db.py +++ b/tests/db/test_squeak_db.py @@ -207,3 +207,16 @@ def test_get_contact_profile(squeak_db, address, contact_profile): assert retrieved_profile.profile_name == contact_profile.profile_name assert retrieved_profile.address == contact_profile.address + + +def test_set_profile_following(squeak_db, address, contact_profile): + profile_id = squeak_db.insert_profile(contact_profile) + squeak_db.set_profile_following(profile_id, True) + retrieved_profile = squeak_db.get_profile(profile_id) + + assert retrieved_profile.following + + squeak_db.set_profile_following(profile_id, False) + retrieved_profile = squeak_db.get_profile(profile_id) + + assert not retrieved_profile.following