From 02c3c4d5f11fe6e71b403b6b54558738f0bed9d3 Mon Sep 17 00:00:00 2001
From: padamsen_fisker <padamsen@fiskerinc.com>
Date: Tue, 4 Jun 2024 16:45:50 -0400
Subject: [PATCH] CEC-6051 - Fix UI for bulk actions

---
 src/components/Fleets/Status/Vehicles/Table/index.jsx | 4 +++-
 src/components/Table/HeaderSortable/index.jsx         | 1 +
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/components/Fleets/Status/Vehicles/Table/index.jsx b/src/components/Fleets/Status/Vehicles/Table/index.jsx
index e2fe7c2..2222820 100644
--- a/src/components/Fleets/Status/Vehicles/Table/index.jsx
+++ b/src/components/Fleets/Status/Vehicles/Table/index.jsx
@@ -70,6 +70,7 @@ const MainForm = ({ name }) => {
   const [order, setOrder] = useState("desc");
   const [search, setSearch] = useState("");
   const [selected, setSelected] = useState([]);
+  const [selectAll, setSelectAll] = useState(false);
   const [fleetVINs, setFleetVINs] = useState([]);
   const componentMounted = useRef(true);
   const classes = useStyles();
@@ -155,7 +156,8 @@ const MainForm = ({ name }) => {
   };
 
   const handleSelectAll = () => {
-    setSelected(fleetVINs);
+    setSelectAll(!selectAll);
+    setSelected(!selectAll ? fleetVINs : []);
   };
 
   const handleSelect = (vin, setState) => {
diff --git a/src/components/Table/HeaderSortable/index.jsx b/src/components/Table/HeaderSortable/index.jsx
index 3824234..c4de1db 100644
--- a/src/components/Table/HeaderSortable/index.jsx
+++ b/src/components/Table/HeaderSortable/index.jsx
@@ -106,6 +106,7 @@ HeaderSortable.propTypes = {
   selectCount: PropTypes.number,
   totalRows: PropTypes.number,
   onSelectAll: PropTypes.func,
+  selectAllForAllPages: PropTypes.bool,
 };
 
 export default HeaderSortable;