mirror of
https://github.com/cupcakearmy/cryptgeon.git
synced 2024-12-22 00:06:28 +00:00
104 lines
3.3 KiB
TypeScript
104 lines
3.3 KiB
TypeScript
import { expect, type Page } from '@playwright/test'
|
|
import { execFile } from 'node:child_process'
|
|
import { promisify } from 'node:util'
|
|
import { getFileChecksum } from './files'
|
|
|
|
const exec = promisify(execFile)
|
|
|
|
type CreatePage = {
|
|
text?: string
|
|
files?: string[]
|
|
views?: number
|
|
expiration?: number
|
|
error?: string
|
|
password?: string
|
|
}
|
|
export async function createNote(page: Page, options: CreatePage): Promise<string> {
|
|
await page.goto('/')
|
|
|
|
if (options.text) {
|
|
await page.getByTestId('text-field').fill(options.text)
|
|
} else if (options.files) {
|
|
await page.getByTestId('switch-file').click()
|
|
|
|
const [fileChooser] = await Promise.all([
|
|
page.waitForEvent('filechooser'),
|
|
page.locator('text=No Files Selected').click(),
|
|
])
|
|
await fileChooser.setFiles(options.files)
|
|
}
|
|
|
|
if (options.views || options.expiration || options.password) await page.getByTestId('switch-advanced').click()
|
|
if (options.views) {
|
|
await page.getByTestId('field-views').fill(options.views.toString())
|
|
} else if (options.expiration) {
|
|
await page.getByTestId('switch-advanced-toggle').click()
|
|
await page.getByTestId('field-expiration').fill(options.expiration.toString())
|
|
}
|
|
if (options.password) {
|
|
await page.getByTestId('custom-password').click()
|
|
await page.getByTestId('password').fill(options.password)
|
|
}
|
|
|
|
await page.locator('button:has-text("create")').click()
|
|
|
|
if (options.error) {
|
|
await expect(page.locator('.error-text')).toContainText(options.error, { timeout: 60_000 })
|
|
}
|
|
|
|
// Return share link
|
|
return await page.getByTestId('share-link').inputValue()
|
|
}
|
|
|
|
type CheckLinkBase = {
|
|
link: string
|
|
text: string
|
|
password?: string
|
|
}
|
|
|
|
export async function checkLinkForDownload(page: Page, options: CheckLinkBase & { checksum: string }) {
|
|
await page.goto('/')
|
|
await page.goto(options.link)
|
|
if (options.password) await page.getByTestId('show-note-password').fill(options.password)
|
|
await page.getByTestId('show-note-button').click()
|
|
|
|
const [download] = await Promise.all([
|
|
page.waitForEvent('download'),
|
|
page.getByTestId(`result`).locator(`text=${options.text}`).click(),
|
|
])
|
|
const path = await download.path()
|
|
if (!path) throw new Error('Download failed')
|
|
const cs = await getFileChecksum(path)
|
|
await expect(cs).toBe(options.checksum)
|
|
}
|
|
|
|
export async function checkLinkForText(page: Page, options: CheckLinkBase) {
|
|
await page.goto('/')
|
|
await page.goto(options.link)
|
|
if (options.password) await page.getByTestId('show-note-password').fill(options.password)
|
|
await page.getByTestId('show-note-button').click()
|
|
const text = await page.getByTestId('result').locator('.note').innerText()
|
|
await expect(text).toContain(options.text)
|
|
}
|
|
|
|
export async function checkLinkDoesNotExist(page: Page, link: string) {
|
|
await page.goto('/') // Required due to firefox: https://github.com/microsoft/playwright/issues/15781
|
|
await page.goto(link)
|
|
await expect(page.locator('main')).toContainText('note was not found or was already deleted')
|
|
}
|
|
|
|
export async function CLI(...args: string[]) {
|
|
return await exec('./packages/cli/dist/index.cjs', args, {
|
|
env: {
|
|
...process.env,
|
|
CRYPTGEON_SERVER: 'http://localhost:1234',
|
|
},
|
|
})
|
|
}
|
|
|
|
export function getLinkFromCLI(output: string): string {
|
|
const match = output.match(/(https?:\/\/[^\s]+)/)
|
|
if (!match) throw new Error('No link found in CLI output')
|
|
return match[0]
|
|
}
|