[Fleet] Render extension replacement component after package policy i…

…nit (#151876)
elastic · Feb 26, 2023 · 5de208b · 5de208b
1 parent 74b7929
commit 5de208b
Showing 1 changed file with 17 additions and 12 deletions.
diff --git a/...tions/fleet/sections/agent_policy/create_package_policy_page/single_page_layout/index.tsx b/...tions/fleet/sections/agent_policy/create_package_policy_page/single_page_layout/index.tsx
@@ -278,18 +278,23 @@ export const CreatePackagePolicySinglePage: CreatePackagePolicyParams = ({
     );
   }
 
-  const replaceStepConfigurePackagePolicy = replaceDefineStepView && packageInfo?.name && (
-    <ExtensionWrapper>
-      <replaceDefineStepView.Component
-        agentPolicy={agentPolicy}
-        packageInfo={packageInfo}
-        newPolicy={packagePolicy}
-        onChange={handleExtensionViewOnChange}
-        validationResults={validationResults}
-        isEditPage={false}
-      />
-    </ExtensionWrapper>
-  );
+  const replaceStepConfigurePackagePolicy =
+    replaceDefineStepView && packageInfo?.name ? (
+      !isInitialized ? (
+        <Loading />
+      ) : (
+        <ExtensionWrapper>
+          <replaceDefineStepView.Component
+            agentPolicy={agentPolicy}
+            packageInfo={packageInfo}
+            newPolicy={packagePolicy}
+            onChange={handleExtensionViewOnChange}
+            validationResults={validationResults}
+            isEditPage={false}
+          />
+        </ExtensionWrapper>
+      )
+    ) : undefined;
 
   const stepConfigurePackagePolicy = useMemo(
     () =>