Merge pull request #557 from colts661/remove_test_iu

Removed extra inserted interested users in `test_perk_fulfillment_repo.py`
biocore · Dec 18, 2023 · 2a6644a · 2a6644a
2 parents 8d4c0bd + cc1b952
commit 2a6644a
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/microsetta_private_api/repo/tests/test_perk_fulfillment_repo.py b/microsetta_private_api/repo/tests/test_perk_fulfillment_repo.py
@@ -428,7 +428,8 @@ def tearDown(self):
             )
             cur.execute(
                 "DELETE FROM campaign.transaction "
-                "WHERE id IN %s",
+                "WHERE id IN %s "
+                "RETURNING interested_user_id",
                 (
                     (
                         FFQ_TRANSACTION_ID,
@@ -439,10 +440,16 @@ def tearDown(self):
                     ),
                 )
             )
+            inserted_iu_ids = [tup[0] for tup in cur.fetchall()]
             cur.execute(
                 "DELETE FROM campaign.interested_users AS iu "
                 "WHERE iu.interested_user_id IN %s",
-                (tuple(self.iu_ids), )
+                (
+                    (
+                        *inserted_iu_ids,
+                        *self.iu_ids
+                    ),
+                )  # self.iu_ids contains iu without transaction
             )
             cur.execute(
                 "DELETE FROM campaign.campaigns_projects "