88 lines
3.4 KiB
TypeScript
88 lines
3.4 KiB
TypeScript
import { net, protocol } from 'electron/main';
|
|
|
|
import { expect } from 'chai';
|
|
|
|
import * as path from 'node:path';
|
|
import * as url from 'node:url';
|
|
|
|
import { defer } from './lib/spec-helpers';
|
|
|
|
describe('net module custom protocols', () => {
|
|
it('can request file:// URLs', async () => {
|
|
const resp = await net.fetch(url.pathToFileURL(path.join(__dirname, 'fixtures', 'hello.txt')).toString());
|
|
expect(resp.ok).to.be.true();
|
|
// trimRight instead of asserting the whole string to avoid line ending shenanigans on WOA
|
|
expect((await resp.text()).trimRight()).to.equal('hello world');
|
|
});
|
|
|
|
it('can make requests to custom protocols', async () => {
|
|
protocol.registerStringProtocol('electron-test', (req, cb) => { cb('hello ' + req.url); });
|
|
defer(() => {
|
|
protocol.unregisterProtocol('electron-test');
|
|
});
|
|
const body = await net.fetch('electron-test://foo').then(r => r.text());
|
|
expect(body).to.equal('hello electron-test://foo');
|
|
});
|
|
|
|
it('runs through intercept handlers', async () => {
|
|
protocol.interceptStringProtocol('http', (req, cb) => { cb('hello ' + req.url); });
|
|
defer(() => {
|
|
protocol.uninterceptProtocol('http');
|
|
});
|
|
const body = await net.fetch('http://foo').then(r => r.text());
|
|
expect(body).to.equal('hello http://foo/');
|
|
});
|
|
|
|
it('file: runs through intercept handlers', async () => {
|
|
protocol.interceptStringProtocol('file', (req, cb) => { cb('hello ' + req.url); });
|
|
defer(() => {
|
|
protocol.uninterceptProtocol('file');
|
|
});
|
|
const body = await net.fetch('file://foo').then(r => r.text());
|
|
expect(body).to.equal('hello file://foo/');
|
|
});
|
|
|
|
it('can be redirected', async () => {
|
|
protocol.interceptStringProtocol('file', (req, cb) => { cb({ statusCode: 302, headers: { location: 'electron-test://bar' } }); });
|
|
defer(() => {
|
|
protocol.uninterceptProtocol('file');
|
|
});
|
|
protocol.registerStringProtocol('electron-test', (req, cb) => { cb('hello ' + req.url); });
|
|
defer(() => {
|
|
protocol.unregisterProtocol('electron-test');
|
|
});
|
|
const body = await net.fetch('file://foo').then(r => r.text());
|
|
expect(body).to.equal('hello electron-test://bar');
|
|
});
|
|
|
|
it('should not follow redirect when redirect: error', async () => {
|
|
protocol.registerStringProtocol('electron-test', (req, cb) => {
|
|
if (/redirect/.test(req.url)) return cb({ statusCode: 302, headers: { location: 'electron-test://bar' } });
|
|
cb('hello ' + req.url);
|
|
});
|
|
defer(() => {
|
|
protocol.unregisterProtocol('electron-test');
|
|
});
|
|
await expect(net.fetch('electron-test://redirect', { redirect: 'error' })).to.eventually.be.rejectedWith('Attempted to redirect, but redirect policy was \'error\'');
|
|
});
|
|
|
|
it('a 307 redirected POST request preserves the body', async () => {
|
|
const bodyData = 'Hello World!';
|
|
let postedBodyData: any;
|
|
protocol.registerStringProtocol('electron-test', async (req, cb) => {
|
|
if (/redirect/.test(req.url)) return cb({ statusCode: 307, headers: { location: 'electron-test://bar' } });
|
|
postedBodyData = req.uploadData![0].bytes.toString();
|
|
cb('hello ' + req.url);
|
|
});
|
|
defer(() => {
|
|
protocol.unregisterProtocol('electron-test');
|
|
});
|
|
const response = await net.fetch('electron-test://redirect', {
|
|
method: 'POST',
|
|
body: bodyData
|
|
});
|
|
expect(response.status).to.equal(200);
|
|
await response.text();
|
|
expect(postedBodyData).to.equal(bodyData);
|
|
});
|
|
});
|