X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/5c513f2c78337482da4cc6055706fd022c26c007..9ab26fe1f0838a97818416bd6b833fc74968ed65:/src/components/data-table/data-table.test.tsx

diff --git a/src/components/data-table/data-table.test.tsx b/src/components/data-table/data-table.test.tsx
index b9d112520a..6dbccb5e7d 100644
--- a/src/components/data-table/data-table.test.tsx
+++ b/src/components/data-table/data-table.test.tsx
@@ -6,14 +6,19 @@ import * as React from "react";
 import { mount, configure } from "enzyme";
 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 DataTable, { DataColumns, DataItem } from "./data-table";
 import DataTableFilters from "../data-table-filters/data-table-filters";
+import { SortDirection } from "./data-column";
 
 configure({ adapter: new Adapter() });
 
+export interface MockItem extends DataItem {
+    name: string;
+}
+
 describe("<DataTable />", () => {
     it("shows only selected columns", () => {
-        const columns: DataColumns<string> = [
+        const columns: DataColumns<MockItem> = [
             {
                 name: "Column 1",
                 render: () => <span />,
@@ -32,7 +37,7 @@ describe("<DataTable />", () => {
         ];
         const dataTable = mount(<DataTable
             columns={columns}
-            items={["item 1"]}
+            items={[{ key: "1", name: "item 1" }] as MockItem[]}
             onFiltersChange={jest.fn()}
             onRowClick={jest.fn()}
             onRowContextMenu={jest.fn()}
@@ -41,7 +46,7 @@ describe("<DataTable />", () => {
     });
 
     it("renders column name", () => {
-        const columns: DataColumns<string> = [
+        const columns: DataColumns<MockItem> = [
             {
                 name: "Column 1",
                 render: () => <span />,
@@ -50,7 +55,7 @@ describe("<DataTable />", () => {
         ];
         const dataTable = mount(<DataTable
             columns={columns}
-            items={["item 1"]}
+            items={[{ key: "1", name: "item 1" }] as MockItem[]}
             onFiltersChange={jest.fn()}
             onRowClick={jest.fn()}
             onRowContextMenu={jest.fn()}
@@ -59,7 +64,7 @@ describe("<DataTable />", () => {
     });
 
     it("uses renderHeader instead of name prop", () => {
-        const columns: DataColumns<string> = [
+        const columns: DataColumns<MockItem> = [
             {
                 name: "Column 1",
                 renderHeader: () => <span>Column Header</span>,
@@ -69,7 +74,7 @@ describe("<DataTable />", () => {
         ];
         const dataTable = mount(<DataTable
             columns={columns}
-            items={["item 1"]}
+            items={[{ key: "1", name: "item 1" }] as MockItem[]}
             onFiltersChange={jest.fn()}
             onRowClick={jest.fn()}
             onRowContextMenu={jest.fn()}
@@ -78,7 +83,7 @@ describe("<DataTable />", () => {
     });
 
     it("passes column key prop to corresponding cells", () => {
-        const columns: DataColumns<string> = [
+        const columns: DataColumns<MockItem> = [
             {
                 name: "Column 1",
                 key: "column-1-key",
@@ -88,7 +93,7 @@ describe("<DataTable />", () => {
         ];
         const dataTable = mount(<DataTable
             columns={columns}
-            items={["item 1"]}
+            items={[{ key: "1", name: "item 1" }] as MockItem[]}
             onFiltersChange={jest.fn()}
             onRowClick={jest.fn()}
             onRowContextMenu={jest.fn()}
@@ -98,21 +103,21 @@ describe("<DataTable />", () => {
     });
 
     it("renders items", () => {
-        const columns: DataColumns<string> = [
+        const columns: DataColumns<MockItem> = [
             {
                 name: "Column 1",
-                render: (item) => <Typography>{item}</Typography>,
+                render: (item) => <Typography>{item.name}</Typography>,
                 selected: true
             },
             {
                 name: "Column 2",
-                render: (item) => <Button>{item}</Button>,
+                render: (item) => <Button>{item.name}</Button>,
                 selected: true
             }
         ];
-        const dataTable = mount(<DataTable 
-            columns={columns} 
-            items={["item 1"]}
+        const dataTable = mount(<DataTable
+            columns={columns}
+            items={[{ key: "1", name: "item 1" }] as MockItem[]}
             onFiltersChange={jest.fn()}
             onRowClick={jest.fn()}
             onRowContextMenu={jest.fn()}
@@ -122,41 +127,41 @@ describe("<DataTable />", () => {
     });
 
     it("passes sorting props to <TableSortLabel />", () => {
-        const columns: DataColumns<string> = [{
+        const columns: DataColumns<MockItem> = [{
             name: "Column 1",
-            sortDirection: "asc",
+            sortDirection: SortDirection.Asc,
             selected: true,
-            render: (item) => <Typography>{item}</Typography>
+            render: (item) => <Typography>{item.name}</Typography>
         }];
         const onSortToggle = jest.fn();
-        const dataTable = mount(<DataTable 
-            columns={columns} 
-            items={["item 1"]} 
+        const dataTable = mount(<DataTable
+            columns={columns}
+            items={[{ key: "1", name: "item 1" }] as MockItem[]}
             onFiltersChange={jest.fn()}
             onRowClick={jest.fn()}
             onRowContextMenu={jest.fn()}
-            onSortToggle={onSortToggle}/>);
+            onSortToggle={onSortToggle} />);
         expect(dataTable.find(TableSortLabel).prop("active")).toBeTruthy();
         dataTable.find(TableSortLabel).at(0).simulate("click");
         expect(onSortToggle).toHaveBeenCalledWith(columns[0]);
     });
 
     it("passes filter props to <DataTableFilter />", () => {
-        const columns: DataColumns<string> = [{
+        const columns: DataColumns<MockItem> = [{
             name: "Column 1",
-            sortDirection: "asc",
+            sortDirection: SortDirection.Asc,
             selected: true,
-            filters: [{name: "Filter 1", selected: true}],
-            render: (item) => <Typography>{item}</Typography>
+            filters: [{ name: "Filter 1", selected: true }],
+            render: (item) => <Typography>{item.name}</Typography>
         }];
         const onFiltersChange = jest.fn();
-        const dataTable = mount(<DataTable 
-            columns={columns} 
-            items={["item 1"]} 
+        const dataTable = mount(<DataTable
+            columns={columns}
+            items={[{ key: "1", name: "item 1" }] as MockItem[]}
             onFiltersChange={onFiltersChange}
             onRowClick={jest.fn()}
             onRowContextMenu={jest.fn()}
-            onSortToggle={jest.fn()}/>);
+            onSortToggle={jest.fn()} />);
         expect(dataTable.find(DataTableFilters).prop("filters")).toBe(columns[0].filters);
         dataTable.find(DataTableFilters).prop("onChange")([]);
         expect(onFiltersChange).toHaveBeenCalledWith([], columns[0]);
