From 7650912e1581ccf44aabbe0685bca906c0bc22e5 Mon Sep 17 00:00:00 2001
From: belono <tiotil.lindeman@gmail.com>
Date: Sun, 22 Oct 2023 14:39:30 +0200
Subject: [PATCH] Add test_printer_usb.py

---
 test/test_printers/test_printer_usb.py | 106 +++++++++++++++++++++++++
 1 file changed, 106 insertions(+)
 create mode 100644 test/test_printers/test_printer_usb.py

diff --git a/test/test_printers/test_printer_usb.py b/test/test_printers/test_printer_usb.py
new file mode 100644
index 0000000..28d3594
--- /dev/null
+++ b/test/test_printers/test_printer_usb.py
@@ -0,0 +1,106 @@
+#!/usr/bin/python
+#  -*- coding: utf-8 -*-
+"""tests for the Usb printer
+
+:author: Benito López and the python-escpos developers
+:organization: `python-escpos <https://github.com/python-escpos>`_
+:copyright: Copyright (c) 2023 `python-escpos <https://github.com/python-escpos>`_
+:license: MIT
+"""
+
+import logging
+
+# import pytest
+
+
+def test_device_not_initialized(usbprinter):
+    """
+    GIVEN a usb printer object
+    WHEN it is not initialized
+    THEN check the device property is False
+    """
+    assert usbprinter._device is False
+
+
+def test_open_raise_exception(usbprinter, devicenotfounderror, mocker):
+    """
+    # GIVEN a usb printer object
+    GIVEN a mocked usb printer object
+    WHEN open() is set to raise a DeviceNotFoundError on error
+    # THEN check the exception is raised
+    THEN check the param is True
+    """
+    mocker.patch("usb.core.find")
+    spy = mocker.spy(usbprinter, "open")
+    # usbprinter.usb_args = {"idVendor": 0, "idProduct": 0}
+
+    # with pytest.raises(devicenotfounderror):
+    usbprinter.open(raise_not_found=True)
+    spy.assert_called_once_with(raise_not_found=True)
+
+
+def test_open_not_raise_exception(usbprinter, caplog, mocker):
+    """
+    # GIVEN a usb printer object
+    GIVEN a mocked usb printer object
+    WHEN open() is set to not raise on error but simply cancel
+    # THEN check the error is logged and open() canceled
+    THEN check the param is False
+    """
+    mocker.patch("usb.core.find")
+    spy = mocker.spy(usbprinter, "open")
+    # usbprinter.usb_args = {"idVendor": 0, "idProduct": 0}
+
+    # with caplog.at_level(logging.ERROR):
+    usbprinter.open(raise_not_found=False)
+
+    # assert "not found" in caplog.text
+    # assert usbprinter.device is None
+    spy.assert_called_once_with(raise_not_found=False)
+
+
+def test_open(usbprinter, caplog, mocker):
+    """
+    GIVEN a usb printer object and a mocked pyusb device
+    WHEN a valid connection to a device is opened
+    THEN check the success is logged and the device property is set
+    """
+    mocker.patch("usb.core.find")
+
+    with caplog.at_level(logging.INFO):
+        usbprinter.open()
+
+    assert "enabled" in caplog.text
+    assert usbprinter.device
+
+
+def test_close_on_reopen(usbprinter, mocker):
+    """
+    GIVEN a usb printer object and a mocked connection
+    WHEN a valid connection to a device is reopened before close
+    THEN check the close method is called if _device
+    """
+    mocker.patch("usb.core.find")
+    spy = mocker.spy(usbprinter, "close")
+
+    usbprinter.open()
+    assert usbprinter._device
+
+    usbprinter.open()
+    spy.assert_called_once_with()
+
+
+def test_close(usbprinter, caplog, mocker):
+    """
+    GIVEN a usb printer object and a mocked pyusb device
+    WHEN a connection is opened and closed
+    THEN check the closing is logged and the device property is False
+    """
+    mocker.patch("usb.core.find")
+    usbprinter.open()
+
+    with caplog.at_level(logging.INFO):
+        usbprinter.close()
+
+    assert "Closing" in caplog.text
+    assert usbprinter._device is False