aboutsummaryrefslogtreecommitdiffstats
path: root/packages/order-utils/test/utils/test_order_factory.ts
blob: 2c5d8cf61414499bed61134cf7e964facb8d3ec9 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import { Order, OrderRelevantState, SignedOrder } from '@0xproject/types';
import { BigNumber } from '@0xproject/utils';
import * as _ from 'lodash';

import { constants, orderFactory } from '../../src';

const BASE_TEST_ORDER: Order = orderFactory.createOrder(
    constants.NULL_ADDRESS,
    constants.NULL_ADDRESS,
    constants.NULL_ADDRESS,
    constants.ZERO_AMOUNT,
    constants.ZERO_AMOUNT,
    constants.ZERO_AMOUNT,
    constants.NULL_BYTES,
    constants.ZERO_AMOUNT,
    constants.NULL_BYTES,
    constants.NULL_ADDRESS,
    constants.NULL_ADDRESS,
);
const BASE_TEST_SIGNED_ORDER: SignedOrder = {
    ...BASE_TEST_ORDER,
    signature: constants.NULL_BYTES,
};
const BASE_TEST_ORDER_RELEVANT_STATE: OrderRelevantState = {
    makerBalance: constants.ZERO_AMOUNT,
    makerProxyAllowance: constants.ZERO_AMOUNT,
    makerFeeBalance: constants.ZERO_AMOUNT,
    makerFeeProxyAllowance: constants.ZERO_AMOUNT,
    filledTakerAssetAmount: constants.ZERO_AMOUNT,
    remainingFillableMakerAssetAmount: constants.ZERO_AMOUNT,
    remainingFillableTakerAssetAmount: constants.ZERO_AMOUNT,
};

export const testOrderFactory = {
    generateTestSignedOrder(partialOrder: Partial<SignedOrder>): SignedOrder {
        return transformObject(BASE_TEST_SIGNED_ORDER, partialOrder);
    },
    generateTestSignedOrders(partialOrder: Partial<SignedOrder>, numOrders: number): SignedOrder[] {
        const baseTestOrders = generateArrayOfInput(BASE_TEST_SIGNED_ORDER, numOrders);
        return transformObjects(baseTestOrders, partialOrder);
    },
    generateTestOrderRelevantState(partialOrderRelevantState: Partial<OrderRelevantState>): OrderRelevantState {
        return transformObject(BASE_TEST_ORDER_RELEVANT_STATE, partialOrderRelevantState);
    },
    generateTestOrderRelevantStates(
        partialOrderRelevantState: Partial<OrderRelevantState>,
        numOrderStates: number,
    ): OrderRelevantState[] {
        const baseTestOrderStates = generateArrayOfInput(BASE_TEST_ORDER_RELEVANT_STATE, numOrderStates);
        return transformObjects(baseTestOrderStates, partialOrderRelevantState);
    },
};

function generateArrayOfInput<T>(input: T, rangeLength: number): T[] {
    return _.map(_.range(rangeLength), () => input);
}
function transformObject<T>(input: T, transformation: Partial<T>): T {
    const copy = _.cloneDeep(input);
    return _.assign(copy, transformation);
}
function transformObjects<T>(inputs: T[], transformation: Partial<T>): T[] {
    return _.map(inputs, input => transformObject(input, transformation));
}