bot/apps/viewer/playwright/tests/fileUpload.spec.ts

import test, { expect } from '@playwright/test'
import cuid from 'cuid'
import path from 'path'
import { parse } from 'papaparse'
import { typebotViewer } from '../services/selectorUtils'
import { createResults, importTypebotInDatabase } from '../services/database'
import { readFileSync } from 'fs'
import { isDefined } from 'utils'

const THREE_GIGABYTES = 3 * 1024 * 1024 * 1024

test('should work as expected', async ({ page, browser }) => {
  const typebotId = cuid()
  await importTypebotInDatabase(
    path.join(__dirname, '../fixtures/typebots/fileUpload.json'),
    { id: typebotId, publicId: `${typebotId}-public` }
  )
  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.goto(`${process.env.BUILDER_URL}/typebots/${typebotId}/results`)
  await expect(page.locator('text="api.json"')).toHaveAttribute(
    'href',
    /.+\/api\.json/
  )
  await expect(page.locator('text="fileUpload.json"')).toHaveAttribute(
    'href',
    /.+\/fileUpload\.json/
  )
  await expect(page.locator('text="hugeGroup.json"')).toHaveAttribute(
    'href',
    /.+\/hugeGroup\.json/
  )

  await page.click('[data-testid="checkbox"] >> nth=0')
  const [download] = await Promise.all([
    page.waitForEvent('download'),
    page.locator('text="Export"').click(),
  ])
  const downloadPath = await download.path()
  expect(downloadPath).toBeDefined()
  const file = readFileSync(downloadPath as string).toString()
  const { data } = parse(file)
  expect(data).toHaveLength(2)
  expect((data[1] as unknown[])[1]).toContain('http://localhost:9000')

  const urls = (
    await Promise.all(
      [
        page.locator('text="api.json"'),
        page.locator('text="fileUpload.json"'),
        page.locator('text="hugeGroup.json"'),
      ].map((elem) => elem.getAttribute('href'))
    )
  ).filter(isDefined)

  const page2 = await browser.newPage()
  await page2.goto(urls[0])
  await expect(page2.locator('pre')).toBeVisible()

  await page.locator('button >> text="Delete"').click()
  await page.locator('button >> text="Delete" >> nth=1').click()
  await expect(page.locator('text="api.json"')).toBeHidden()
  await page2.goto(urls[0])
  await expect(
    page2.locator('span:has-text("The specified key does not exist.")')
  ).toBeVisible()
  await page2.goto(urls[1])
  await expect(
    page2.locator('span:has-text("The specified key does not exist.")')
  ).toBeVisible()
  await page2.goto(urls[2])
  await expect(
    page2.locator('span:has-text("The specified key does not exist.")')
  ).toBeVisible()
})

test.describe('Storage limit is reached', () => {
  const typebotId = 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("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.BUILDER_URL}/typebots/${typebotId}/results`)
    await expect(page.locator('text="150%"')).toBeVisible()
    await expect(page.locator('text="api.json"')).toBeHidden()
  })
})