3 files changed, 85 insertions, 0 deletions
diff --git a/packages/dev-utils/src/blockchain_lifecycle.ts b/packages/dev-utils/src/blockchain_lifecycle.ts
new file mode 100644
index 000000000..18f5d5c61
--- /dev/null
+++ b/packages/dev-utils/src/blockchain_lifecycle.ts
@@ -0,0 +1,23 @@
+import { RPC } from './rpc';
+
+export class BlockchainLifecycle {
+    private _rpc: RPC;
+    private _snapshotIdsStack: number[];
+    constructor(url: string) {
+        this._rpc = new RPC(url);
+        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._rpc.takeSnapshotAsync();
+        this._snapshotIdsStack.push(snapshotId);
+    }
+    public async revertAsync(): Promise<void> {
+        const snapshotId = this._snapshotIdsStack.pop() as number;
+        const didRevert = await this._rpc.revertSnapshotAsync(snapshotId);
+        if (!didRevert) {
+            throw new Error(`Snapshot with id #${snapshotId} failed to revert`);
+        }
+    }
+}
diff --git a/packages/dev-utils/src/index.ts b/packages/dev-utils/src/index.ts
new file mode 100644
index 000000000..9ba0cb5cf
--- /dev/null
+++ b/packages/dev-utils/src/index.ts
@@ -0,0 +1,2 @@
+export { RPC } from './rpc';
+export { BlockchainLifecycle } from './blockchain_lifecycle';
diff --git a/packages/dev-utils/src/rpc.ts b/packages/dev-utils/src/rpc.ts
new file mode 100644
index 000000000..36f8b1ef9
--- /dev/null
+++ b/packages/dev-utils/src/rpc.ts
@@ -0,0 +1,60 @@
+import * as ethUtil from 'ethereumjs-util';
+import * as request from 'request-promise-native';
+
+export class RPC {
+    private _url: string;
+    private _id: number;
+    constructor(url: string) {
+        this._url = url;
+        this._id = 0;
+    }
+    public async takeSnapshotAsync(): Promise<number> {
+        const method = 'evm_snapshot';
+        const params: any[] = [];
+        const payload = this._toPayload(method, params);
+        const snapshotIdHex = await this._sendAsync(payload);
+        const snapshotId = ethUtil.bufferToInt(ethUtil.toBuffer(snapshotIdHex));
+        return snapshotId;
+    }
+    public async revertSnapshotAsync(snapshotId: number): Promise<boolean> {
+        const method = 'evm_revert';
+        const params = [snapshotId];
+        const payload = this._toPayload(method, params);
+        const didRevert = await this._sendAsync(payload);
+        return didRevert;
+    }
+    public async increaseTimeAsync(time: number) {
+        const method = 'evm_increaseTime';
+        const params = [time];
+        const payload = this._toPayload(method, params);
+        return this._sendAsync(payload);
+    }
+    public async mineBlockAsync(): Promise<void> {
+        const method = 'evm_mine';
+        const params: any[] = [];
+        const payload = this._toPayload(method, params);
+        await this._sendAsync(payload);
+    }
+    private _toPayload(method: string, params: any[] = []): string {
+        const payload = JSON.stringify({
+            id: this._id,
+            method,
+            params,
+        });
+        this._id += 1;
+        return payload;
+    }
+    private async _sendAsync(payload: string): Promise<any> {
+        const opts = {
+            method: 'POST',
+            uri: this._url,
+            body: payload,
+            headers: {
+                'content-type': 'application/json',
+            },
+        };
+        const bodyString = await request(opts);
+        const body = JSON.parse(bodyString);
+        return body.result;
+    }
+}