|
21 | 21 | async def test_refresh_external_sources__removed_user_is_kicked( |
22 | 22 | db_session: Session, |
23 | 23 | ): |
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) |
26 | 26 |
|
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) |
29 | 29 |
|
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 | + ) |
32 | 36 |
|
33 | | - source = TelegramChatWhitelistExternalSourceFactory.create( |
| 37 | + source = TelegramChatWhitelistExternalSourceFactory.with_session(db_session).create( |
34 | 38 | chat=chat, |
35 | 39 | group=group, |
36 | 40 | content=[1001, 1002], |
@@ -73,13 +77,15 @@ async def test_refresh_external_sources__removed_user_is_kicked( |
73 | 77 | async def test_refresh_external_sources__no_removed_users__no_kicks( |
74 | 78 | db_session: Session, |
75 | 79 | ): |
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) |
78 | 82 |
|
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 | + ) |
81 | 87 |
|
82 | | - TelegramChatWhitelistExternalSourceFactory.create( |
| 88 | + TelegramChatWhitelistExternalSourceFactory.with_session(db_session).create( |
83 | 89 | chat=chat, |
84 | 90 | group=group, |
85 | 91 | content=[1001], |
|
0 commit comments