🔧 Enable usage limits on viewer

2022-12-05 13:53:20 +01:00
parent 4b2f42b40d
commit cfcecaaa17
4 changed files with 334 additions and 263 deletions
--- a/apps/viewer/src/features/usage/usage.spec.ts
+++ b/apps/viewer/src/features/usage/usage.spec.ts
@ -1,19 +1,75 @@
+import { getTestAsset } from '@/test/utils/playwright'
 import test, { expect } from '@playwright/test'
 import cuid from 'cuid'
+import { Plan } from 'db'
+import { defaultSettings } from 'models'
+import {
+  createWorkspaces,
+  importTypebotInDatabase,
+  injectFakeResults,
+} from 'utils/playwright/databaseActions'
+import { typebotViewer } from 'utils/playwright/testHelpers'

-// TODO: uncomment on 1st of November
+test('should not start if chat limit is reached', async ({ page, context }) => {
+  await test.step('Free plan', async () => {
+    const workspaceId = cuid()
+    const typebotId = cuid()
+    await createWorkspaces([{ id: workspaceId, plan: Plan.FREE }])
+    await importTypebotInDatabase(getTestAsset('typebots/fileUpload.json'), {
+      id: typebotId,
+      publicId: `${typebotId}-public`,
+      workspaceId,
+    })
+    await injectFakeResults({ typebotId, count: 400 })
+    await page.goto(`/${typebotId}-public`)
+    await expect(page.locator('text="This bot is now closed."')).toBeVisible()
+    await page.goto(`${process.env.NEXTAUTH_URL}/typebots/${typebotId}/results`)
+    await expect(page.locator('text="133%"')).toBeVisible()
+  })

-// test('should not start if chat limit is reached', async ({ page }) => {
-//   const typebotId = cuid()
-//   await importTypebotInDatabase(
-//     path.join(__dirname, '../fixtures/typebots/fileUpload.json'),
-//     {
-//       id: typebotId,
-//       publicId: `${typebotId}-public`,
-//       workspaceId: limitTestWorkspaceId,
-//     }
-//   )
-//   await createResults({ typebotId, count: 320 })
-//   await page.goto(`/${typebotId}-public`)
-//   await expect(page.locator('text="This bot is now closed."')).toBeVisible()
-// })
+  await test.step('Lifetime plan', async () => {
+    const workspaceId = cuid()
+    const typebotId = cuid()
+    await createWorkspaces([{ id: workspaceId, plan: Plan.LIFETIME }])
+    await importTypebotInDatabase(getTestAsset('typebots/fileUpload.json'), {
+      id: typebotId,
+      publicId: `${typebotId}-public`,
+      workspaceId,
+    })
+    await injectFakeResults({ typebotId, count: 3000 })
+    await page.goto(`/${typebotId}-public`)
+    await expect(
+      typebotViewer(page).locator('text="Hey there, upload please"')
+    ).toBeVisible()
+  })
+
+  await test.step('Custom plan', async () => {
+    const workspaceId = cuid()
+    const typebotId = cuid()
+    await createWorkspaces([
+      { id: workspaceId, plan: Plan.CUSTOM, customChatsLimit: 1000 },
+    ])
+    await importTypebotInDatabase(getTestAsset('typebots/fileUpload.json'), {
+      id: typebotId,
+      publicId: `${typebotId}-public`,
+      workspaceId,
+      settings: {
+        ...defaultSettings,
+        general: {
+          ...defaultSettings.general,
+          isNewResultOnRefreshEnabled: true,
+        },
+      },
+    })
+    const page = await context.newPage()
+    await page.goto(`/${typebotId}-public`)
+    await expect(
+      typebotViewer(page).locator('text="Hey there, upload please"')
+    ).toBeVisible()
+    await injectFakeResults({ typebotId, count: 2000 })
+    await page.goto(`/${typebotId}-public`)
+    await expect(page.locator('text="This bot is now closed."')).toBeVisible()
+    await page.goto(`${process.env.NEXTAUTH_URL}/typebots/${typebotId}/results`)
+    await expect(page.locator('text="200%"')).toBeVisible()
+  })
+})