diff options
author | 2019-09-18 14:00:57 -0700 | |
---|---|---|
committer | 2019-09-18 14:10:06 -0700 | |
commit | 2ac0c6df978f20a488b3eb026753c8a972cb2554 (patch) | |
tree | 43fc70b14b517139dea5324e762ba92a9e211e28 /tests/test_pagination.py | |
parent | Docstring linting chunk 7 (diff) | |
parent | Merge pull request #436 from python-discord/enhance-offtopicnames-search (diff) |
Merge branch 'master' into flake8-plugins
Diffstat (limited to 'tests/test_pagination.py')
-rw-r--r-- | tests/test_pagination.py | 29 |
1 files changed, 29 insertions, 0 deletions
diff --git a/tests/test_pagination.py b/tests/test_pagination.py new file mode 100644 index 000000000..11d6541ae --- /dev/null +++ b/tests/test_pagination.py @@ -0,0 +1,29 @@ +from unittest import TestCase + +import pytest + +from bot import pagination + + +class LinePaginatorTests(TestCase): + def setUp(self): + self.paginator = pagination.LinePaginator(prefix='', suffix='', max_size=30) + + def test_add_line_raises_on_too_long_lines(self): + message = f"Line exceeds maximum page size {self.paginator.max_size - 2}" + with pytest.raises(RuntimeError, match=message): + self.paginator.add_line('x' * self.paginator.max_size) + + def test_add_line_works_on_small_lines(self): + self.paginator.add_line('x' * (self.paginator.max_size - 3)) + + +class ImagePaginatorTests(TestCase): + def setUp(self): + self.paginator = pagination.ImagePaginator() + + def test_add_image_appends_image(self): + image = 'lemon' + self.paginator.add_image(image) + + assert self.paginator.images == [image] |