From ba77f15690a9c4d54eb84cf4d7c87fd93c6f4ef1 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Jan=20Procha=CC=81zka?=
 <90197375+P-R-O-C-H-Y@users.noreply.github.com>
Date: Fri, 7 Jul 2023 12:20:41 +0200
Subject: [PATCH] fix(arduino): separated flash settings for each target

---
 pytest-embedded-arduino/pytest_embedded_arduino/app.py   | 9 ++++++++-
 .../pytest_embedded_arduino/serial.py                    | 2 +-
 2 files changed, 9 insertions(+), 2 deletions(-)

diff --git a/pytest-embedded-arduino/pytest_embedded_arduino/app.py b/pytest-embedded-arduino/pytest_embedded_arduino/app.py
index d9c1e6fe..975b1147 100644
--- a/pytest-embedded-arduino/pytest_embedded_arduino/app.py
+++ b/pytest-embedded-arduino/pytest_embedded_arduino/app.py
@@ -17,7 +17,14 @@ class ArduinoApp(App):
     """
 
     #: dict of flash settings
-    flash_settings = {'flash_mode': 'dio', 'flash_size': 'detect', 'flash_freq': '80m'}
+    flash_settings = {
+        'esp32': {'flash_mode': 'dio', 'flash_size': 'detect', 'flash_freq': '80m'},
+        'esp32s2': {'flash_mode': 'dio', 'flash_size': 'detect', 'flash_freq': '80m'},
+        'esp32c3': {'flash_mode': 'dio', 'flash_size': 'detect', 'flash_freq': '80m'},
+        'esp32s3': {'flash_mode': 'dio', 'flash_size': 'detect', 'flash_freq': '80m'},
+        'esp32c6': {'flash_mode': 'dio', 'flash_size': 'detect', 'flash_freq': '80m'},
+        'esp32h2': {'flash_mode': 'dio', 'flash_size': 'detect', 'flash_freq': '48m'},
+    }
 
     #: dict of binaries' offset.
     binary_offsets = {
diff --git a/pytest-embedded-arduino/pytest_embedded_arduino/serial.py b/pytest-embedded-arduino/pytest_embedded_arduino/serial.py
index 22658943..b5761a2b 100644
--- a/pytest-embedded-arduino/pytest_embedded_arduino/serial.py
+++ b/pytest-embedded-arduino/pytest_embedded_arduino/serial.py
@@ -57,7 +57,7 @@ def flash(self) -> None:
             'chip': self.app.target,
         }
 
-        default_kwargs.update(self.app.flash_settings)
+        default_kwargs.update(self.app.flash_settings[self.app.target])
         flash_args = EsptoolArgs(**default_kwargs)
 
         try: