added more unit tests for PLP DEX sampling

packages/asset-swapper/src/utils/market_operation_utils/sampler.ts

@@ -75,6 +75,40 @@ const samplerOperations = {
             },
         };
     },
+    getPLPSellQuotes(
+        plpRegistryAddress: string,
+        takerToken: string,
+        makerToken: string,
+        takerFillAmounts: BigNumber[],
+    ): BatchedOperation<BigNumber[]> {
+        return {
+            encodeCall: contract => {
+                return contract
+                    .sampleSellsFromLiquidityProviderRegistry(plpRegistryAddress, takerToken, makerToken, takerFillAmounts)
+                    .getABIEncodedTransactionData();
+            },
+            handleCallResultsAsync: async (contract, callResults) => {
+                return contract.getABIDecodedReturnData<BigNumber[]>('sampleSellsFromLiquidityProviderRegistry', callResults);
+            },
+        };
+    },
+    getPLPBuyQuotes(
+        plpRegistryAddress: string,
+        takerToken: string,
+        makerToken: string,
+        makerFillAmounts: BigNumber[],
+    ): BatchedOperation<BigNumber[]> {
+        return {
+            encodeCall: contract => {
+                return contract
+                    .sampleBuysFromLiquidityProviderRegistry(plpRegistryAddress, takerToken, makerToken, makerFillAmounts)
+                    .getABIEncodedTransactionData();
+            },
+            handleCallResultsAsync: async (contract, callResults) => {
+                return contract.getABIDecodedReturnData<BigNumber[]>('sampleBuysFromLiquidityProviderRegistry', callResults);
+            },
+        };
+    },
     getEth2DaiSellQuotes(
         makerToken: string,
         takerToken: string,
@@ -186,12 +220,12 @@ const samplerOperations = {
     },
     getLiquidityProviderFromRegistry(
         registryAddress: string,
-        makerToken: string,
         takerToken: string,
+        makerToken: string,
     ): BatchedOperation<string> {
         return {
             encodeCall: contract => {
-                return contract.getLiquidityProviderFromRegistry(registryAddress, makerToken, takerToken).getABIEncodedTransactionData()
+                return contract.getLiquidityProviderFromRegistry(registryAddress, takerToken, makerToken).getABIEncodedTransactionData();
             },
             handleCallResultsAsync: async (contract, callResults) => {
                 return contract.getABIDecodedReturnData<string>('getLiquidityProviderFromRegistry', callResults);
@@ -203,6 +237,7 @@ const samplerOperations = {
         makerToken: string,
         takerToken: string,
         takerFillAmounts: BigNumber[],
+        plpRegistryAddress?: string | undefined,
     ): BatchedOperation<DexSample[][]> {
         const subOps = sources
             .map(source => {
@@ -229,6 +264,13 @@ const samplerOperations = {
                             takerFillAmounts,
                         );
                     }
+                } else if (source === ERC20BridgeSource.Plp) {
+                    if (plpRegistryAddress === undefined) {
+                        throw new Error('Cannot sample liquidity from a PLP liquidity pool, if a registry is not provided.');
+                    }
+                    batchedOperation = samplerOperations.getPLPSellQuotes(
+                        plpRegistryAddress, takerToken, makerToken, takerFillAmounts,
+                    );
                 } else {
                         throw new Error(`Unsupported sell sample source: ${source}`);
                 }
@@ -265,12 +307,20 @@ const samplerOperations = {
         makerToken: string,
         takerToken: string,
         makerFillAmounts: BigNumber[],
+        plpRegistryAddress?: string | undefined,
     ): BatchedOperation<DexSample[][]> {
         const subOps = sources.map(source => {
             if (source === ERC20BridgeSource.Eth2Dai) {
                 return samplerOperations.getEth2DaiBuyQuotes(makerToken, takerToken, makerFillAmounts);
             } else if (source === ERC20BridgeSource.Uniswap) {
                 return samplerOperations.getUniswapBuyQuotes(makerToken, takerToken, makerFillAmounts);
+            } else if (source === ERC20BridgeSource.Plp) {
+                if (plpRegistryAddress === undefined) {
+                    throw new Error('Cannot sample liquidity from a PLP liquidity pool, if a registry is not provided.');
+                }
+                return samplerOperations.getPLPBuyQuotes(
+                    plpRegistryAddress, takerToken, makerToken, makerFillAmounts,
+                );
             } else {
                 throw new Error(`Unsupported buy sample source: ${source}`);
             }

packages/asset-swapper/src/utils/market_operation_utils/types.ts

@@ -31,6 +31,7 @@ export enum ERC20BridgeSource {
     CurveUsdcDai = 'Curve_USDC_DAI',
     CurveUsdcDaiUsdt = 'Curve_USDC_DAI_USDT',
     CurveUsdcDaiUsdtTusd = 'Curve_USDC_DAI_USDT_TUSD',
+    Plp = 'PLP',
 }
 
 // Internal `fillData` field for `Fill` objects.

packages/asset-swapper/test/dex_sampler_test.ts

@@ -1,14 +1,14 @@
-import { constants, expect, getRandomFloat, getRandomInteger, randomAddress, provider, txDefaults } from '@0x/contracts-test-utils';
+import { constants, expect, getRandomFloat, getRandomInteger, provider, randomAddress, txDefaults } from '@0x/contracts-test-utils';
 import { assetDataUtils, generatePseudoRandomSalt } from '@0x/order-utils';
 import { SignedOrder } from '@0x/types';
 import { BigNumber, hexUtils, NULL_ADDRESS } from '@0x/utils';
 import * as _ from 'lodash';
 
 import { DexOrderSampler, getSampleAmounts } from '../src/utils/market_operation_utils/sampler';
-import { ERC20BridgeSource } from '../src/utils/market_operation_utils/types';
+import { ERC20BridgeSource, DexSample } from '../src/utils/market_operation_utils/types';
 
 import { MockSamplerContract } from './utils/mock_sampler_contract';
-import { DummyLiquidityProviderRegistryContract, ERC20BridgeSamplerContract, IERC20BridgeSamplerContract } from '@0x/contract-wrappers';
+import { DummyLiquidityProviderRegistryContract, ERC20BridgeSamplerContract, IERC20BridgeSamplerContract, DummyLiquidityProviderContract } from '@0x/contract-wrappers';
 import { artifacts as erc20BridgeSamplerArtifacts } from '@0x/contracts-erc20-bridge-sampler/lib/src/artifacts';
 
 const CHAIN_ID = 1;
@@ -329,31 +329,36 @@ describe('DexSampler tests', () => {
             expect(quotes).to.deep.eq(expectedQuotes);
         });
 
-        it.only('getLiquidityProviderFromRegistry()', async () => {
+        describe.only('PLP Operations', () => {
             const xAsset = randomAddress();
             const yAsset = randomAddress();
             const zAsset = randomAddress();
             const liquidityPool1 = randomAddress();
             const liquidityPool2 = randomAddress();
 
-            // Deploy Registry
-            const registryContract = await DummyLiquidityProviderRegistryContract.deployFrom0xArtifactAsync(
+            let registryContract: DummyLiquidityProviderRegistryContract;
+            let samplerContract: ERC20BridgeSamplerContract;
+            beforeEach(async () => {
+                registryContract = await DummyLiquidityProviderRegistryContract.deployFrom0xArtifactAsync(
                     erc20BridgeSamplerArtifacts.DummyLiquidityProviderRegistry,
                     provider,
                     txDefaults,
                     {},
                 );
-            // Write 2 new liquidity pools
-            await registryContract.setLiquidityProviderForMarket(xAsset, yAsset, liquidityPool1).awaitTransactionSuccessAsync();
-            await registryContract.setLiquidityProviderForMarket(xAsset, zAsset, liquidityPool2).awaitTransactionSuccessAsync();
-
-            // Deploy the sampler
-            const samplerContract = await ERC20BridgeSamplerContract.deployFrom0xArtifactAsync(
+                samplerContract = await ERC20BridgeSamplerContract.deployFrom0xArtifactAsync(
                     erc20BridgeSamplerArtifacts.ERC20BridgeSampler,
                     provider,
                     txDefaults,
                     {},
                 );
+            });
+            it('getLiquidityProviderFromRegistry()', async () => {
+                // Deploy Registry
+                // Write 2 new liquidity pools
+                await registryContract.setLiquidityProviderForMarket(xAsset, yAsset, liquidityPool1).awaitTransactionSuccessAsync().txHashPromise;
+                await registryContract.setLiquidityProviderForMarket(xAsset, zAsset, liquidityPool2).awaitTransactionSuccessAsync().txHashPromise;
+
+                // Deploy the sampler
 
                 // Test multiple combinations: 2 pools that are present, 1 pool that is not present.
                 const dexOrderSampler = new DexOrderSampler(new IERC20BridgeSamplerContract(samplerContract.address, provider));
@@ -366,6 +371,49 @@ describe('DexSampler tests', () => {
                 expect(xzPool).to.eql(liquidityPool2);
                 expect(yzPool).to.eql(NULL_ADDRESS);
             });
+            it('is able to sample DEX liquidity from PLP', async () => {
+                const fakeLiquidityPool = await DummyLiquidityProviderContract.deployFrom0xArtifactAsync(
+                    erc20BridgeSamplerArtifacts.DummyLiquidityProvider,
+                    provider,
+                    txDefaults,
+                    {},
+                );
+                await registryContract.setLiquidityProviderForMarket(xAsset, yAsset, fakeLiquidityPool.address).awaitTransactionSuccessAsync().txHashPromise;
+
+                const dexOrderSampler = new DexOrderSampler(new IERC20BridgeSamplerContract(samplerContract.address, provider));
+                const [buyQuotes, sellQuotes] = await dexOrderSampler.executeBatchAsync([
+                    DexOrderSampler.ops.getBuyQuotes(
+                        [ERC20BridgeSource.Plp],
+                        xAsset,
+                        yAsset,
+                        [new BigNumber(10), new BigNumber(100)],
+                        registryContract.address,
+                    ),
+                    DexOrderSampler.ops.getSellQuotes(
+                        [ERC20BridgeSource.Plp],
+                        xAsset,
+                        yAsset,
+                        [new BigNumber(10), new BigNumber(100), new BigNumber(500)],
+                        registryContract.address,
+                    ),
+                ]);
+                expect(buyQuotes.length).to.eql(1);
+                const plpBuyQuotes: DexSample[] = buyQuotes[0];
+                expect(plpBuyQuotes.length).to.eql(2);
+                for(const quote of plpBuyQuotes) {
+                    expect(quote.source).to.bignumber.eql(ERC20BridgeSource.Plp);
+                    expect(quote.input.plus(1)).to.bignumber.eql(quote.output);
+                }
+
+                expect(sellQuotes.length).to.eql(1);
+                const plpSellQuotes: DexSample[] = sellQuotes[0];
+                expect(plpSellQuotes.length).to.eql(3);
+                for(const quote of plpSellQuotes) {
+                    expect(quote.source).to.bignumber.eql(ERC20BridgeSource.Plp);
+                    expect(quote.input.minus(1)).to.bignumber.eql(quote.output);
+                }
+            })
+        });
     });
 
     describe('batched operations', () => {