From 3ca3b1c8230f1909957357fca8537d4e417f79de Mon Sep 17 00:00:00 2001
From: RachelElysia <71795832+RachelElysia@users.noreply.github.com>
Date: Wed, 13 Oct 2021 11:30:01 -0400
Subject: [PATCH] Manage Host Page (Bug fix): Transfer and delete hosts updates
 total host count (#2501)

* refetchHosts includes retrieveHosts and retrieveHostCount
---
 changes/fix-2501-transfer-delete-updates-host-count    |  1 +
 .../pages/hosts/ManageHostsPage/ManageHostsPage.tsx    | 10 ++++++++--
 2 files changed, 9 insertions(+), 2 deletions(-)
 create mode 100644 changes/fix-2501-transfer-delete-updates-host-count

diff --git a/changes/fix-2501-transfer-delete-updates-host-count b/changes/fix-2501-transfer-delete-updates-host-count
new file mode 100644
index 0000000000..3d06b4b7ca
--- /dev/null
+++ b/changes/fix-2501-transfer-delete-updates-host-count
@@ -0,0 +1 @@
+* Bug fix: Transferring and deleting hosts updates host count
\ No newline at end of file
diff --git a/frontend/pages/hosts/ManageHostsPage/ManageHostsPage.tsx b/frontend/pages/hosts/ManageHostsPage/ManageHostsPage.tsx
index 61a7b29557..6b90facc7d 100644
--- a/frontend/pages/hosts/ManageHostsPage/ManageHostsPage.tsx
+++ b/frontend/pages/hosts/ManageHostsPage/ManageHostsPage.tsx
@@ -332,6 +332,11 @@ const ManageHostsPage = ({
     }
   };
 
+  const refetchHosts = (options: IHostCountLoadOptions) => {
+    retrieveHosts(options);
+    retrieveHostCount(options);
+  };
+
   // triggered every time the route is changed
   // which means every filter click and text search
   useDeepEffect(() => {
@@ -743,7 +748,7 @@ const ManageHostsPage = ({
           : `Hosts successfully transferred to  ${team.name}.`;
 
       dispatch(renderFlash("success", successMessage));
-      retrieveHosts({
+      refetchHosts({
         selectedLabels: selectedFilters,
         globalFilter: searchQuery,
         sortBy,
@@ -788,7 +793,7 @@ const ManageHostsPage = ({
       } successfully deleted.`;
 
       dispatch(renderFlash("success", successMessage));
-      retrieveHosts({
+      refetchHosts({
         selectedLabels: selectedFilters,
         globalFilter: searchQuery,
         sortBy,
@@ -796,6 +801,7 @@ const ManageHostsPage = ({
         policyId,
         policyResponse,
       });
+
       refetchLabels();
       toggleDeleteHostModal();
       setSelectedHostIds([]);