# 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)