crowetic/protocol
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

decouple state (SimulationEnvironment) and Simulation

Browse Source
This commit is contained in:
Michael Zhu
2019-11-05 11:08:58 -08:00
parent 3c7a0bcd85
commit 44f268a7ee
5 changed files with 36 additions and 29 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
contracts/integrations/test/simulation/pool_management_fuzz_test.ts
View File

@@ -6,19 +6,24 @@ import { PoolOperator, Staker } from '../actors';
import { DeploymentManager } from '../utils/deployment_manager';
import { AssertionResult } from '../utils/function_assertions';
import { Simulation } from './simulation';
import { Simulation, SimulationEnvironment } from './simulation';
class PoolManagementSimulation extends Simulation {
constructor(balanceStore: BlockchainBalanceStore, deployment: DeploymentManager) {
super(balanceStore, deployment);
constructor(environment: SimulationEnvironment) {
super(environment);
}
protected async *_assertionGenerator(): AsyncIterableIterator<AssertionResult> {
const staker = new Staker({ name: 'Staker', deployment: this._deployment, simulation: this });
await staker.configureERC20TokenAsync(this._deployment.tokens.zrx);
this.balanceStore.registerTokenOwner(staker.address, staker.name);
const { deployment, balanceStore } = this.environment;
const staker = new Staker({ name: 'Staker', deployment, simulationEnvironment: this.environment });
await staker.configureERC20TokenAsync(deployment.tokens.zrx);
balanceStore.registerTokenOwner(staker.address, staker.name);
const operator = new PoolOperator({ name: 'Operator', deployment: this._deployment, simulation: this });
const operator = new PoolOperator({
name: 'Operator',
deployment,
simulationEnvironment: this.environment,
});
const actions = [
staker.simulationActions.validStake,
@@ -49,7 +54,7 @@ blockchainTests.only('Pool management fuzz test', env => {
{ erc20: { ZRX: deployment.tokens.zrx } },
);
const sim = new PoolManagementSimulation(balanceStore, deployment);
const sim = new PoolManagementSimulation({ balanceStore, deployment });
return sim.fuzzAsync();
});
});

11
contracts/integrations/test/simulation/simulation.ts
View File

@@ -4,14 +4,15 @@ import * as _ from 'lodash';
import { DeploymentManager } from '../utils/deployment_manager';
import { AssertionResult } from '../utils/function_assertions';
export interface SimulationEnvironment {
balanceStore: BlockchainBalanceStore;
deployment: DeploymentManager;
}
export abstract class Simulation {
public poolIds = [];
private readonly _generator = this._assertionGenerator();
protected constructor(
public readonly balanceStore: BlockchainBalanceStore,
protected readonly _deployment: DeploymentManager,
) {}
protected constructor(public readonly environment: SimulationEnvironment) {}
public async stepAsync(): Promise<void> {
await this._generator.next();