Ensure upgrade page displays as upgrade even without from prop

kpollich · kpollich · commit ceb5ad8fcc89 · 2021-08-26T09:21:40.000-04:00
diff --git a/x-pack/plugins/fleet/public/applications/fleet/sections/agent_policy/edit_package_policy_page/index.tsx b/x-pack/plugins/fleet/public/applications/fleet/sections/agent_policy/edit_package_policy_page/index.tsx
@@ -71,8 +71,9 @@ export const EditPackagePolicyPage = memo(() => {
 
 export const EditPackagePolicyForm = memo<{
   packagePolicyId: string;
+  isUpgrade?: boolean;
   from?: EditPackagePolicyFrom;
-}>(({ packagePolicyId, from = 'edit' }) => {
+}>(({ packagePolicyId, isUpgrade = false, from = 'edit' }) => {
   const { application, notifications } = useStartServices();
   const {
     agents: { enabled: isFleetEnabled },
@@ -99,9 +100,6 @@ export const EditPackagePolicyForm = memo<{
   >();
   const [dryRunData, setDryRunData] = useState<UpgradePackagePolicyDryRunResponse>();
 
-  const isUpgrade =
-    from === 'upgrade-from-fleet-policy-list' || from === 'upgrade-from-integrations-policy-list';
-
   const policyId = agentPolicy?.id ?? '';
 
   // Retrieve agent policy, package, and package policy info
diff --git a/x-pack/plugins/fleet/public/applications/fleet/sections/agent_policy/upgrade_package_policy_page/index.tsx b/x-pack/plugins/fleet/public/applications/fleet/sections/agent_policy/upgrade_package_policy_page/index.tsx
@@ -30,5 +30,5 @@ export const UpgradePackagePolicyPage = memo(() => {
     from = 'upgrade-from-integrations-policy-list';
   }
 
-  return <EditPackagePolicyForm packagePolicyId={packagePolicyId} from={from} />;
+  return <EditPackagePolicyForm packagePolicyId={packagePolicyId} from={from} isUpgrade />;
 });

Original file line number	Diff line number	Diff line change
`@@ -30,5 +30,5 @@ export const UpgradePackagePolicyPage = memo(() => {`
`30`	`30`	`from = 'upgrade-from-integrations-policy-list';`
`31`	`31`	`}`
`32`	`32`
`33`		`- return <EditPackagePolicyForm packagePolicyId={packagePolicyId} from={from} />;`
	`33`	`+ return <EditPackagePolicyForm packagePolicyId={packagePolicyId} from={from} isUpgrade />;`
`34`	`34`	`});`