igniteui-angular-sovn/src/lib/data-operations/test-util/data-generator.ts

Ignite UI for Angular is a dependency-free Angular toolkit for building modern web apps

import { GridColumnDataType } from '../data-util';

/**
 * @hidden
 */
const COUNT_ROWS = 5;

/**
 * @hidden
 */
const COUNT_COLS = 4;

/**
 * @hidden
 */
export interface IDataColumn {
    fieldName: string;
    type: GridColumnDataType;
}

/**
 * @hidden
 */
export class DataGenerator {
    public columns: IDataColumn[] = [];
    public data: any[] = [];
    constructor(countRows = COUNT_ROWS, countCols = COUNT_COLS) {
        this.columns = this.generateColumns(countCols);
        this.data = this.generateData(countRows);
    }
    public generateArray(startValue, endValue) {
        const len = Math.abs(startValue - endValue);
        const decrement = startValue > endValue;
        return Array.from({ length: len + 1 }, (e, i) => decrement ? startValue - i : startValue + i);
    }
    public getValuesForColumn(data, fieldName) {
        return data.map((x) => x[fieldName]);
    }
    public getGroupRecords(data) {
        return data.map((x) => x['groupParent']);
    }
    public isSuperset(haystack, arr) {
        return arr.every((val) => haystack.indexOf(val) >= 0);
    }
    private generateColumns(countCols): IDataColumn[] {
        let i: number;
        const defaultColumns: IDataColumn[] = [
            {
                fieldName: 'number',
                type: GridColumnDataType.Number
            },
            {
                fieldName: 'string',
                type: GridColumnDataType.String
            },
            {
                fieldName: 'date',
                type: GridColumnDataType.Date
            },
            {
                fieldName: 'boolean',
                type: GridColumnDataType.Boolean
            }
        ];
        if (countCols <= 0) {
            return defaultColumns;
        }
        if (countCols <= defaultColumns.length) {
            return defaultColumns.slice(0, countCols);
        }
        const len = countCols - defaultColumns.length;
        const res = defaultColumns;
        for (i = 0; i < len; i++) {
            res.push({
                fieldName: `col${i}`,
                type: GridColumnDataType.String
            });
        }
        return res;
    }
    private generateData(countRows: number) {
        let i;
        let j;
        let rec;
        let val;
        let col;
        const data = [];
        for (i = 0; i < countRows; i++) {
            rec = {};
            for (j = 0; j < this.columns.length; j++) {
                col = this.columns[j];
                switch (col.type) {
                    case GridColumnDataType.Number:
                        val = i;
                        break;
                    case GridColumnDataType.Date:
                        val = new Date(Date.now() + i * 24 * 60 * 60 * 1000);
                        break;
                    case GridColumnDataType.Boolean:
                        val = !!(i % 2);
                        break;
                    default:
                        val = `row${i}, col${j}`;
                        break;
                }
                rec[col.fieldName] = val;
            }
            data.push(rec);
        }
        return data;
    }
}