From dcac61e4d9ab7c6dd1985cb2c4d6d46d2c905d4a Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ivar=20Conradi=20=C3=98sthus?= <ivar@getunleash.ai>
Date: Fri, 16 Jun 2023 13:35:24 +0200
Subject: [PATCH] fix: add trial expired warning for enterprise (#3997)

Adds the trial expired warning for enterprise as well.
---
 .../BillingPlan/BillingPlan.tsx               |  1 +
 .../common/InstanceStatus/InstanceStatus.tsx  |  2 +-
 .../InstanceStatus/InstanceStatusBar.test.tsx | 14 +++++++
 .../InstanceStatusBar.test.tsx.snap           | 40 +++++++++++++++++++
 frontend/src/interfaces/instance.ts           |  1 +
 5 files changed, 57 insertions(+), 1 deletion(-)

diff --git a/frontend/src/component/admin/billing/BillingDashboard/BillingPlan/BillingPlan.tsx b/frontend/src/component/admin/billing/BillingDashboard/BillingPlan/BillingPlan.tsx
index d5e67e046c..d42edabd77 100644
--- a/frontend/src/component/admin/billing/BillingDashboard/BillingPlan/BillingPlan.tsx
+++ b/frontend/src/component/admin/billing/BillingDashboard/BillingPlan/BillingPlan.tsx
@@ -82,6 +82,7 @@ export const BillingPlan: FC<IBillingPlanProps> = ({ instanceStatus }) => {
         [InstancePlan.PRO]: 80,
         [InstancePlan.COMPANY]: 0,
         [InstancePlan.TEAM]: 0,
+        [InstancePlan.ENTERPRISE]: 0,
         [InstancePlan.UNKNOWN]: 0,
         user: 15,
     };
diff --git a/frontend/src/component/common/InstanceStatus/InstanceStatus.tsx b/frontend/src/component/common/InstanceStatus/InstanceStatus.tsx
index 25282e2077..62f1213dc3 100644
--- a/frontend/src/component/common/InstanceStatus/InstanceStatus.tsx
+++ b/frontend/src/component/common/InstanceStatus/InstanceStatus.tsx
@@ -104,7 +104,7 @@ export const InstanceStatus: FC = ({ children }) => {
     return (
         <>
             <ConditionallyRender
-                condition={isBilling && Boolean(instanceStatus)}
+                condition={Boolean(instanceStatus)}
                 show={() => (
                     <>
                         <InstanceStatusBarMemo
diff --git a/frontend/src/component/common/InstanceStatus/InstanceStatusBar.test.tsx b/frontend/src/component/common/InstanceStatus/InstanceStatusBar.test.tsx
index d78c26248d..4ebea4ce82 100644
--- a/frontend/src/component/common/InstanceStatus/InstanceStatusBar.test.tsx
+++ b/frontend/src/component/common/InstanceStatus/InstanceStatusBar.test.tsx
@@ -95,6 +95,20 @@ test('InstanceStatusBar should warn when the trial has expired', async () => {
     expect(await screen.findByTestId(INSTANCE_STATUS_BAR_ID)).toMatchSnapshot();
 });
 
+test('InstanceStatusBar should warn when the trial has expired for enterprise', async () => {
+    render(
+        <InstanceStatusBar
+            instanceStatus={{
+                plan: InstancePlan.ENTERPRISE,
+                state: InstanceState.EXPIRED,
+            }}
+        />
+    );
+
+    expect(screen.getByTestId(INSTANCE_STATUS_BAR_ID)).toBeInTheDocument();
+    expect(await screen.findByTestId(INSTANCE_STATUS_BAR_ID)).toMatchSnapshot();
+});
+
 test('InstanceStatusBar should warn when the trial has churned', async () => {
     render(
         <InstanceStatusBar
diff --git a/frontend/src/component/common/InstanceStatus/__snapshots__/InstanceStatusBar.test.tsx.snap b/frontend/src/component/common/InstanceStatus/__snapshots__/InstanceStatusBar.test.tsx.snap
index c0d921daa1..f76ede60a5 100644
--- a/frontend/src/component/common/InstanceStatus/__snapshots__/InstanceStatusBar.test.tsx.snap
+++ b/frontend/src/component/common/InstanceStatus/__snapshots__/InstanceStatusBar.test.tsx.snap
@@ -80,6 +80,46 @@ exports[`InstanceStatusBar should warn when the trial has expired 1`] = `
 </aside>
 `;
 
+exports[`InstanceStatusBar should warn when the trial has expired for enterprise 1`] = `
+<aside
+  class="css-163g6kh"
+  data-testid="INSTANCE_STATUS_BAR_ID"
+>
+  <svg
+    aria-hidden="true"
+    class="MuiSvgIcon-root MuiSvgIcon-fontSizeMedium css-prk1jy-MuiSvgIcon-root"
+    data-testid="WarningAmberIcon"
+    focusable="false"
+    viewBox="0 0 24 24"
+  >
+    <path
+      d="M12 5.99 19.53 19H4.47L12 5.99M12 2 1 21h22L12 2z"
+    />
+    <path
+      d="M13 16h-2v2h2zm0-6h-2v5h2z"
+    />
+  </svg>
+  <p
+    class="MuiTypography-root MuiTypography-body1 css-rviqjc-MuiTypography-root"
+  >
+    <strong>
+      Warning!
+    </strong>
+     Your free 
+    Enterprise
+     trial has expired. 
+    <strong>
+      Upgrade trial
+    </strong>
+     otherwise your
+     
+    <strong>
+      account will be deleted.
+    </strong>
+  </p>
+</aside>
+`;
+
 exports[`InstanceStatusBar should warn when the trial is about to expire 1`] = `
 <aside
   class="css-f22srx"
diff --git a/frontend/src/interfaces/instance.ts b/frontend/src/interfaces/instance.ts
index 6f04f759eb..cdd3397e6c 100644
--- a/frontend/src/interfaces/instance.ts
+++ b/frontend/src/interfaces/instance.ts
@@ -20,5 +20,6 @@ export enum InstancePlan {
     PRO = 'Pro',
     COMPANY = 'Company',
     TEAM = 'Team',
+    ENTERPRISE = 'Enterprise',
     UNKNOWN = 'Unknown',
 }