From 30c6fe08ec5f67fa3679c0043a24ffc420974964 Mon Sep 17 00:00:00 2001
From: Brandon Millman <brandon.millman@gmail.com>
Date: Wed, 1 Aug 2018 15:08:34 -0700
Subject: Add tests

---
 .../src/utils/market_orders_optimization_utils.ts  | 37 +++++++++++-----------
 1 file changed, 19 insertions(+), 18 deletions(-)

(limited to 'packages/contract-wrappers/src/utils')

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,
+    };
+};
-- 
cgit v1.2.3