🔧 Enable usage limits on viewer

apps/viewer/src/features/blocks/inputs/fileUpload/fileUpload.spec.ts

@@ -3,11 +3,16 @@ import cuid from 'cuid'
 import { parse } from 'papaparse'
 import { readFileSync } from 'fs'
 import { isDefined } from 'utils'
-import { importTypebotInDatabase } from 'utils/playwright/databaseActions'
+import {
+  createWorkspaces,
+  importTypebotInDatabase,
+  injectFakeResults,
+} from 'utils/playwright/databaseActions'
 import { typebotViewer } from 'utils/playwright/testHelpers'
 import { getTestAsset } from '@/test/utils/playwright'
+import { Plan } from 'db'
 
-// const THREE_GIGABYTES = 3 * 1024 * 1024 * 1024
+const THREE_GIGABYTES = 3 * 1024 * 1024 * 1024
 
 test('should work as expected', async ({ page, browser }) => {
   const typebotId = cuid()
@@ -73,47 +78,45 @@ test('should work as expected', async ({ page, browser }) => {
   await expect(page2.locator('pre')).toBeHidden()
 })
 
-// TODO: uncomment on 1st of November
-
-// test.describe('Storage limit is reached', () => {
-//   const typebotId = cuid()
+test.describe('Storage limit is reached', () => {
+  const typebotId = cuid()
+  const workspaceId = cuid()
 
-//   test.beforeAll(async () => {
-//     await importTypebotInDatabase(
-//       path.join(__dirname, '../fixtures/typebots/fileUpload.json'),
-//       {
-//         id: typebotId,
-//         publicId: `${typebotId}-public`,
-//       }
-//     )
-//     await createResults({
-//       typebotId,
-//       count: 20,
-//       fakeStorage: THREE_GIGABYTES,
-//     })
-//   })
+  test.beforeAll(async () => {
+    await createWorkspaces([{ id: workspaceId, plan: Plan.STARTER }])
+    await importTypebotInDatabase(getTestAsset('typebots/fileUpload.json'), {
+      id: typebotId,
+      publicId: `${typebotId}-public`,
+      workspaceId,
+    })
+    await injectFakeResults({
+      typebotId,
+      count: 20,
+      fakeStorage: THREE_GIGABYTES,
+    })
+  })
 
-//   test("shouldn't upload anything if limit has been reached", async ({
-//     page,
-//   }) => {
-//     await page.goto(`/${typebotId}-public`)
-//     await typebotViewer(page)
-//       .locator(`input[type="file"]`)
-//       .setInputFiles([
-//         path.join(__dirname, '../fixtures/typebots/api.json'),
-//         path.join(__dirname, '../fixtures/typebots/fileUpload.json'),
-//         path.join(__dirname, '../fixtures/typebots/hugeGroup.json'),
-//       ])
-//     await expect(typebotViewer(page).locator(`text="3"`)).toBeVisible()
-//     await typebotViewer(page).locator('text="Upload 3 files"').click()
-//     await expect(
-//       typebotViewer(page).locator(`text="3 files uploaded"`)
-//     ).toBeVisible()
-//     await page.evaluate(() =>
-//       window.localStorage.setItem('workspaceId', 'starterWorkspace')
-//     )
-//     await page.goto(`${process.env.NEXTAUTH_URL}/typebots/${typebotId}/results`)
-//     await expect(page.locator('text="150%"')).toBeVisible()
-//     await expect(page.locator('text="api.json"')).toBeHidden()
-//   })
-// })
+  test("shouldn't upload anything if limit has been reached", async ({
+    page,
+  }) => {
+    await page.goto(`/${typebotId}-public`)
+    await typebotViewer(page)
+      .locator(`input[type="file"]`)
+      .setInputFiles([
+        getTestAsset('typebots/api.json'),
+        getTestAsset('typebots/fileUpload.json'),
+        getTestAsset('typebots/hugeGroup.json'),
+      ])
+    await expect(typebotViewer(page).locator(`text="3"`)).toBeVisible()
+    await typebotViewer(page).locator('text="Upload 3 files"').click()
+    await expect(
+      typebotViewer(page).locator(`text="3 files uploaded"`)
+    ).toBeVisible()
+    await page.evaluate(() =>
+      window.localStorage.setItem('workspaceId', 'starterWorkspace')
+    )
+    await page.goto(`${process.env.NEXTAUTH_URL}/typebots/${typebotId}/results`)
+    await expect(page.locator('text="150%"')).toBeVisible()
+    await expect(page.locator('text="api.json"')).toBeHidden()
+  })
+})

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()
+  })
+})