From a59b0b560d17a834e08cffcacff0910c1c2f87ba Mon Sep 17 00:00:00 2001
From: Ryan Liang <jiallian@amazon.com>
Date: Tue, 12 Mar 2024 17:11:20 -0700
Subject: [PATCH] Implement the refresh button

Signed-off-by: Ryan Liang <jiallian@amazon.com>
---
 .../__snapshots__/data_connection.test.tsx.snap     |  6 +++++-
 .../manage/accelerations/acceleration_table.tsx     | 13 +++++++++----
 2 files changed, 14 insertions(+), 5 deletions(-)

diff --git a/public/components/datasources/components/__tests__/__snapshots__/data_connection.test.tsx.snap b/public/components/datasources/components/__tests__/__snapshots__/data_connection.test.tsx.snap
index f6dfdb125e..562e73c31c 100644
--- a/public/components/datasources/components/__tests__/__snapshots__/data_connection.test.tsx.snap
+++ b/public/components/datasources/components/__tests__/__snapshots__/data_connection.test.tsx.snap
@@ -304,12 +304,16 @@ exports[`Data Connection Page test Renders Prometheus data connection page with
                 class="euiFlexItem euiFlexItem--flexGrowZero"
               >
                 <button
-                  class="euiButton euiButton--primary"
+                  class="euiButton euiButton--primary euiButton-isDisabled"
+                  disabled=""
                   type="button"
                 >
                   <span
                     class="euiButtonContent euiButton__content"
                   >
+                    <span
+                      class="euiLoadingSpinner euiLoadingSpinner--medium euiButtonContent__spinner"
+                    />
                     <span
                       class="euiButton__text"
                     >
diff --git a/public/components/datasources/components/manage/accelerations/acceleration_table.tsx b/public/components/datasources/components/manage/accelerations/acceleration_table.tsx
index 06e6de6881..8761f500ea 100644
--- a/public/components/datasources/components/manage/accelerations/acceleration_table.tsx
+++ b/public/components/datasources/components/manage/accelerations/acceleration_table.tsx
@@ -69,12 +69,17 @@ export const AccelerationTable = () => {
     }
   }, [loadStatus]);
 
+  const handleRefresh = () => {
+    console.log('Initiating refresh...');
+    setIsRefreshing(true);
+    loadAccelerations('mys3');
+  };
+
   const RefreshButton = () => {
-    // TODO: Implement logic for refreshing acceleration
     return (
-      <>
-        <EuiButton onClick={() => console.log('clicked on refresh button')}>Refresh</EuiButton>
-      </>
+      <EuiButton onClick={handleRefresh} isLoading={isRefreshing}>
+        Refresh
+      </EuiButton>
     );
   };