first commit

test/utils/EVM.ts

@@ -0,0 +1,83 @@
+/*
+
+    Copyright 2020 DODO ZOO.
+    SPDX-License-Identifier: Apache-2.0
+
+*/
+
+// require('dotenv-flow').config();
+
+import { JsonRpcPayload, JsonRpcResponse } from 'web3-core-helpers';
+import Web3 from 'web3';
+
+export function getDefaultWeb3() {
+  return new Web3(process.env.RPC_NODE_URI)
+}
+
+export class EVM {
+  private provider = new Web3.providers.HttpProvider(process.env.RPC_NODE_URI);
+
+  public async reset(id: string): Promise<string> {
+    if (!id) {
+      throw new Error('id must be set');
+    }
+
+    await this.callJsonrpcMethod('evm_revert', [id]);
+
+    return this.snapshot();
+  }
+
+  public async snapshot(): Promise<string> {
+    return this.callJsonrpcMethod('evm_snapshot');
+  }
+
+  public async evmRevert(id: string): Promise<string> {
+    return this.callJsonrpcMethod('evm_revert', [id]);
+  }
+
+  public async stopMining(): Promise<string> {
+    return this.callJsonrpcMethod('miner_stop');
+  }
+
+  public async startMining(): Promise<string> {
+    return this.callJsonrpcMethod('miner_start');
+  }
+
+  public async mineBlock(): Promise<string> {
+    return this.callJsonrpcMethod('evm_mine');
+  }
+
+  public async increaseTime(duration: number): Promise<string> {
+    return this.callJsonrpcMethod('evm_increaseTime', [duration]);
+  }
+
+  public async callJsonrpcMethod(method: string, params?: (any[])): Promise<string> {
+    const args: JsonRpcPayload = {
+      method,
+      params,
+      jsonrpc: '2.0',
+      id: new Date().getTime(),
+    };
+
+    const response = await this.send(args);
+
+    return response.result;
+  }
+
+  private async send(args: JsonRpcPayload): Promise<any> {
+    return new Promise((resolve, reject) => {
+      const callback: any = (error: Error, val: JsonRpcResponse): void => {
+        if (error) {
+          reject(error);
+        } else {
+          resolve(val);
+        }
+      };
+
+      this.provider.send(
+        args,
+        callback,
+      );
+    });
+  }
+}