Skip to content

Commit 49c00a2

Browse files
authored
tests: refactor factories to use db_session in refresh_external_source tests (#235)
2 parents f3dd002 + ccb3bfa commit 49c00a2

File tree

1 file changed

+18
-12
lines changed

1 file changed

+18
-12
lines changed

backend/tests/unit/community_manager/actions/test_refresh_external_sources.py

Lines changed: 18 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -21,16 +21,20 @@
2121
async def test_refresh_external_sources__removed_user_is_kicked(
2222
db_session: Session,
2323
):
24-
chat = TelegramChatFactory.create(is_full_control=True)
25-
group = TelegramChatRuleGroupFactory.create(chat=chat)
24+
chat = TelegramChatFactory.with_session(db_session).create(is_full_control=True)
25+
group = TelegramChatRuleGroupFactory.with_session(db_session).create(chat=chat)
2626

27-
user_stays = UserFactory.create(telegram_id=1001)
28-
user_removed = UserFactory.create(telegram_id=1002)
27+
user_stays = UserFactory.with_session(db_session).create(telegram_id=1001)
28+
user_removed = UserFactory.with_session(db_session).create(telegram_id=1002)
2929

30-
TelegramChatUserFactory.create(chat=chat, user=user_stays, is_managed=True)
31-
TelegramChatUserFactory.create(chat=chat, user=user_removed, is_managed=True)
30+
TelegramChatUserFactory.with_session(db_session).create(
31+
chat=chat, user=user_stays, is_managed=True
32+
)
33+
TelegramChatUserFactory.with_session(db_session).create(
34+
chat=chat, user=user_removed, is_managed=True
35+
)
3236

33-
source = TelegramChatWhitelistExternalSourceFactory.create(
37+
source = TelegramChatWhitelistExternalSourceFactory.with_session(db_session).create(
3438
chat=chat,
3539
group=group,
3640
content=[1001, 1002],
@@ -73,13 +77,15 @@ async def test_refresh_external_sources__removed_user_is_kicked(
7377
async def test_refresh_external_sources__no_removed_users__no_kicks(
7478
db_session: Session,
7579
):
76-
chat = TelegramChatFactory.create(is_full_control=True)
77-
group = TelegramChatRuleGroupFactory.create(chat=chat)
80+
chat = TelegramChatFactory.with_session(db_session).create(is_full_control=True)
81+
group = TelegramChatRuleGroupFactory.with_session(db_session).create(chat=chat)
7882

79-
user = UserFactory.create(telegram_id=1001)
80-
TelegramChatUserFactory.create(chat=chat, user=user, is_managed=True)
83+
user = UserFactory.with_session(db_session).create(telegram_id=1001)
84+
TelegramChatUserFactory.with_session(db_session).create(
85+
chat=chat, user=user, is_managed=True
86+
)
8187

82-
TelegramChatWhitelistExternalSourceFactory.create(
88+
TelegramChatWhitelistExternalSourceFactory.with_session(db_session).create(
8389
chat=chat,
8490
group=group,
8591
content=[1001],

0 commit comments

Comments
 (0)