python/aiogram
1
0
Fork 0
mirror of https://github.com/aiogram/aiogram.git synced 2026-04-08 16:37:47 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

refactor: remove MediaGroupFilter and fix rst heading

Browse source
This commit is contained in:
Vitaly312 2026-04-05 18:15:59 +03:00
parent 24fdd285fd
commit 28626d124a
4 changed files with 6 additions and 147 deletions
Download patch file Download diff file Expand all files Collapse all files

73
tests/test_filters/test_media_group.py
View file

@ -1,73 +0,0 @@
import datetime
import pytest
from aiogram.filters.media_group import DEFAULT_MAX_MEDIA_COUNT, MIN_MEDIA_COUNT, MediaGroupFilter
from aiogram.types import Chat, Message
class TestMediaGroupFilter:
@pytest.mark.parametrize(
"args,min_count,max_count",
[
((), MIN_MEDIA_COUNT, DEFAULT_MAX_MEDIA_COUNT),
((3,), 3, 3),
((11,), 11, 11),
((None, 11, None), 11, 11),
((None, 3), 3, DEFAULT_MAX_MEDIA_COUNT),
((None, None, 3), MIN_MEDIA_COUNT, 3),
],
)
def test_init_range(self, args, min_count, max_count):
filter = MediaGroupFilter(*args)
assert filter.max_media_count == max_count
assert filter.min_media_count == min_count
@pytest.mark.parametrize(
"count,min_count,max_count",
[
(1, None, 1),
(1, 1, None),
(None, 1, None),
(None, None, 1),
(1, None, None),
(None, 5, 3),
],
)
def test_raise_error(self, count, min_count, max_count):
with pytest.raises(ValueError):
MediaGroupFilter(count, min_count, max_count)
@pytest.mark.parametrize(
"min_count,max_count,media_count,result",
[
[2, 2, 1, False],
[2, 2, 2, True],
[2, 2, 3, False],
[2, 5, 2, True],
[2, 5, 5, True],
[2, 5, 6, False],
],
)
async def test_call(self, min_count, max_count, media_count, result):
filter = MediaGroupFilter(min_media_count=min_count, max_media_count=max_count)
album = [
Message(
message_id=i,
date=datetime.datetime.now(),
chat=Chat(id=42, type="private"),
)
for i in range(media_count)
]
response = await filter(album[0], album)
assert bool(response) is result
if result:
assert response.get("media_count") == media_count
def test_str_count(self):
filter = MediaGroupFilter(5)
assert str(filter) == "MediaGroupFilter(count=5)"
def test_str_range(self):
filter = MediaGroupFilter(min_media_count=2, max_media_count=5)
assert str(filter) == "MediaGroupFilter(min_media_count=2, max_media_count=5)"