playwright/test/downloads-path.spec.ts

124 lines
4.4 KiB
TypeScript
Raw Normal View History

2020-08-03 23:41:48 +03:00
/**
* Copyright (c) Microsoft Corporation.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
2020-10-12 19:16:02 +03:00
import { folio } from './fixtures';
import fs from 'fs';
import type { Browser, BrowserContext } from '..';
2020-08-03 23:41:48 +03:00
type TestState = {
downloadsBrowser: Browser;
persistentDownloadsContext: BrowserContext;
};
2020-10-14 08:40:25 +03:00
const fixtures = folio.extend<TestState>();
2020-08-03 23:41:48 +03:00
fixtures.downloadsBrowser.init(async ({ server, browserType, browserOptions, testInfo }, test) => {
2020-10-12 19:16:02 +03:00
server.setRoute('/download', (req, res) => {
res.setHeader('Content-Type', 'application/octet-stream');
res.setHeader('Content-Disposition', 'attachment; filename=file.txt');
res.end(`Hello world`);
});
const browser = await browserType.launch({
...browserOptions,
2020-10-12 19:16:02 +03:00
downloadsPath: testInfo.outputPath(''),
});
await test(browser);
await browser.close();
2020-08-03 23:41:48 +03:00
});
fixtures.persistentDownloadsContext.init(async ({ server, launchPersistent, testInfo, browserChannel }, test) => {
2020-10-12 19:16:02 +03:00
server.setRoute('/download', (req, res) => {
res.setHeader('Content-Type', 'application/octet-stream');
res.setHeader('Content-Disposition', 'attachment; filename=file.txt');
res.end(`Hello world`);
});
const { context, page } = await launchPersistent(
{
downloadsPath: testInfo.outputPath(''),
acceptDownloads: true,
2020-10-12 19:16:02 +03:00
}
);
await page.setContent(`<a href="${server.PREFIX}/download">download</a>`);
await test(context);
await context.close();
});
const { it, expect } = fixtures.build();
2020-08-03 23:41:48 +03:00
it('should keep downloadsPath folder', async ({downloadsBrowser, testInfo, server}) => {
2020-08-03 23:41:48 +03:00
const page = await downloadsBrowser.newPage();
await page.setContent(`<a href="${server.PREFIX}/download">download</a>`);
const [ download ] = await Promise.all([
page.waitForEvent('download'),
page.click('a')
]);
expect(download.url()).toBe(`${server.PREFIX}/download`);
expect(download.suggestedFilename()).toBe(`file.txt`);
await download.path().catch(e => void 0);
2020-08-03 23:41:48 +03:00
await page.close();
await downloadsBrowser.close();
expect(fs.existsSync(testInfo.outputPath(''))).toBeTruthy();
2020-08-03 23:41:48 +03:00
});
it('should delete downloads when context closes', async ({downloadsBrowser, server}) => {
2020-08-03 23:41:48 +03:00
const page = await downloadsBrowser.newPage({ acceptDownloads: true });
await page.setContent(`<a href="${server.PREFIX}/download">download</a>`);
const [ download ] = await Promise.all([
page.waitForEvent('download'),
page.click('a')
]);
const path = await download.path();
expect(fs.existsSync(path)).toBeTruthy();
await page.close();
expect(fs.existsSync(path)).toBeFalsy();
});
2020-08-05 21:43:40 +03:00
it('should report downloads in downloadsPath folder', async ({downloadsBrowser, testInfo, server}) => {
2020-08-03 23:41:48 +03:00
const page = await downloadsBrowser.newPage({ acceptDownloads: true });
await page.setContent(`<a href="${server.PREFIX}/download">download</a>`);
const [ download ] = await Promise.all([
page.waitForEvent('download'),
page.click('a')
]);
const path = await download.path();
expect(path.startsWith(testInfo.outputPath(''))).toBeTruthy();
2020-08-03 23:41:48 +03:00
await page.close();
});
it('should accept downloads in persistent context', async ({persistentDownloadsContext, testInfo, server}) => {
2020-08-03 23:41:48 +03:00
const page = persistentDownloadsContext.pages()[0];
const [ download ] = await Promise.all([
page.waitForEvent('download'),
page.click('a'),
2020-08-03 23:41:48 +03:00
]);
expect(download.url()).toBe(`${server.PREFIX}/download`);
expect(download.suggestedFilename()).toBe(`file.txt`);
const path = await download.path();
expect(path.startsWith(testInfo.outputPath(''))).toBeTruthy();
2020-08-03 23:41:48 +03:00
});
it('should delete downloads when persistent context closes', async ({persistentDownloadsContext}) => {
2020-08-03 23:41:48 +03:00
const page = persistentDownloadsContext.pages()[0];
const [ download ] = await Promise.all([
page.waitForEvent('download'),
page.click('a'),
2020-08-03 23:41:48 +03:00
]);
const path = await download.path();
expect(fs.existsSync(path)).toBeTruthy();
await persistentDownloadsContext.close();
expect(fs.existsSync(path)).toBeFalsy();
2020-08-03 23:41:48 +03:00
});