diff --git a/pytest-embedded-serial-esp/pytest_embedded_serial_esp/serial.py b/pytest-embedded-serial-esp/pytest_embedded_serial_esp/serial.py index 11e3710f..6f9056bd 100644 --- a/pytest-embedded-serial-esp/pytest_embedded_serial_esp/serial.py +++ b/pytest-embedded-serial-esp/pytest_embedded_serial_esp/serial.py @@ -101,7 +101,7 @@ def __init__( ) with contextlib.redirect_stdout(msg_queue): - esp = esptool.get_default_connected_device( + self.esp = esptool.get_default_connected_device( ports, port=port, connect_attempts=3, @@ -109,14 +109,14 @@ def __init__( chip=esptool_target, ) - if not esp: + if not self.esp: raise ValueError('Couldn\'t auto detect chip. Please manually specify with "--port"') self.esp: esptool.ESPLoader = None # type: ignore self.stub: esptool.ESPLoader = None # type: ignore - target = esp.CHIP_NAME.lower().replace('-', '') - logging.info('Target: %s, Port: %s', target, esp.serial_port) + target = self.esp.CHIP_NAME.lower().replace('-', '') + logging.info('Target: %s, Port: %s', target, self.esp.serial_port) self.target = target @@ -124,7 +124,7 @@ def __init__( self.erase_all = erase_all self.esptool_baud = esptool_baud - super().__init__(msg_queue=msg_queue, port=esp._port, baud=baud, meta=meta, **kwargs) + super().__init__(msg_queue=msg_queue, port=self.esp._port, baud=baud, meta=meta, **kwargs) def _post_init(self): if self._meta: @@ -157,8 +157,6 @@ def decorator(func): def wrapper(self, *args, **kwargs): with self.disable_redirect_thread(): with contextlib.redirect_stdout(self._q): - self.esp = esptool.detect_chip(self.proc, self.baud) - self.esp.connect('hard_reset') ret = func(self, *args, **kwargs) return ret