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

refactor tests, and add more tests

Browse Source
This commit is contained in:
Daniel Pyrathon 2020-02-28 12:28:09 -08:00
parent e5df51a83a
commit bcc3e5ebb0
1 changed files with 19 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

23
packages/asset-swapper/test/market_operation_utils_test.ts
View File

@ -159,7 +159,7 @@ describe('MarketOperationUtils tests', () => {
}; };
} }
function createGetMultipleSellQuotesOperationFromRatesAndRetainPLPParams(rates: RatesBySource): [{sources: ERC20BridgeSource[], plpRegistryAddress?: string}, GetMultipleQuotesOperation] { function callTradeOperationAndRetainPLPParams(tradeOperation: (rates: RatesBySource) => GetMultipleQuotesOperation, rates: RatesBySource): [{sources: ERC20BridgeSource[], plpRegistryAddress?: string}, GetMultipleQuotesOperation] {
const plpParams: {sources: ERC20BridgeSource[], plpRegistryAddress?: string} = { const plpParams: {sources: ERC20BridgeSource[], plpRegistryAddress?: string} = {
sources: [], sources: [],
plpRegistryAddress: undefined, plpRegistryAddress: undefined,
@ -167,7 +167,7 @@ describe('MarketOperationUtils tests', () => {
const fn = (sources: ERC20BridgeSource[], makerToken: string, takerToken: string, fillAmounts: BigNumber[], plpRegistryAddress: string | undefined) => { const fn = (sources: ERC20BridgeSource[], makerToken: string, takerToken: string, fillAmounts: BigNumber[], plpRegistryAddress: string | undefined) => {
plpParams.plpRegistryAddress = plpRegistryAddress; plpParams.plpRegistryAddress = plpRegistryAddress;
plpParams.sources = sources; plpParams.sources = sources;
return createGetMultipleSellQuotesOperationFromRates(rates)( return tradeOperation(rates)(
sources, makerToken, takerToken, fillAmounts, plpRegistryAddress, sources, makerToken, takerToken, fillAmounts, plpRegistryAddress,
); );
}; };
@ -366,7 +366,7 @@ describe('MarketOperationUtils tests', () => {
}); });
it('polls the liquidity provider when the registry is provided in the arguments', async () => { it('polls the liquidity provider when the registry is provided in the arguments', async () => {
const [args, fn] = createGetMultipleSellQuotesOperationFromRatesAndRetainPLPParams(DEFAULT_RATES); const [args, fn] = callTradeOperationAndRetainPLPParams(createGetMultipleSellQuotesOperationFromRates, DEFAULT_RATES);
replaceSamplerOps({ replaceSamplerOps({
getSellQuotes: fn, getSellQuotes: fn,
}); });
@ -651,6 +651,21 @@ describe('MarketOperationUtils tests', () => {
expect(sourcesPolled).to.deep.eq(BUY_SOURCES); expect(sourcesPolled).to.deep.eq(BUY_SOURCES);
}); });
it('polls the liquidity provider when the registry is provided in the arguments', async () => {
const [args, fn] = callTradeOperationAndRetainPLPParams(createGetMultipleBuyQuotesOperationFromRates, DEFAULT_RATES);
replaceSamplerOps({
getBuyQuotes: fn,
});
const registryAddress = randomAddress();
const newMarketOperationUtils = new MarketOperationUtils(MOCK_SAMPLER, contractAddresses, ORDER_DOMAIN, registryAddress);
await newMarketOperationUtils.getMarketBuyOrdersAsync(ORDERS, FILL_AMOUNT, {
...DEFAULT_OPTS,
excludedSources: [],
});
expect(args.sources.sort()).to.deep.eq(BUY_SOURCES.concat([ERC20BridgeSource.Plp]).sort());
expect(args.plpRegistryAddress).to.eql(registryAddress);
});
it('does not poll DEXes in `excludedSources`', async () => { it('does not poll DEXes in `excludedSources`', async () => {
const excludedSources = _.sampleSize(SELL_SOURCES, _.random(1, SELL_SOURCES.length)); const excludedSources = _.sampleSize(SELL_SOURCES, _.random(1, SELL_SOURCES.length));
let sourcesPolled: ERC20BridgeSource[] = []; let sourcesPolled: ERC20BridgeSource[] = [];
@ -838,7 +853,7 @@ describe('MarketOperationUtils tests', () => {
const yAsset = randomAddress(); const yAsset = randomAddress();
const toSell = Web3Wrapper.toBaseUnitAmount(10, 18); const toSell = Web3Wrapper.toBaseUnitAmount(10, 18);
const [getSellQuiotesParams, getSellQuotesFn] = createGetMultipleSellQuotesOperationFromRatesAndRetainPLPParams({ const [getSellQuiotesParams, getSellQuotesFn] = callTradeOperationAndRetainPLPParams(createGetMultipleSellQuotesOperationFromRates, {
[ERC20BridgeSource.Plp]: createDecreasingRates(5), [ERC20BridgeSource.Plp]: createDecreasingRates(5),
}); });
const [getLiquidityProviderParams, getLiquidityProviderFn] = getLiquidityProviderFromRegistryAndReturnCallParameters(liquidityPoolAddress); const [getLiquidityProviderParams, getLiquidityProviderFn] = getLiquidityProviderFromRegistryAndReturnCallParameters(liquidityPoolAddress);