From 9b6d14c7528ab37b3cdce3308d940fc8cdb2eae5 Mon Sep 17 00:00:00 2001
From: Alex <33497058+bexsoft@users.noreply.github.com>
Date: Tue, 20 Jul 2021 20:33:01 -0500
Subject: [PATCH] Fixed policy page crash (#882)

Co-authored-by: Benjamin Perez <benjamin@bexsoft.net>
---
 portal-ui/src/screens/Console/Policies/PolicyDetails.tsx | 9 +++++----
 1 file changed, 5 insertions(+), 4 deletions(-)

diff --git a/portal-ui/src/screens/Console/Policies/PolicyDetails.tsx b/portal-ui/src/screens/Console/Policies/PolicyDetails.tsx
index b986b2dbd..b2b70ea0a 100644
--- a/portal-ui/src/screens/Console/Policies/PolicyDetails.tsx
+++ b/portal-ui/src/screens/Console/Policies/PolicyDetails.tsx
@@ -30,7 +30,7 @@ import TableWrapper from "../Common/TableWrapper/TableWrapper";
 import api from "../../../common/api";
 import PageHeader from "../Common/PageHeader/PageHeader";
 import { Link } from "react-router-dom";
-import { setErrorSnackMessage } from "../../../actions";
+import { setErrorSnackMessage, setSnackBarMessage } from "../../../actions";
 import CodeMirrorWrapper from "../Common/FormComponents/CodeMirrorWrapper/CodeMirrorWrapper";
 import history from "../../../history";
 import InputAdornment from "@material-ui/core/InputAdornment";
@@ -44,8 +44,8 @@ import TextField from "@material-ui/core/TextField";
 interface IPolicyDetailsProps {
   classes: any;
   match: any;
-  closeModalAndRefresh: (refresh: boolean) => void;
   setErrorSnackMessage: typeof setErrorSnackMessage;
+  setSnackBarMessage: typeof setSnackBarMessage;
 }
 
 const styles = (theme: Theme) =>
@@ -160,8 +160,8 @@ const styles = (theme: Theme) =>
 const PolicyDetails = ({
   classes,
   match,
-  closeModalAndRefresh,
   setErrorSnackMessage,
+  setSnackBarMessage,
 }: IPolicyDetailsProps) => {
   const [selectedTab, setSelectedTab] = useState<number>(0);
   const [policy, setPolicy] = useState<Policy | null>(null);
@@ -191,7 +191,7 @@ const PolicyDetails = ({
       })
       .then((res) => {
         setAddLoading(false);
-        closeModalAndRefresh(true);
+        setSnackBarMessage("Policy successfully updated")
       })
       .catch((err) => {
         setAddLoading(false);
@@ -441,6 +441,7 @@ const PolicyDetails = ({
 
 const connector = connect(null, {
   setErrorSnackMessage,
+  setSnackBarMessage
 });
 
 export default withStyles(styles)(connector(PolicyDetails));