25 lines
1014 B
TypeScript
25 lines
1014 B
TypeScript
import { Web3Wrapper } from '@0xproject/web3-wrapper';
|
|
import * as Web3 from 'web3';
|
|
|
|
export class BlockchainLifecycle {
|
|
private _web3Wrapper: Web3Wrapper;
|
|
private _snapshotIdsStack: number[];
|
|
constructor(web3Wrapper: Web3Wrapper) {
|
|
this._web3Wrapper = web3Wrapper;
|
|
this._snapshotIdsStack = [];
|
|
}
|
|
// TODO: In order to run these tests on an actual node, we should check if we are running against
|
|
// TestRPC, if so, use snapshots, otherwise re-deploy contracts before every test
|
|
public async startAsync(): Promise<void> {
|
|
const snapshotId = await this._web3Wrapper.takeSnapshotAsync();
|
|
this._snapshotIdsStack.push(snapshotId);
|
|
}
|
|
public async revertAsync(): Promise<void> {
|
|
const snapshotId = this._snapshotIdsStack.pop() as number;
|
|
const didRevert = await this._web3Wrapper.revertSnapshotAsync(snapshotId);
|
|
if (!didRevert) {
|
|
throw new Error(`Snapshot with id #${snapshotId} failed to revert`);
|
|
}
|
|
}
|
|
}
|