From cc1b952c23c7d12b1285ee0a206abc052ff74047 Mon Sep 17 00:00:00 2001 From: colts661 Date: Wed, 13 Dec 2023 11:30:57 -0800 Subject: [PATCH] removed all inserted iu --- .../repo/tests/test_perk_fulfillment_repo.py | 11 +++++++++-- 1 file changed, 9 insertions(+), 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 index 395892120..641370db8 100644 --- 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 "