From cf6d94e82bf5d93d22424cb45b8c12cdf09093c3 Mon Sep 17 00:00:00 2001
From: Peter Dragun <peter.dragun@espressif.com>
Date: Wed, 13 Mar 2024 14:28:13 +0100
Subject: [PATCH] feat: add UF2 IDs for ESP32-C5 and ESP32-C61

https://github.com/microsoft/uf2/pull/80
---
 esptool/targets/esp32c5.py  | 2 ++
 esptool/targets/esp32c61.py | 2 ++
 test/test_uf2_ids.py        | 2 +-
 3 files changed, 5 insertions(+), 1 deletion(-)

diff --git a/esptool/targets/esp32c5.py b/esptool/targets/esp32c5.py
index 802cc52ab..d4517af91 100644
--- a/esptool/targets/esp32c5.py
+++ b/esptool/targets/esp32c5.py
@@ -45,6 +45,8 @@ class ESP32C5ROM(ESP32C6ROM):
         [0x600FE000, 0x60100000, "MEM_INTERNAL2"],
     ]
 
+    UF2_FAMILY_ID = 0xF71C0343
+
     def get_chip_description(self):
         chip_name = {
             0: "ESP32-C5",
diff --git a/esptool/targets/esp32c61.py b/esptool/targets/esp32c61.py
index 96f8ff93e..58867a433 100644
--- a/esptool/targets/esp32c61.py
+++ b/esptool/targets/esp32c61.py
@@ -58,6 +58,8 @@ class ESP32C61ROM(ESP32C6ROM):
         [0x600FE000, 0x60100000, "MEM_INTERNAL2"],
     ]
 
+    UF2_FAMILY_ID = 0x77D850C4
+
     def get_chip_description(self):
         chip_name = {
             0: "ESP32-C61",
diff --git a/test/test_uf2_ids.py b/test/test_uf2_ids.py
index 1919013f7..8be38b91c 100644
--- a/test/test_uf2_ids.py
+++ b/test/test_uf2_ids.py
@@ -53,7 +53,7 @@ def test_check_uf2(uf2_json):
         out = []
         # there was a difference between the chip support
         for chip in diff:
-            if chip in esptool_chips:
+            if chip not in esptool_chips:
                 out.append(
                     f"Missing chip definition for '{chip}' in esptool "
                     "which was defined in Microsoft UF2 Github repo."