@gooddata/react-components/src/components/visualizations/chart/test/helper.ts

GoodData.UI - A powerful JavaScript library for building analytical applications

// (C) 2007-2019 GoodData Corporation
import {
    findAttributeInDimension,
    findMeasureGroupInDimensions,
} from "../../../../helpers/executionResultHelper";
import { getChartOptions } from "../chartOptionsBuilder";

import {
    VIEW_BY_DIMENSION_INDEX,
    STACK_BY_DIMENSION_INDEX,
    PARENT_ATTRIBUTE_INDEX,
    PRIMARY_ATTRIBUTE_INDEX,
} from "../constants";

import { IHeaderPredicate } from "../../../../interfaces/HeaderPredicate";
import { barChartWithStackByAndViewByAttributes } from "../../../../../stories/test_data/fixtures";
import { IChartOptions } from "../../../../interfaces/Config";

export function generateChartOptions(
    dataSet: any = barChartWithStackByAndViewByAttributes,
    config: any = {
        type: "column",
        stacking: false,
    },
    drillableItems: IHeaderPredicate[] = [],
): IChartOptions {
    const {
        executionRequest: { afm, resultSpec },
        executionResponse,
        executionResult: { data, headerItems },
    } = dataSet;
    return getChartOptions(afm, resultSpec, executionResponse, data, headerItems, config, drillableItems);
}

export function getMVS(dataSet: any) {
    const {
        executionResponse: { dimensions },
        executionResult: { headerItems },
    } = dataSet;
    const measureGroup = findMeasureGroupInDimensions(dimensions);
    const viewByAttribute = findAttributeInDimension(
        dimensions[VIEW_BY_DIMENSION_INDEX],
        headerItems[VIEW_BY_DIMENSION_INDEX],
    );
    const stackByAttribute = findAttributeInDimension(
        dimensions[STACK_BY_DIMENSION_INDEX],
        headerItems[STACK_BY_DIMENSION_INDEX],
    );
    return {
        measureGroup,
        viewByAttribute,
        stackByAttribute,
    };
}

export function getMVSForViewByTwoAttributes(dataSet: any) {
    const mvs = getMVS(dataSet);
    const {
        executionResponse: { dimensions },
        executionResult: { headerItems },
    } = dataSet;
    const viewByParentAttribute = findAttributeInDimension(
        dimensions[VIEW_BY_DIMENSION_INDEX],
        headerItems[VIEW_BY_DIMENSION_INDEX],
        PARENT_ATTRIBUTE_INDEX,
    );
    const viewByAttribute = findAttributeInDimension(
        dimensions[VIEW_BY_DIMENSION_INDEX],
        headerItems[VIEW_BY_DIMENSION_INDEX],
        PRIMARY_ATTRIBUTE_INDEX,
    );
    return {
        ...mvs,
        viewByAttribute,
        viewByParentAttribute,
    };
}