python-escpos/test/test_function_text.py

#!/usr/bin/python
"""tests for the text printing function

:author: `Patrick Kanzler <patrick.kanzler@fablab.fau.de>`_
:organization: `python-escpos <https://github.com/python-escpos>`_
:copyright: Copyright (c) 2016 `python-escpos <https://github.com/python-escpos>`_
:license: MIT
"""


import pytest
import mock
from hypothesis import given, assume
import hypothesis.strategies as st
from escpos.printer import Dummy


def get_printer():
    return Dummy(magic_encode_args={'disabled': True, 'encoding': 'CP437'})


@given(text=st.text())
def test_text(text):
    """Test that text() calls the MagicEncode object.
    """
    instance = get_printer()
    instance.magic.write = mock.Mock()
    instance.text(text)
    instance.magic.write.assert_called_with(text)


def test_block_text():
    printer = get_printer()
    printer.block_text(
        "All the presidents men were eating falafel for breakfast.", font='a')
    assert printer.output == \
        b'All the presidents men were eating falafel\nfor breakfast.'


def test_textln():
    printer = get_printer()
    printer.textln('hello, world')
    assert printer.output == b'hello, world\n'


def test_textln_empty():
    printer = get_printer()
    printer.textln()
    assert printer.output == b'\n'


def test_ln():
    printer = get_printer()
    printer.ln()
    assert printer.output == b'\n'


def test_multiple_ln():
    printer = get_printer()
    printer.ln(3)
    assert printer.output == b'\n\n\n'