mirror of
https://github.com/jazzband/django-downloadview.git
synced 2026-03-16 22:40:25 +00:00
53 lines
1.6 KiB
Python
53 lines
1.6 KiB
Python
# coding=utf-8
|
|
"""Tests around :mod:`django_downloadview.io`."""
|
|
import unittest
|
|
|
|
from django_downloadview import TextIteratorIO, BytesIteratorIO
|
|
|
|
|
|
HELLO_TEXT = u'Hello world!\né\n'
|
|
HELLO_BYTES = b'Hello world!\n\xc3\xa9\n'
|
|
|
|
|
|
def generate_hello_text():
|
|
"""Generate u'Hello world!\n'."""
|
|
yield u'Hello '
|
|
yield u'world!'
|
|
yield u'\n'
|
|
yield u'é'
|
|
yield u'\n'
|
|
|
|
|
|
def generate_hello_bytes():
|
|
"""Generate b'Hello world!\n'."""
|
|
yield b'Hello '
|
|
yield b'world!'
|
|
yield b'\n'
|
|
yield b'\xc3\xa9'
|
|
yield b'\n'
|
|
|
|
|
|
class TextIteratorIOTestCase(unittest.TestCase):
|
|
"""Tests around :class:`~django_downloadview.io.TextIteratorIO`."""
|
|
def test_read_text(self):
|
|
"""TextIteratorIO obviously accepts text generator."""
|
|
file_obj = TextIteratorIO(generate_hello_text())
|
|
self.assertEqual(file_obj.read(), HELLO_TEXT)
|
|
|
|
def test_read_bytes(self):
|
|
"""TextIteratorIO converts bytes as text."""
|
|
file_obj = TextIteratorIO(generate_hello_bytes())
|
|
self.assertEqual(file_obj.read(), HELLO_TEXT)
|
|
|
|
|
|
class BytesIteratorIOTestCase(unittest.TestCase):
|
|
"""Tests around :class:`~django_downloadview.io.BytesIteratorIO`."""
|
|
def test_read_bytes(self):
|
|
"""BytesIteratorIO obviously accepts bytes generator."""
|
|
file_obj = BytesIteratorIO(generate_hello_bytes())
|
|
self.assertEqual(file_obj.read(), HELLO_BYTES)
|
|
|
|
def test_read_text(self):
|
|
"""BytesIteratorIO converts text as bytes."""
|
|
file_obj = BytesIteratorIO(generate_hello_text())
|
|
self.assertEqual(file_obj.read(), HELLO_BYTES)
|