From 0c7dafea2ae17db54be06f74b9f2d2d7fba16f2f Mon Sep 17 00:00:00 2001
From: Terry Sutton <saltcod@gmail.com>
Date: Thu, 6 Oct 2022 17:15:14 -0230
Subject: [PATCH] Adds a test for showing upgrade prompt if user on free plan

---
 studio/tests/pages/projects/logs-query.test.js | 16 ++++------------
 1 file changed, 4 insertions(+), 12 deletions(-)

diff --git a/studio/tests/pages/projects/logs-query.test.js b/studio/tests/pages/projects/logs-query.test.js
index cfb7e5d4dfc..41093ed8cb5 100644
--- a/studio/tests/pages/projects/logs-query.test.js
+++ b/studio/tests/pages/projects/logs-query.test.js
@@ -167,7 +167,7 @@ test('datepicker interaction updates query params', async () => {
   }
 })
 
-test('Upgrade prompt displays if user on free plan and tries to view beyond 24hrs', async () => {
+test.only('Upgrade prompt displays if user on free plan and tries to view beyond 24hrs', async () => {
   render(<LogsExplorerPage />)
   render(<UpgradePrompt />)
 
@@ -177,23 +177,15 @@ test('Upgrade prompt displays if user on free plan and tries to view beyond 24hr
   useProjectSubscription.mockReturnValue({
     subscription: {
       tier: {
-        supabase_prod_id: 'tier_pro',
+        supabase_prod_id: 'free',
       },
     },
   })
 
   const { subscription } = useProjectSubscription()
 
-  if (subscription.tier.supabase_prod_id === 'tier_pro') {
-    const router = useRouter()
-    expect(router.push).toBeCalled()
-    expect(router.push).toBeCalledWith(
-      expect.objectContaining({
-        query: expect.objectContaining({
-          its: expect.any(String),
-        }),
-      })
-    )
+  if (subscription.tier.supabase_prod_id !== 'free') {
+    expect(getByText('Logs can be retained up to a')).toBeInTheDocument()
   }
 })