1 files changed, 19 insertions, 0 deletions
diff --git a/test/utils/blockchain_clean.ts b/test/utils/blockchain_clean.ts
new file mode 100644
index 000000000..6468dbec7
--- /dev/null
+++ b/test/utils/blockchain_clean.ts
@@ -0,0 +1,19 @@
+import {RPC} from './rpc';
+
+export class BlockchainClean {
+    private rpc: RPC;
+    private snapshotId: number;
+    constructor() {
+        this.rpc = new RPC();
+    }
+    // TODO: Check if running on TestRPC or on actual node, if actual node, re-deploy contracts instead
+    public async setupAsync() {
+        this.snapshotId = await this.rpc.takeSnapshotAsync();
+    }
+    public async restoreAsync() {
+        const didRevert = await this.rpc.revertSnapshotAsync(this.snapshotId);
+        if (!didRevert) {
+            throw new Error(`Snapshot with id #${this.snapshotId} failed to revert`);
+        }
+    }
+};