456f5b7aa6
* add a method to check barcode code format ensure that the code to print is compatible with the ESC/POS formats and also automatically check this format before printing (barcode() method). * rewrite test using pytest's parametrize functionality * add test for the 'check' argument * update authors list
53 lines
1.5 KiB
Python
53 lines
1.5 KiB
Python
#!/usr/bin/python
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
from __future__ import unicode_literals
|
|
|
|
import escpos.printer as printer
|
|
from escpos.constants import BARCODE_TYPE_A, BARCODE_TYPE_B
|
|
from escpos.capabilities import Profile, BARCODE_B
|
|
from escpos.exceptions import BarcodeTypeError, BarcodeCodeError
|
|
import pytest
|
|
|
|
|
|
@pytest.mark.parametrize("bctype,data,expected", [
|
|
('EAN13', '4006381333931',
|
|
b'\x1ba\x01\x1dh@\x1dw\x03\x1df\x00\x1dH\x02\x1dk\x024006381333931\x00')
|
|
])
|
|
def test_barcode(bctype, data, expected):
|
|
"""should generate different barcode types correctly.
|
|
"""
|
|
instance = printer.Dummy()
|
|
instance.barcode(data, bctype)
|
|
assert instance.output == expected
|
|
|
|
|
|
@pytest.mark.parametrize("bctype,supports_b", [
|
|
('invalid', True),
|
|
('CODE128', False),
|
|
])
|
|
def test_lacks_support(bctype, supports_b):
|
|
"""should raise an error if the barcode type is not supported.
|
|
"""
|
|
profile = Profile(features={BARCODE_B: supports_b})
|
|
instance = printer.Dummy(profile=profile)
|
|
with pytest.raises(BarcodeTypeError):
|
|
instance.barcode('test', bctype)
|
|
|
|
assert instance.output == b''
|
|
|
|
|
|
@pytest.mark.parametrize("bctype,data", [
|
|
('EAN13', 'AA'),
|
|
('CODE128', '{D2354AA'),
|
|
])
|
|
def test_code_check(bctype, data):
|
|
"""should raise an error if the barcode code is invalid.
|
|
"""
|
|
instance = printer.Dummy()
|
|
with pytest.raises(BarcodeCodeError):
|
|
instance.barcode(data, bctype)
|
|
|
|
assert instance.output == b''
|