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