From a685743c74a159632616ff3f478cf225d976431c Mon Sep 17 00:00:00 2001
From: Adam Velebil <adam.velebil@yubico.com>
Date: Fri, 9 Feb 2024 14:52:21 +0100
Subject: [PATCH] adds missing keys to ButtonSegments

---
 lib/app/views/reset_dialog.dart | 10 +++++++++-
 1 file changed, 9 insertions(+), 1 deletion(-)

diff --git a/lib/app/views/reset_dialog.dart b/lib/app/views/reset_dialog.dart
index f93cb820..e7ce016d 100644
--- a/lib/app/views/reset_dialog.dart
+++ b/lib/app/views/reset_dialog.dart
@@ -208,7 +208,15 @@ class _ResetDialogState extends ConsumerState<ResetDialog> {
                       .where((c) => supported & c.value != 0)
                       .map((c) => ButtonSegment(
                             value: c,
-                            icon: Icon(c._icon),
+                            icon: Icon(
+                              c._icon,
+                              key: switch (c) {
+                                Capability.oath => factoryResetPickResetOath,
+                                Capability.fido2 => factoryResetPickResetFido2,
+                                Capability.piv => factoryResetPickResetPiv,
+                                _ => const Key('_invalid') // no reset
+                              },
+                            ),
                             label: showLabels
                                 ? Text(c.getDisplayName(l10n))
                                 : null,