X Tutup
Skip to content

Commit ecfcc69

Browse files
committed
Fixed tests
1 parent 7e81a15 commit ecfcc69

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

tests/test_filters.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -214,34 +214,39 @@ def test_group_fileter(self):
214214
self.assertTrue(Filters.group(self.message))
215215

216216
def test_filters_chat(self):
217-
with self.assertRaisesRegexp(ValueError, 'either chat_id or username'):
218-
Filters.chat(chat_id=0, username='chat')
217+
with self.assertRaisesRegexp(ValueError, 'chat_id or username'):
218+
Filters.chat(chat_id=-1, username='chat')
219+
with self.assertRaisesRegexp(ValueError, 'chat_id or username'):
219220
Filters.chat()
220221

221222
def test_filters_chat_id(self):
222223
self.assertFalse(Filters.chat(chat_id=-1)(self.message))
223224
self.message.chat.id = -1
224225
self.assertTrue(Filters.chat(chat_id=-1)(self.message))
225226
self.message.chat.id = -2
227+
self.assertTrue(Filters.chat(chat_id=[-1, -2])(self.message))
226228
self.assertFalse(Filters.chat(chat_id=-1)(self.message))
227229

228230
def test_filters_chat_username(self):
229231
self.assertFalse(Filters.chat(username='chat')(self.message))
230232
self.message.chat.username = 'chat'
231233
self.assertTrue(Filters.chat(username='@chat')(self.message))
232234
self.assertTrue(Filters.chat(username='chat')(self.message))
233-
self.assertFalse(Filters.chat(username='@chat_username')(self.message))
235+
self.assertTrue(Filters.chat(username=['chat1', 'chat', 'chat2'])(self.message))
236+
self.assertFalse(Filters.chat(username=['@chat1', 'chat_2'])(self.message))
234237

235238
def test_filters_user(self):
236-
with self.assertRaisesRegexp(ValueError, 'either user_id or username'):
237-
Filters.user(user_id=0, username='user')
239+
with self.assertRaisesRegexp(ValueError, 'user_id or username'):
240+
Filters.user(user_id=1, username='user')
241+
with self.assertRaisesRegexp(ValueError, 'user_id or username'):
238242
Filters.user()
239243

240244
def test_filters_user_id(self):
241245
self.assertFalse(Filters.user(user_id=1)(self.message))
242246
self.message.from_user.id = 1
243247
self.assertTrue(Filters.user(user_id=1)(self.message))
244248
self.message.from_user.id = 2
249+
self.assertTrue(Filters.user(user_id=[1, 2])(self.message))
245250
self.assertFalse(Filters.user(user_id=1)(self.message))
246251

247252
def test_filters_username(self):
@@ -250,7 +255,8 @@ def test_filters_username(self):
250255
self.message.from_user.username = 'user'
251256
self.assertTrue(Filters.user(username='@user')(self.message))
252257
self.assertTrue(Filters.user(username='user')(self.message))
253-
self.assertFalse(Filters.user(username='@username')(self.message))
258+
self.assertTrue(Filters.user(username=['user1', 'user', 'user2'])(self.message))
259+
self.assertFalse(Filters.user(username=['@username', '@user_2'])(self.message))
254260

255261
def test_and_filters(self):
256262
self.message.text = 'test'

0 commit comments

Comments
 (0)
X Tutup