Update policy back link to return to selected team (#3190)

changes/issue-3130-fix-policy-back-link

@@ -0,0 +1 @@
+* Update policy back link to return to selected team

frontend/pages/policies/ManagePoliciesPage/ManagePoliciesPage.tsx

@@ -1,5 +1,4 @@
 import React, { useCallback, useContext, useEffect, useState } from "react";
-import { Link } from "react-router";
 import { useQuery } from "react-query";
 import { useDispatch } from "react-redux";
 import { noop } from "lodash";
@@ -261,7 +260,7 @@ const ManagePolicyPage = (managePoliciesPageProps: {
     if (userTeams && !userTeams.find((t) => t.id === teamId)) {
       if (isOnGlobalTeam) {
         // For global users, default to zero (i.e. all teams).
-        if (teamId !== 0) {
+        if (teamId === undefined) {
           handleTeamSelect(0);
           return;
         }

frontend/pages/policies/PolicyPage/screens/QueryEditor.tsx

@@ -140,9 +140,14 @@ const QueryEditor = ({
     return null;
   }
 
+  const backPath = policyTeamId ? `?team_id=${policyTeamId}` : "";
+
   return (
     <div className={`${baseClass}__form body-wrap`}>
-      <Link to={PATHS.MANAGE_POLICIES} className={`${baseClass}__back-link`}>
+      <Link
+        to={`${PATHS.MANAGE_POLICIES}/${backPath}`}
+        className={`${baseClass}__back-link`}
+      >
         <img src={BackChevron} alt="back chevron" id="back-chevron" />
         <span>Back to policies</span>
       </Link>

frontend/pages/queries/QueryPage/screens/QueryEditor.tsx

@@ -44,7 +44,7 @@ const QueryEditor = ({
   goToSelectTargets,
   onOpenSchemaSidebar,
   renderLiveQueryWarning,
-}: IQueryEditorProps) => {
+}: IQueryEditorProps): JSX.Element | null => {
   const dispatch = useDispatch();
   const { currentUser } = useContext(AppContext);
 

frontend/pages/queries/QueryPage/screens/RunQuery.tsx

@@ -32,7 +32,7 @@ const RunQuery = ({
   queryIdForEdit,
   setSelectedTargets,
   goToQueryEditor,
-}: IRunQueryProps) => {
+}: IRunQueryProps): JSX.Element | null => {
   const dispatch = useDispatch();
 
   const [isQueryFinished, setIsQueryFinished] = useState<boolean>(false);