mirror of
https://github.com/lora-aprs/LoRa_APRS_iGate.git
synced 2025-12-06 07:42:00 +01:00
79 lines
2.6 KiB
Python
79 lines
2.6 KiB
Python
import os
|
|
import pytest
|
|
import serial
|
|
import logging
|
|
from HIL.common import runProcess
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class EspFlash:
|
|
def __init__(self, port):
|
|
self.pio_package_path = "$HOME/.platformio/packages"
|
|
self.port = port
|
|
logger.info(f"pio package path: {self.pio_package_path}")
|
|
logger.info(f"port: {self.port}")
|
|
|
|
def runESPTool(self, cmd):
|
|
runProcess(
|
|
f"/usr/bin/python3 {self.pio_package_path}/tool-esptoolpy/esptool.py --chip esp32 --port {self.port} {cmd}")
|
|
|
|
def erase(self):
|
|
logger.info("erase flash")
|
|
self.runESPTool("erase_flash")
|
|
|
|
def write(self, addr, bin_file):
|
|
logger.info(f"write flash on addr {addr} with file {bin_file}")
|
|
self.runESPTool(
|
|
f"--baud 460800 --before default_reset --after hard_reset write_flash -z --flash_mode dio --flash_size detect {addr} {bin_file}")
|
|
|
|
def verify(self, addr, bin_file):
|
|
logger.info(f"verify flash on addr {addr} with file {bin_file}")
|
|
self.runESPTool(
|
|
f"--baud 460800 --before default_reset --after hard_reset verify_flash --flash_mode dio --flash_size detect {addr} {bin_file}")
|
|
|
|
def make_spiffs(self, fs_path, fs_bin):
|
|
logger.info(f"make spiffs, fs_path: {fs_path}, fs_bin: {fs_bin}")
|
|
runProcess(
|
|
f"{self.pio_package_path}/tool-mkspiffs/mkspiffs_espressif32_arduino -c {fs_path} -p 256 -b 4096 -s 1507328 {fs_bin}")
|
|
|
|
|
|
class EspDut:
|
|
def __init__(self, port):
|
|
self.port = port
|
|
self.baudrate = 115200
|
|
self.serial = None
|
|
self.flash = EspFlash(self.port)
|
|
|
|
def writeFlash(self, bin_dir):
|
|
logger.info("write flash")
|
|
self.flash.erase()
|
|
self.flash.write("0x1000", f"{bin_dir}/bootloader_dio_40m.bin")
|
|
self.flash.write("0x8000", f"{bin_dir}/partitions.bin")
|
|
self.flash.write("0xe000", f"{bin_dir}/boot_app0.bin")
|
|
self.flash.write("0x10000", f"{bin_dir}/firmware.bin")
|
|
|
|
def writeConfig(self, fs_path):
|
|
logger.info("write config")
|
|
fs_bin = "spiffs.bin"
|
|
self.flash.make_spiffs(fs_path, fs_bin)
|
|
self.flash.write("0x290000", fs_bin)
|
|
|
|
def openPort(self):
|
|
logger.info(f"open port: {self.port}, baudrate: {self.baudrate}")
|
|
self.serial = serial.Serial(
|
|
self.port, baudrate=self.baudrate, timeout=0)
|
|
|
|
def getLine(self):
|
|
return self.serial.readline()
|
|
|
|
def closePort(self):
|
|
if self.serial:
|
|
self.serial.close()
|
|
self.serial = None
|
|
|
|
|
|
@pytest.fixture
|
|
def ESP():
|
|
return EspDut(os.environ["ESP_PORT"])
|