Add tests

1 files changed, 19 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,
+    };
+};