aboutsummaryrefslogtreecommitdiffstats
path: root/packages
diff options
context:
space:
mode:
authorBrandon Millman <brandon.millman@gmail.com>2018-08-02 06:08:34 +0800
committerBrandon Millman <brandon.millman@gmail.com>2018-08-02 11:47:59 +0800
commit30c6fe08ec5f67fa3679c0043a24ffc420974964 (patch)
tree5db5611ce603f59a626e34e7e222144e6bf9fa96 /packages
parent7c864b81e0d958560e098ebd8bd241385e4aadff (diff)
downloaddexon-sol-tools-30c6fe08ec5f67fa3679c0043a24ffc420974964.tar
dexon-sol-tools-30c6fe08ec5f67fa3679c0043a24ffc420974964.tar.gz
dexon-sol-tools-30c6fe08ec5f67fa3679c0043a24ffc420974964.tar.bz2
dexon-sol-tools-30c6fe08ec5f67fa3679c0043a24ffc420974964.tar.lz
dexon-sol-tools-30c6fe08ec5f67fa3679c0043a24ffc420974964.tar.xz
dexon-sol-tools-30c6fe08ec5f67fa3679c0043a24ffc420974964.tar.zst
dexon-sol-tools-30c6fe08ec5f67fa3679c0043a24ffc420974964.zip
Add tests
Diffstat (limited to 'packages')
-rw-r--r--packages/contract-wrappers/src/utils/market_orders_optimization_utils.ts37
-rw-r--r--packages/contract-wrappers/test/market_orders_optimization_utils_test.ts66
2 files changed, 85 insertions, 18 deletions
diff --git a/packages/contract-wrappers/src/utils/market_orders_optimization_utils.ts b/packages/contract-wrappers/src/utils/market_orders_optimization_utils.ts
index 0041f1a64..e35b2eadc 100644
--- a/packages/contract-wrappers/src/utils/market_orders_optimization_utils.ts
+++ b/packages/contract-wrappers/src/utils/market_orders_optimization_utils.ts
@@ -11,15 +11,12 @@ export const marketOrdersOptimizationUtils = {
* @returns optimized orders
*/
optimizeMarketOrders(orders: SignedOrder[]): SignedOrder[] {
- const optimizedOrders = _.map(orders, (order, index) => {
- const makerAssetData = index === 0 ? order.makerAssetData : constants.NULL_BYTES;
- const takerAssetData = constants.NULL_BYTES;
- return {
- ...order,
- makerAssetData,
- takerAssetData,
- };
- });
+ const optimizedOrders = _.map(orders, (order, index) =>
+ transformOrder(order, {
+ makerAssetData: index === 0 ? order.makerAssetData : constants.NULL_BYTES,
+ takerAssetData: constants.NULL_BYTES,
+ }),
+ );
return optimizedOrders;
},
/**
@@ -29,15 +26,19 @@ export const marketOrdersOptimizationUtils = {
* @returns optimized orders
*/
optimizeFeeOrders(orders: SignedOrder[]): SignedOrder[] {
- const optimizedOrders = _.map(orders, order => {
- const makerAssetData = constants.NULL_BYTES;
- const takerAssetData = constants.NULL_BYTES;
- return {
- ...order,
- makerAssetData,
- takerAssetData,
- };
- });
+ const optimizedOrders = _.map(orders, (order, index) =>
+ transformOrder(order, {
+ makerAssetData: constants.NULL_BYTES,
+ takerAssetData: constants.NULL_BYTES,
+ }),
+ );
return optimizedOrders;
},
};
+
+const transformOrder = (order: SignedOrder, partialOrder: Partial<SignedOrder>) => {
+ return {
+ ...order,
+ ...partialOrder,
+ };
+};
diff --git a/packages/contract-wrappers/test/market_orders_optimization_utils_test.ts b/packages/contract-wrappers/test/market_orders_optimization_utils_test.ts
new file mode 100644
index 000000000..742294df2
--- /dev/null
+++ b/packages/contract-wrappers/test/market_orders_optimization_utils_test.ts
@@ -0,0 +1,66 @@
+import { orderFactory } from '@0xproject/order-utils';
+import * as chai from 'chai';
+import * as _ from 'lodash';
+import 'mocha';
+
+import { constants } from '../src/utils/constants';
+import { marketOrdersOptimizationUtils } from '../src/utils/market_orders_optimization_utils';
+
+import { chaiSetup } from './utils/chai_setup';
+import { assert } from '../src/utils/assert';
+import { NULL_BYTES } from '@0xproject/utils';
+
+chaiSetup.configure();
+const expect = chai.expect;
+
+// utility for generating a set of order objects with mostly NULL values
+// except for a specified makerAssetData and takerAssetData
+const FAKE_ORDERS_COUNT = 5;
+const generateFakeOrders = (makerAssetData: string, takerAssetData: string) =>
+ _.map(_.range(FAKE_ORDERS_COUNT), index => {
+ const order = orderFactory.createOrder(
+ constants.NULL_ADDRESS,
+ constants.NULL_ADDRESS,
+ constants.NULL_ADDRESS,
+ constants.ZERO_AMOUNT,
+ constants.ZERO_AMOUNT,
+ constants.ZERO_AMOUNT,
+ makerAssetData,
+ constants.ZERO_AMOUNT,
+ takerAssetData,
+ constants.NULL_ADDRESS,
+ constants.NULL_ADDRESS,
+ );
+ return {
+ ...order,
+ signature: 'dummy signature',
+ };
+ });
+
+describe('marketOrdersOptimizationUtils', () => {
+ const fakeMakerAssetData = 'fakeMakerAssetData';
+ const fakeTakerAssetData = 'fakeTakerAssetData';
+ const orders = generateFakeOrders(fakeMakerAssetData, fakeTakerAssetData);
+ describe('#optimizeMarketOrders', () => {
+ it('should make makerAssetData `0x` unless first order', () => {
+ const optimizedOrders = marketOrdersOptimizationUtils.optimizeMarketOrders(orders);
+ expect(optimizedOrders[0].makerAssetData).to.equal(fakeMakerAssetData);
+ const ordersWithoutHead = _.slice(optimizedOrders, 1);
+ _.forEach(ordersWithoutHead, order => expect(order.makerAssetData).to.equal(constants.NULL_BYTES));
+ });
+ it('should make all takerAssetData `0x`', () => {
+ const optimizedOrders = marketOrdersOptimizationUtils.optimizeMarketOrders(orders);
+ _.forEach(optimizedOrders, order => expect(order.takerAssetData).to.equal(constants.NULL_BYTES));
+ });
+ });
+ describe('#optimizeFeeOrders', () => {
+ it('should make all makerAssetData `0x`', () => {
+ const optimizedOrders = marketOrdersOptimizationUtils.optimizeFeeOrders(orders);
+ _.forEach(optimizedOrders, order => expect(order.makerAssetData).to.equal(constants.NULL_BYTES));
+ });
+ it('should make all takerAssetData `0x`', () => {
+ const optimizedOrders = marketOrdersOptimizationUtils.optimizeFeeOrders(orders);
+ _.forEach(optimizedOrders, order => expect(order.takerAssetData).to.equal(constants.NULL_BYTES));
+ });
+ });
+});