aboutsummaryrefslogtreecommitdiffstats
path: root/test/utils/blockchain_lifecycle.ts
diff options
context:
space:
mode:
Diffstat (limited to 'test/utils/blockchain_lifecycle.ts')
-rw-r--r--test/utils/blockchain_lifecycle.ts19
1 files changed, 19 insertions, 0 deletions
diff --git a/test/utils/blockchain_lifecycle.ts b/test/utils/blockchain_lifecycle.ts
new file mode 100644
index 000000000..f7342b680
--- /dev/null
+++ b/test/utils/blockchain_lifecycle.ts
@@ -0,0 +1,19 @@
+import {RPC} from './rpc';
+
+export class BlockchainLifecycle {
+ 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 startAsync(): Promise<void> {
+ this.snapshotId = await this.rpc.takeSnapshotAsync();
+ }
+ public async revertAsync(): Promise<void> {
+ const didRevert = await this.rpc.revertSnapshotAsync(this.snapshotId);
+ if (!didRevert) {
+ throw new Error(`Snapshot with id #${this.snapshotId} failed to revert`);
+ }
+ }
+};