diff --git a/packages/asset-swapper/test/dex_sampler_test.ts b/packages/asset-swapper/test/dex_sampler_test.ts index e13ec25e22..ec6c4916b9 100644 --- a/packages/asset-swapper/test/dex_sampler_test.ts +++ b/packages/asset-swapper/test/dex_sampler_test.ts @@ -12,8 +12,8 @@ import { getRandomInteger, provider, randomAddress, - txDefaults, toBaseUnitAmount, + txDefaults, } from '@0x/contracts-test-utils'; import { assetDataUtils, generatePseudoRandomSalt } from '@0x/order-utils'; import { SignedOrder } from '@0x/types'; @@ -172,12 +172,49 @@ describe('DexSampler tests', () => { registry, ), ); - expect(result).to.deep.equal([[{ - source: 'LiquidityProvider', - output: toBaseUnitAmount(1001), - input: toBaseUnitAmount(1000), - }]]); - }) + expect(result).to.deep.equal([ + [ + { + source: 'LiquidityProvider', + output: toBaseUnitAmount(1001), + input: toBaseUnitAmount(1000), + }, + ], + ]); + }); + + it('getLiquidityProviderBuyQuotes()', async () => { + const expectedMakerToken = randomAddress(); + const expectedTakerToken = randomAddress(); + const registry = randomAddress(); + const sampler = new MockSamplerContract({ + sampleBuysFromLiquidityProviderRegistry: (registryAddress, takerToken, makerToken, fillAmounts) => { + expect(registryAddress).to.eq(registry); + expect(takerToken).to.eq(expectedTakerToken); + expect(makerToken).to.eq(expectedMakerToken); + return [toBaseUnitAmount(999)]; + }, + }); + const dexOrderSampler = new DexOrderSampler(sampler); + const [result] = await dexOrderSampler.executeAsync( + DexOrderSampler.ops.getBuyQuotes( + [ERC20BridgeSource.LiquidityProvider], + expectedMakerToken, + expectedTakerToken, + [toBaseUnitAmount(1000)], + registry, + ), + ); + expect(result).to.deep.equal([ + [ + { + source: 'LiquidityProvider', + output: toBaseUnitAmount(999), + input: toBaseUnitAmount(1000), + }, + ], + ]); + }); it('getEth2DaiSellQuotes()', async () => { const expectedTakerToken = randomAddress(); diff --git a/packages/asset-swapper/test/market_operation_utils_test.ts b/packages/asset-swapper/test/market_operation_utils_test.ts index 0485dc2323..7e92f6d4c3 100644 --- a/packages/asset-swapper/test/market_operation_utils_test.ts +++ b/packages/asset-swapper/test/market_operation_utils_test.ts @@ -11,7 +11,7 @@ import { import { Web3Wrapper } from '@0x/dev-utils'; import { assetDataUtils, generatePseudoRandomSalt } from '@0x/order-utils'; import { AssetProxyId, ERC20BridgeAssetData, SignedOrder } from '@0x/types'; -import { BigNumber, hexUtils, NULL_ADDRESS, fromTokenUnitAmount } from '@0x/utils'; +import { BigNumber, fromTokenUnitAmount, hexUtils, NULL_ADDRESS } from '@0x/utils'; import * as _ from 'lodash'; import { constants as assetSwapperConstants } from '../src/constants';