// Copyright (C) The Arvados Authors. All rights reserved.
//
// SPDX-License-Identifier: AGPL-3.0

import * as React from "react";
import { mount, configure } from "enzyme";
import { pipe } from 'lodash/fp';
import { TableHead, TableCell, Typography, TableBody, Button, TableSortLabel } from "@material-ui/core";
import * as Adapter from "enzyme-adapter-react-16";
import { DataTable, DataColumns } from "./data-table";
import { SortDirection, createDataColumn } from "./data-column";
import { DataTableFiltersPopover } from 'components/data-table-filters/data-table-filters-popover';
import { createTree, setNode, initTreeNode } from 'models/tree';
import { DataTableFilterItem } from "components/data-table-filters/data-table-filters-tree";

configure({ adapter: new Adapter() });

describe("<DataTable />", () => {
    it("shows only selected columns", () => {
        const columns: DataColumns<string> = [
            createDataColumn({
                name: "Column 1",
                render: () => <span />,
                selected: true,
                configurable: true
            }),
            createDataColumn({
                name: "Column 2",
                render: () => <span />,
                selected: true,
                configurable: true
            }),
            createDataColumn({
                name: "Column 3",
                render: () => <span />,
                selected: false,
                configurable: true
            }),
        ];
        const dataTable = mount(<DataTable
            columns={columns}
            items={[{ key: "1", name: "item 1" }]}
            onFiltersChange={jest.fn()}
            onRowClick={jest.fn()}
            onRowDoubleClick={jest.fn()}
            onContextMenu={jest.fn()}
            onSortToggle={jest.fn()} />);
        expect(dataTable.find(TableHead).find(TableCell)).toHaveLength(2);
    });

    it("renders column name", () => {
        const columns: DataColumns<string> = [
            createDataColumn({
                name: "Column 1",
                render: () => <span />,
                selected: true,
                configurable: true
            }),
        ];
        const dataTable = mount(<DataTable
            columns={columns}
            items={["item 1"]}
            onFiltersChange={jest.fn()}
            onRowClick={jest.fn()}
            onRowDoubleClick={jest.fn()}
            onContextMenu={jest.fn()}
            onSortToggle={jest.fn()} />);
        expect(dataTable.find(TableHead).find(TableCell).text()).toBe("Column 1");
    });

    it("uses renderHeader instead of name prop", () => {
        const columns: DataColumns<string> = [
            createDataColumn({
                name: "Column 1",
                renderHeader: () => <span>Column Header</span>,
                render: () => <span />,
                selected: true,
                configurable: true
            }),
        ];
        const dataTable = mount(<DataTable
            columns={columns}
            items={[]}
            onFiltersChange={jest.fn()}
            onRowClick={jest.fn()}
            onRowDoubleClick={jest.fn()}
            onContextMenu={jest.fn()}
            onSortToggle={jest.fn()} />);
        expect(dataTable.find(TableHead).find(TableCell).text()).toBe("Column Header");
    });

    it("passes column key prop to corresponding cells", () => {
        const columns: DataColumns<string> = [
            createDataColumn({
                name: "Column 1",
                key: "column-1-key",
                render: () => <span />,
                selected: true,
                configurable: true
            })
        ];
        const dataTable = mount(<DataTable
            columns={columns}
            items={["item 1"]}
            onFiltersChange={jest.fn()}
            onRowClick={jest.fn()}
            onRowDoubleClick={jest.fn()}
            onContextMenu={jest.fn()}
            onSortToggle={jest.fn()} />);
        expect(dataTable.find(TableHead).find(TableCell).key()).toBe("column-1-key");
        expect(dataTable.find(TableBody).find(TableCell).key()).toBe("column-1-key");
    });

    it("renders items", () => {
        const columns: DataColumns<string> = [
            createDataColumn({
                name: "Column 1",
                render: (item) => <Typography>{item}</Typography>,
                selected: true,
                configurable: true
            }),
            createDataColumn({
                name: "Column 2",
                render: (item) => <Button>{item}</Button>,
                selected: true,
                configurable: true
            })
        ];
        const dataTable = mount(<DataTable
            columns={columns}
            items={["item 1"]}
            onFiltersChange={jest.fn()}
            onRowClick={jest.fn()}
            onRowDoubleClick={jest.fn()}
            onContextMenu={jest.fn()}
            onSortToggle={jest.fn()} />);
        expect(dataTable.find(TableBody).find(Typography).text()).toBe("item 1");
        expect(dataTable.find(TableBody).find(Button).text()).toBe("item 1");
    });

    it("passes sorting props to <TableSortLabel />", () => {
        const columns: DataColumns<string> = [
            createDataColumn({
                name: "Column 1",
                sortDirection: SortDirection.ASC,
                selected: true,
                configurable: true,
                render: (item) => <Typography>{item}</Typography>
            })];
        const onSortToggle = jest.fn();
        const dataTable = mount(<DataTable
            columns={columns}
            items={["item 1"]}
            onFiltersChange={jest.fn()}
            onRowClick={jest.fn()}
            onRowDoubleClick={jest.fn()}
            onContextMenu={jest.fn()}
            onSortToggle={onSortToggle} />);
        expect(dataTable.find(TableSortLabel).prop("active")).toBeTruthy();
        dataTable.find(TableSortLabel).at(0).simulate("click");
        expect(onSortToggle).toHaveBeenCalledWith(columns[0]);
    });

    it("does not display <DataTableFiltersPopover /> if there is no filters provided", () => {
        const columns: DataColumns<string> = [{
            name: "Column 1",
            sortDirection: SortDirection.ASC,
            selected: true,
            configurable: true,
            filters: [],
            render: (item) => <Typography>{item}</Typography>
        }];
        const onFiltersChange = jest.fn();
        const dataTable = mount(<DataTable
            columns={columns}
            items={[]}
            onFiltersChange={onFiltersChange}
            onRowClick={jest.fn()}
            onRowDoubleClick={jest.fn()}
            onSortToggle={jest.fn()}
            onContextMenu={jest.fn()} />);
        expect(dataTable.find(DataTableFiltersPopover)).toHaveLength(0);
    });

    it("passes filter props to <DataTableFiltersPopover />", () => {
        const filters = pipe(
            () => createTree<DataTableFilterItem>(),
            setNode(initTreeNode({ id: 'filter', value: { name: 'filter' } }))
        );
        const columns: DataColumns<string> = [{
            name: "Column 1",
            sortDirection: SortDirection.ASC,
            selected: true,
            configurable: true,
            filters: filters(),
            render: (item) => <Typography>{item}</Typography>
        }];
        const onFiltersChange = jest.fn();
        const dataTable = mount(<DataTable
            columns={columns}
            items={[]}
            onFiltersChange={onFiltersChange}
            onRowClick={jest.fn()}
            onRowDoubleClick={jest.fn()}
            onSortToggle={jest.fn()}
            onContextMenu={jest.fn()} />);
        expect(dataTable.find(DataTableFiltersPopover).prop("filters")).toBe(columns[0].filters);
        dataTable.find(DataTableFiltersPopover).prop("onChange")([]);
        expect(onFiltersChange).toHaveBeenCalledWith([], columns[0]);
    });
});