33 lines
979 B
TypeScript
33 lines
979 B
TypeScript
import { BlockchainBalanceStore } from '@0x/contracts-exchange';
|
|
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 {
|
|
private readonly _generator = this._assertionGenerator();
|
|
|
|
protected constructor(public readonly environment: SimulationEnvironment) {}
|
|
|
|
public async stepAsync(): Promise<void> {
|
|
await this._generator.next();
|
|
}
|
|
|
|
public async fuzzAsync(steps?: number): Promise<void> {
|
|
if (steps !== undefined) {
|
|
_.times(steps, async () => await this.stepAsync());
|
|
} else {
|
|
while (true) {
|
|
await this.stepAsync();
|
|
}
|
|
}
|
|
}
|
|
|
|
protected abstract _assertionGenerator(): AsyncIterableIterator<AssertionResult>;
|
|
}
|