From b46b3f6e98bbbfb01bfc82316ef0f86c901f7f41 Mon Sep 17 00:00:00 2001
From: Sebastian Malton <sebastian@malton.name>
Date: Tue, 16 Aug 2022 15:03:40 -0400
Subject: [PATCH] Add data-warning prop on button for better snapshot testing

Signed-off-by: Sebastian Malton <sebastian@malton.name>
---
 .../navigation-using-application-menu.test.tsx.snap         | 1 +
 .../navigation-using-application-menu.test.ts.snap          | 1 +
 .../add-custom-helm-repository-in-preferences.test.ts.snap  | 1 +
 .../showing-details-for-helm-release.test.ts.snap           | 6 ++++++
 src/renderer/components/button/button.tsx                   | 6 +++++-
 .../__snapshots__/kube-object-menu.test.tsx.snap            | 4 ++++
 6 files changed, 18 insertions(+), 1 deletion(-)

diff --git a/src/features/add-cluster/__snapshots__/navigation-using-application-menu.test.tsx.snap b/src/features/add-cluster/__snapshots__/navigation-using-application-menu.test.tsx.snap
index 991a77480a..87957f581a 100644
--- a/src/features/add-cluster/__snapshots__/navigation-using-application-menu.test.tsx.snap
+++ b/src/features/add-cluster/__snapshots__/navigation-using-application-menu.test.tsx.snap
@@ -298,6 +298,7 @@ exports[`add-cluster - navigation using application menu when navigating to add
             >
               <button
                 class="Button primary"
+                data-waiting="false"
                 disabled=""
                 type="button"
               >
diff --git a/src/features/extensions/__snapshots__/navigation-using-application-menu.test.ts.snap b/src/features/extensions/__snapshots__/navigation-using-application-menu.test.ts.snap
index c3d9431eb2..118c5b9cc3 100644
--- a/src/features/extensions/__snapshots__/navigation-using-application-menu.test.ts.snap
+++ b/src/features/extensions/__snapshots__/navigation-using-application-menu.test.ts.snap
@@ -306,6 +306,7 @@ exports[`extensions - navigation using application menu when navigating to exten
                   <div>
                     <button
                       class="Button button primary"
+                      data-waiting="false"
                       type="button"
                     >
                       Install
diff --git a/src/features/helm-charts/__snapshots__/add-custom-helm-repository-in-preferences.test.ts.snap b/src/features/helm-charts/__snapshots__/add-custom-helm-repository-in-preferences.test.ts.snap
index 8f83091fcc..c02c727480 100644
--- a/src/features/helm-charts/__snapshots__/add-custom-helm-repository-in-preferences.test.ts.snap
+++ b/src/features/helm-charts/__snapshots__/add-custom-helm-repository-in-preferences.test.ts.snap
@@ -6316,6 +6316,7 @@ exports[`add custom helm repository in preferences when navigating to preference
             <button
               class="Button waiting primary"
               data-testid="custom-helm-repository-submit-button"
+              data-waiting="true"
               type="submit"
             >
               Add
diff --git a/src/features/helm-releases/__snapshots__/showing-details-for-helm-release.test.ts.snap b/src/features/helm-releases/__snapshots__/showing-details-for-helm-release.test.ts.snap
index ae169ab2f4..b717949f66 100644
--- a/src/features/helm-releases/__snapshots__/showing-details-for-helm-release.test.ts.snap
+++ b/src/features/helm-releases/__snapshots__/showing-details-for-helm-release.test.ts.snap
@@ -7430,6 +7430,7 @@ exports[`showing details for helm release given application is started when navi
               <button
                 class="Button primary"
                 data-testid="helm-release-configuration-save-button"
+                data-waiting="false"
                 type="button"
               >
                 Save
@@ -8668,6 +8669,7 @@ exports[`showing details for helm release given application is started when navi
               <button
                 class="Button primary"
                 data-testid="helm-release-configuration-save-button"
+                data-waiting="false"
                 type="button"
               >
                 Save
@@ -9906,6 +9908,7 @@ exports[`showing details for helm release given application is started when navi
               <button
                 class="Button waiting primary"
                 data-testid="helm-release-configuration-save-button"
+                data-waiting="true"
                 type="button"
               >
                 Save
@@ -10971,6 +10974,7 @@ exports[`showing details for helm release given application is started when navi
               <button
                 class="Button primary"
                 data-testid="helm-release-configuration-save-button"
+                data-waiting="false"
                 type="button"
               >
                 Save
@@ -12037,6 +12041,7 @@ exports[`showing details for helm release given application is started when navi
               <button
                 class="Button primary"
                 data-testid="helm-release-configuration-save-button"
+                data-waiting="false"
                 disabled=""
                 type="button"
               >
@@ -13276,6 +13281,7 @@ exports[`showing details for helm release given application is started when navi
               <button
                 class="Button primary"
                 data-testid="helm-release-configuration-save-button"
+                data-waiting="false"
                 type="button"
               >
                 Save
diff --git a/src/renderer/components/button/button.tsx b/src/renderer/components/button/button.tsx
index 756f50a584..8168c2abf3 100644
--- a/src/renderer/components/button/button.tsx
+++ b/src/renderer/components/button/button.tsx
@@ -49,7 +49,11 @@ export const Button = withTooltip((props: ButtonProps) => {
 
   // render as button
   return (
-    <button type="button" {...btnProps}>
+    <button
+      type="button"
+      {...btnProps}
+      data-waiting={typeof waiting === "boolean" ? String(waiting) : undefined}
+    >
       {label}
       {children}
     </button>
diff --git a/src/renderer/components/kube-object-menu/__snapshots__/kube-object-menu.test.tsx.snap b/src/renderer/components/kube-object-menu/__snapshots__/kube-object-menu.test.tsx.snap
index 799e841e5d..053e08a809 100644
--- a/src/renderer/components/kube-object-menu/__snapshots__/kube-object-menu.test.tsx.snap
+++ b/src/renderer/components/kube-object-menu/__snapshots__/kube-object-menu.test.tsx.snap
@@ -122,6 +122,7 @@ exports[`kube-object-menu given kube object when removing kube object renders 1`
         <button
           class="Button ok primary"
           data-testid="confirm"
+          data-waiting="false"
           type="button"
         >
           Remove
@@ -248,6 +249,7 @@ exports[`kube-object-menu given kube object when rerendered with different kube
         <button
           class="Button ok primary"
           data-testid="confirm"
+          data-waiting="false"
           type="button"
         >
           Remove
@@ -340,6 +342,7 @@ exports[`kube-object-menu given kube object with namespace when removing kube ob
         <button
           class="Button ok primary"
           data-testid="confirm"
+          data-waiting="false"
           type="button"
         >
           Remove
@@ -432,6 +435,7 @@ exports[`kube-object-menu given kube object without namespace when removing kube
         <button
           class="Button ok primary"
           data-testid="confirm"
+          data-waiting="false"
           type="button"
         >
           Remove