mirror of
https://github.com/espressif/esp-idf.git
synced 2024-10-05 20:47:46 -04:00
127 lines
4.5 KiB
Python
Executable File
127 lines
4.5 KiB
Python
Executable File
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# SPDX-FileCopyrightText: 2020-2022 Espressif Systems (Shanghai) CO LTD
|
|
# SPDX-License-Identifier: Apache-2.0
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import collections
|
|
import filecmp
|
|
import json
|
|
import os
|
|
import shutil
|
|
import sys
|
|
import tempfile
|
|
import time
|
|
import unittest
|
|
|
|
import pexpect
|
|
|
|
current_dir = os.path.dirname(os.path.realpath(__file__))
|
|
mkdfu_path = os.path.join(current_dir, '..', 'mkdfu.py')
|
|
|
|
|
|
class TestMkDFU(unittest.TestCase):
|
|
def common_test(self, json_input=None, file_args=[], output_to_compare=None, part_size=None):
|
|
'''
|
|
- json_input - input JSON file compatible with mkdfu.py - used when not None
|
|
- file_args - list of (address, path_to_file) tuples
|
|
- output_to_compare - path to the file containing the expected output - tested when not None
|
|
- part_size - partition size - used when not None
|
|
'''
|
|
with tempfile.NamedTemporaryFile(delete=False) as f_out:
|
|
self.addCleanup(os.unlink, f_out.name)
|
|
args = [mkdfu_path, 'write',
|
|
'-o', f_out.name,
|
|
'--pid', '2',
|
|
'--flash-size', '4MB']
|
|
if part_size:
|
|
args += ['--part-size', str(part_size)]
|
|
if json_input:
|
|
args += ['--json', json_input]
|
|
for addr, f_path in file_args:
|
|
args += [str(addr), f_path]
|
|
p = pexpect.spawn(sys.executable, args, timeout=10, encoding='utf-8')
|
|
self.addCleanup(p.terminate, force=True)
|
|
|
|
p.expect_exact('Adding flash chip parameters file with flash_size = 4MB')
|
|
|
|
for addr, f_path in sorted(file_args, key=lambda e: e[0]):
|
|
p.expect_exact('Adding {} at {}'.format(f_path, hex(addr)))
|
|
|
|
p.expect_exact('"{}" has been written. You may proceed with DFU flashing.'.format(f_out.name))
|
|
|
|
# Need to wait for the process to end because the output file is closed when mkdfu exits.
|
|
# Do non-blocking wait instead of the blocking p.wait():
|
|
for _ in range(10):
|
|
if not p.isalive():
|
|
break
|
|
time.sleep(0.5)
|
|
else:
|
|
p.terminate()
|
|
|
|
if output_to_compare:
|
|
self.assertTrue(filecmp.cmp(f_out.name, os.path.join(current_dir, output_to_compare)), 'Output files are different')
|
|
|
|
|
|
class TestHelloWorldExample(TestMkDFU):
|
|
'''
|
|
tests with images prepared in the "1" subdirectory
|
|
'''
|
|
def test_with_json(self):
|
|
with tempfile.NamedTemporaryFile(mode='w', dir=os.path.join(current_dir, '1'), delete=False) as f:
|
|
self.addCleanup(os.unlink, f.name)
|
|
|
|
bins = [('0x1000', '1.bin'), ('0x8000', '2.bin'), ('0x10000', '3.bin')]
|
|
json.dump({'flash_files': collections.OrderedDict(bins)}, f)
|
|
|
|
self.common_test(json_input=f.name, output_to_compare='1/dfu.bin')
|
|
|
|
def test_without_json(self):
|
|
|
|
self.common_test(file_args=[(0x1000, '1/1.bin'),
|
|
(0x8000, '1/2.bin'),
|
|
(0x10000, '1/3.bin')],
|
|
output_to_compare='1/dfu.bin')
|
|
|
|
def test_filenames(self):
|
|
temp_dir = tempfile.mkdtemp(prefix='very_long_directory_name' * 8)
|
|
self.addCleanup(shutil.rmtree, temp_dir, ignore_errors=True)
|
|
|
|
with tempfile.NamedTemporaryFile(prefix='ľščťžýáíéěř\u0420\u043e\u0441\u0441\u0438\u044f',
|
|
dir=temp_dir,
|
|
delete=False) as f:
|
|
bootloader = f.name
|
|
|
|
shutil.copyfile(os.path.join(current_dir, '1', '1.bin'), bootloader)
|
|
|
|
self.common_test(file_args=[(0x1000, bootloader),
|
|
(0x8000, os.path.join(current_dir, '1', '2.bin')),
|
|
(0x10000, os.path.join(current_dir, '1', '3.bin'))])
|
|
|
|
|
|
class TestSplit(TestMkDFU):
|
|
'''
|
|
tests with images prepared in the "2" subdirectory
|
|
|
|
"2/dfu.bin" was prepared with:
|
|
mkdfu.py write --part-size 5 --pid 2 --flash-size 4MB -o 2/dfu.bin 0 bin
|
|
where the content of "bin" is b"\xce" * 10
|
|
'''
|
|
def test_split(self):
|
|
temp_dir = tempfile.mkdtemp(dir=current_dir)
|
|
self.addCleanup(shutil.rmtree, temp_dir, ignore_errors=True)
|
|
|
|
with open(os.path.join(temp_dir, 'bin'), 'wb') as f:
|
|
self.addCleanup(os.unlink, f.name)
|
|
f.write(b'\xce' * 10)
|
|
|
|
self.common_test(file_args=[(0, f.name)],
|
|
part_size=5,
|
|
output_to_compare='2/dfu.bin')
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|