mirror of
https://github.com/espressif/esp-idf.git
synced 2024-10-05 20:47:46 -04:00
Merge branch 'fix/remove_monkeypatch_in_flash_encryption_example' into 'master'
flash_encryption_example: Remove unnecessary monkeypatch See merge request espressif/esp-idf!18583
This commit is contained in:
commit
9f72659da8
@ -1,45 +0,0 @@
|
|||||||
# SPDX-FileCopyrightText: 2022 Espressif Systems (Shanghai) CO LTD
|
|
||||||
# SPDX-License-Identifier: Apache-2.0
|
|
||||||
|
|
||||||
import logging
|
|
||||||
|
|
||||||
import pytest
|
|
||||||
from _pytest.fixtures import FixtureRequest
|
|
||||||
from _pytest.monkeypatch import MonkeyPatch
|
|
||||||
from pytest_embedded_idf.serial import IdfSerial
|
|
||||||
|
|
||||||
|
|
||||||
# This is a custom Serial Class to add the erase_flash functionality
|
|
||||||
class FlashEncSerial(IdfSerial):
|
|
||||||
|
|
||||||
@IdfSerial.use_esptool
|
|
||||||
def erase_partition(self, partition_name: str) -> None:
|
|
||||||
|
|
||||||
if partition_name is None:
|
|
||||||
logging.error('Invalid arguments')
|
|
||||||
return
|
|
||||||
|
|
||||||
if not self.app.partition_table:
|
|
||||||
logging.error('Partition table not parsed.')
|
|
||||||
return
|
|
||||||
|
|
||||||
if partition_name in self.app.partition_table:
|
|
||||||
address = self.app.partition_table[partition_name]['offset']
|
|
||||||
size = self.app.partition_table[partition_name]['size']
|
|
||||||
logging.info('Erasing the partition {0} of size {1} at {2}'.format(partition_name, size, address))
|
|
||||||
self.stub.erase_region(address, size)
|
|
||||||
else:
|
|
||||||
logging.error('partition name {0} not found in app partition table'.format(partition_name))
|
|
||||||
return
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='module')
|
|
||||||
def monkeypatch_module(request: FixtureRequest) -> MonkeyPatch:
|
|
||||||
mp = MonkeyPatch()
|
|
||||||
request.addfinalizer(mp.undo)
|
|
||||||
return mp
|
|
||||||
|
|
||||||
|
|
||||||
@pytest.fixture(scope='module', autouse=True)
|
|
||||||
def replace_dut_class(monkeypatch_module: MonkeyPatch) -> None:
|
|
||||||
monkeypatch_module.setattr('pytest_embedded_idf.serial.IdfSerial', FlashEncSerial)
|
|
Loading…
x
Reference in New Issue
Block a user