import CosmicColumnFormatter from './CosmicColumnFormatter';
import { ICosmicData } from 'shared/model/Cosmic';
import { keywordToCosmic } from 'shared/lib/AnnotationUtils';
import { initMutation } from 'test/MutationMockUtils';
import React from 'react';
import { assert } from 'chai';
import { mount, ReactWrapper } from 'enzyme';
import { DefaultTooltip } from 'cbioportal-frontend-commons';

describe('CosmicColumnFormatter', () => {
    const cosmicMutations = [
        {
            cosmicMutationId: '',
            count: 2,
            keyword: 'TP53 R273 missense',
            proteinChange: 'R273H',
        },
        {
            cosmicMutationId: '',
            count: 3,
            keyword: 'TP53 R273 missense',
            proteinChange: 'R273C',
        },
        {
            cosmicMutationId: '',
            count: 5,
            keyword: 'TP53 R273 missense',
            proteinChange: 'R273L',
        },
        {
            cosmicMutationId: '',
            count: 7,
            keyword: 'PIK3CA R38 missense',
            proteinChange: 'R38H',
        },
        {
            cosmicMutationId: '',
            count: 11,
            keyword: 'PIK3CA R38 missense',
            proteinChange: 'R38C',
        },
        {
            cosmicMutationId: '',
            count: 13,
            keyword: 'PIK3CA R38 missense',
            proteinChange: 'R38S',
        },
    ];

    const mutation273 = initMutation({
        gene: {
            hugoGeneSymbol: 'TP53',
        },
        proteinChange: 'R273S',
        keyword: 'TP53 R273 missense',
    });

    const mutation38 = initMutation({
        gene: {
            hugoGeneSymbol: 'PIK3CA',
        },
        proteinChange: 'R38S',
        keyword: 'PIK3CA R38 missense',
    });

    const mutation666 = initMutation({
        gene: {
            hugoGeneSymbol: 'DIABLO',
        },
        proteinChange: 'V666E',
        keyword: 'DIABLO',
    });

    const tableData = [[mutation273], [mutation38], [mutation666]];

    let component273: ReactWrapper<any, any>;
    let component38: ReactWrapper<any, any>;
    let component666: ReactWrapper<any, any>;
    let cosmicData: ICosmicData;

    beforeAll(() => {
        cosmicData = keywordToCosmic(cosmicMutations);

        let data = [mutation273];

        // mount a single cell component (Td)
        component273 = mount(
            CosmicColumnFormatter.renderFunction(data, cosmicData)
        );

        data = [mutation38];

        // mount a single cell component (Td)
        component38 = mount(
            CosmicColumnFormatter.renderFunction(data, cosmicData)
        );

        data = [mutation666];

        // mount a single cell component (Td)
        component666 = mount(
            CosmicColumnFormatter.renderFunction(data, cosmicData)
        );
    });

    it('generates component tooltip', () => {
        assert.isTrue(
            component273.find(DefaultTooltip).exists(),
            'Tooltip should exists for TP53 R273 missense mutation'
        );
        assert.isTrue(
            component38.find(DefaultTooltip).exists(),
            'Tooltip should exists for PIK3CA R38 missense mutation'
        );
        assert.isFalse(
            component666.find(DefaultTooltip).exists(),
            'Tooltip should not exist for DIABLO mutation'
        );
    });

    it('renders display value', () => {
        assert.isTrue(
            component273
                .find(`div`)
                .text()
                .indexOf((2 + 3 + 5).toString()) > -1,
            'Cosmic count total for TP53 R273 missense mutation is correct'
        );
        assert.isTrue(
            component38
                .find(`div`)
                .text()
                .indexOf((7 + 11 + 13).toString()) > -1,
            'Cosmic count total for PIK3CA R38 missense mutation is correct'
        );
    });

    it('sets component cell value property', () => {
        assert.equal(
            CosmicColumnFormatter.getSortValue([mutation273], cosmicData),
            2 + 3 + 5,
            'Sort value for TP53 R273 missense mutation is correct'
        );
        assert.equal(
            CosmicColumnFormatter.getSortValue([mutation38], cosmicData),
            7 + 11 + 13,
            'Sort value value property for PIK3CA R38 missense mutation is correct'
        );
        assert.equal(
            CosmicColumnFormatter.getSortValue([mutation666], cosmicData),
            null,
            'Sort value value property for DIABLO mutation should be null, no cosmic data'
        );
    });
});