mirror of https://github.com/lukechilds/polar.git
jamaljsr
5 years ago
17 changed files with 322 additions and 313 deletions
@ -1,19 +1 @@ |
|||||
// eslint-disable-next-line @typescript-eslint/no-var-requires
|
module.exports = jest.fn(); |
||||
const { PassThrough } = require('stream'); |
|
||||
|
|
||||
module.exports = () => { |
|
||||
let mockStream; |
|
||||
// return a fake stream when "archiver()" is called in the app
|
|
||||
const ctor = function() { |
|
||||
mockStream = new PassThrough(); |
|
||||
mockStream.file = jest.fn(); |
|
||||
mockStream.directory = jest.fn(); |
|
||||
mockStream.append = jest.fn(); |
|
||||
mockStream.finalize = jest.fn(); |
|
||||
return mockStream; |
|
||||
}; |
|
||||
// attach a func to emit events on the stream from the tests
|
|
||||
ctor.mockEmit = (event, data) => mockStream.emit(event, data); |
|
||||
|
|
||||
return ctor; |
|
||||
}; |
|
||||
|
|||||
@ -1,3 +1,3 @@ |
|||||
module.export = { |
module.exports = { |
||||
Extract: jest.fn(), |
Extract: jest.fn(), |
||||
}; |
}; |
||||
|
|||||
@ -1,107 +1,131 @@ |
|||||
import fsExtra from 'fs-extra'; |
import fsExtra from 'fs-extra'; |
||||
import { join } from 'path'; |
import { join } from 'path'; |
||||
import archiver from 'archiver'; |
import * as archiver from 'archiver'; |
||||
import { PassThrough } from 'stream'; |
import { PassThrough } from 'stream'; |
||||
|
import unzipper from 'unzipper'; |
||||
|
import * as files from 'utils/files'; |
||||
|
import { delay } from './async'; |
||||
|
import { initChartFromNetwork } from './chart'; |
||||
|
import { getNetwork } from './tests'; |
||||
import { unzip, zip } from './zip'; |
import { unzip, zip } from './zip'; |
||||
|
|
||||
|
jest.mock('utils/files'); |
||||
|
|
||||
const fsMock = fsExtra as jest.Mocked<typeof fsExtra>; |
const fsMock = fsExtra as jest.Mocked<typeof fsExtra>; |
||||
|
const filesMock = files as jest.Mocked<typeof files>; |
||||
|
const unzipperMock = unzipper as jest.Mocked<typeof unzipper>; |
||||
const archiverMock = archiver as jest.Mocked<any>; |
const archiverMock = archiver as jest.Mocked<any>; |
||||
|
|
||||
describe('unzip', () => { |
describe('Zip Util', () => { |
||||
it("fail to unzip something that isn't a zip", async () => { |
let unzipStream: PassThrough; |
||||
return expect( |
let zipStream: any; |
||||
unzip(join(__dirname, 'tests', 'resources', 'bar.txt'), 'foobar'), |
let zippedPaths: string[]; |
||||
).rejects.toThrow(); |
const rpcPath = join('alice', 'lightningd', 'regtest', 'lightning-rpc'); |
||||
|
|
||||
|
beforeEach(() => { |
||||
|
const network = getNetwork(); |
||||
|
const chart = initChartFromNetwork(network); |
||||
|
filesMock.read.mockResolvedValue(JSON.stringify({ network, chart })); |
||||
|
fsMock.pathExists.mockResolvedValue(true as never); |
||||
|
fsMock.mkdirp.mockResolvedValue(true as never); |
||||
|
fsMock.copy.mockResolvedValue(true as never); |
||||
|
fsMock.createWriteStream.mockReturnValue(new PassThrough() as any); |
||||
|
fsMock.createReadStream.mockImplementation(() => { |
||||
|
return { |
||||
|
pipe: jest.fn(() => { |
||||
|
// return the mock stream when "pipe()" is called
|
||||
|
unzipStream = new PassThrough(); |
||||
|
return unzipStream; |
||||
|
}), |
||||
|
} as any; |
||||
|
}); |
||||
|
unzipperMock.Extract.mockImplementation(jest.fn()); |
||||
|
archiverMock.mockImplementation(() => { |
||||
|
// return a fake stream when "archiver()" is called in the app
|
||||
|
zipStream = new PassThrough(); |
||||
|
zipStream.file = jest.fn(); |
||||
|
zipStream.directory = jest.fn( |
||||
|
(src: string, dest: string, entryData: archiver.EntryDataFunction) => { |
||||
|
// call the entryData func on some test paths
|
||||
|
const mockPaths = ['test.yml', 'other.json', rpcPath]; |
||||
|
zippedPaths = []; |
||||
|
mockPaths.forEach(name => { |
||||
|
const result = entryData({ name }); |
||||
|
if (result) zippedPaths.push(name); |
||||
|
}); |
||||
|
}, |
||||
|
); |
||||
|
zipStream.append = jest.fn(); |
||||
|
zipStream.finalize = jest.fn(); |
||||
|
// attach a func to emit events on the stream from the tests
|
||||
|
zipStream.mockEmit = (event: any, data: any) => zipStream.emit(event, data); |
||||
|
|
||||
|
return zipStream; |
||||
|
}); |
||||
}); |
}); |
||||
|
|
||||
it("fails to unzip something that doesn't exist", async () => { |
afterEach(() => { |
||||
return expect(unzip('foobar', 'bazfoo')).rejects.toThrow(); |
if (unzipStream) unzipStream.destroy(); |
||||
|
if (zipStream) zipStream.destroy(); |
||||
}); |
}); |
||||
}); |
|
||||
|
|
||||
describe('zip', () => { |
|
||||
// it('zips objects', async () => {
|
|
||||
// const objects: Array<{ name: string; object: any }> = [
|
|
||||
// {
|
|
||||
// name: 'firstObject',
|
|
||||
// object: 2,
|
|
||||
// },
|
|
||||
// {
|
|
||||
// name: 'secondObject',
|
|
||||
// object: { baz: 'baz' },
|
|
||||
// },
|
|
||||
// {
|
|
||||
// name: 'thirdObject',
|
|
||||
// object: [2, { foo: 'foo' }, false],
|
|
||||
// },
|
|
||||
// ];
|
|
||||
|
|
||||
// const zipped = join(tmpdir(), `zip-test-${Date.now()}.zip`);
|
|
||||
// await zip({
|
|
||||
// destination: zipped,
|
|
||||
// objects,
|
|
||||
// paths: [],
|
|
||||
// });
|
|
||||
|
|
||||
// const unzipped = join(tmpdir(), `zip-test-${Date.now()}`);
|
|
||||
// await unzip(zipped, unzipped);
|
|
||||
|
|
||||
// for (const obj of objects) {
|
|
||||
// const read = await fsExtra
|
|
||||
// .readFile(join(unzipped, obj.name))
|
|
||||
// .then(read => JSON.parse(read.toString('utf-8')));
|
|
||||
// expect(read).toEqual(obj.object);
|
|
||||
// }
|
|
||||
// });
|
|
||||
|
|
||||
// it('zips paths', async () => {
|
|
||||
// const files = [
|
|
||||
// join(__dirname, 'tests', 'resources', 'bar.txt'),
|
|
||||
// join(__dirname, 'tests', 'resources', 'foo.json'),
|
|
||||
// join(__dirname, 'tests', 'resources', 'baz'),
|
|
||||
// ];
|
|
||||
// const zipped = join(tmpdir(), `zip-test-${Date.now()}.zip`);
|
|
||||
// await zip({ destination: zipped, objects: [], paths: files });
|
|
||||
|
|
||||
// const unzipped = join(tmpdir(), `zip-test-${Date.now()}`);
|
|
||||
// await unzip(zipped, unzipped);
|
|
||||
|
|
||||
// const entries = await fs..readdir(unzipped, { withFileTypes: true });
|
|
||||
|
|
||||
// const bar = entries.find(e => e.name === 'bar.txt');
|
describe('unzip', () => { |
||||
// const baz = entries.find(e => e.name === 'baz');
|
it('should unzip a file successfully', async () => { |
||||
// const foo = entries.find(e => e.name === 'foo.json');
|
const promise = unzip('source-path', 'destination-path'); |
||||
|
// emit an error after a small delay
|
||||
// expect(bar?.isFile()).toBeTruthy();
|
await delay(100); |
||||
// expect(baz?.isDirectory()).toBeTruthy();
|
unzipStream.emit('close'); |
||||
// expect(foo?.isFile()).toBeTruthy();
|
await expect(promise).resolves.toBeUndefined(); |
||||
// });
|
}); |
||||
|
|
||||
it('should fail if there is an archiver error', async () => { |
it("fail to unzip something that isn't a zip", async () => { |
||||
fsMock.createWriteStream.mockReturnValueOnce(new PassThrough() as any); |
const promise = unzip('source-path', 'destination-file'); |
||||
|
// fail the unzip stream after a small delay
|
||||
const promise = zip('source', 'destination'); |
await delay(100); |
||||
|
unzipStream.emit('error', new Error('test-error')); |
||||
// emit an error after a small delay
|
await expect(promise).rejects.toThrow(); |
||||
const mockError = new Error('test-error'); |
}); |
||||
setTimeout(() => { |
|
||||
archiverMock.mockEmit('error', mockError); |
it("fails to unzip something that doesn't exist", async () => { |
||||
}, 100); |
fsMock.pathExists.mockResolvedValue(false as never); |
||||
|
return expect(unzip('source-path', 'destination-file')).rejects.toThrow(); |
||||
await expect(promise).rejects.toEqual(mockError); |
}); |
||||
}); |
}); |
||||
|
|
||||
it('should fail if there is an archiver warning', async () => { |
describe('zip', () => { |
||||
fsMock.createWriteStream.mockReturnValueOnce(new PassThrough() as any); |
it('should zip a folder successfully', async () => { |
||||
|
const promise = zip('source-path', 'destination-path'); |
||||
const promise = zip('source', 'destination'); |
// emit an error after a small delay
|
||||
|
await delay(100); |
||||
// emit an error after a small delay
|
zipStream.emit('finish'); |
||||
const mockError = new Error('test-warning'); |
await expect(promise).resolves.toBeUndefined(); |
||||
setTimeout(() => { |
}); |
||||
archiverMock.mockEmit('warning', mockError); |
|
||||
}, 100); |
it('should not include the c-lightning RPC in the zip', async () => { |
||||
|
const promise = zip('source-path', 'destination-path'); |
||||
await expect(promise).rejects.toEqual(mockError); |
// emit an error after a small delay
|
||||
|
await delay(100); |
||||
|
zipStream.emit('finish'); |
||||
|
await expect(promise).resolves.toBeUndefined(); |
||||
|
expect(zippedPaths).not.toContain(rpcPath); |
||||
|
expect(zippedPaths).toEqual(['test.yml', 'other.json']); |
||||
|
}); |
||||
|
|
||||
|
it('should fail if there is an archiver error', async () => { |
||||
|
const promise = zip('source-path', 'destination-path'); |
||||
|
// emit an error after a small delay
|
||||
|
await delay(100); |
||||
|
const mockError = new Error('test-error'); |
||||
|
zipStream.emit('error', mockError); |
||||
|
await expect(promise).rejects.toEqual(mockError); |
||||
|
}); |
||||
|
|
||||
|
it('should fail if there is an archiver warning', async () => { |
||||
|
const promise = zip('source-path', 'destination-path'); |
||||
|
// emit a warning after a small delay
|
||||
|
const mockError = new Error('test-warning'); |
||||
|
zipStream.emit('warning', mockError); |
||||
|
await expect(promise).rejects.toEqual(mockError); |
||||
|
}); |
||||
}); |
}); |
||||
}); |
}); |
||||
|
|||||
Loading…
Reference in new issue