X-Git-Url: https://git.arvados.org/arvados-workbench2.git/blobdiff_plain/750a9abc14ba1e17784a386399237ebf35fb279b..53fd16f76b1c949c36c4d119e6ca6346af7c643d:/src/components/tree/tree.test.tsx

diff --git a/src/components/tree/tree.test.tsx b/src/components/tree/tree.test.tsx
index bb0499f8..58484c37 100644
--- a/src/components/tree/tree.test.tsx
+++ b/src/components/tree/tree.test.tsx
@@ -5,51 +5,47 @@ import * as React from 'react';
 import { mount } from 'enzyme';
 import * as Enzyme from 'enzyme';
 import * as Adapter from 'enzyme-adapter-react-16';
-import { Collapse } from '@material-ui/core';
-import CircularProgress from '@material-ui/core/CircularProgress';
 import ListItem from "@material-ui/core/ListItem/ListItem";
 
-import Tree, {TreeItem} from './tree';
-import { Project } from '../../models/project';
+import { Tree, TreeItem } from './tree';
+import { ProjectResource } from '../../models/project';
+import { mockProjectResource } from '../../models/test-utils';
+
 Enzyme.configure({ adapter: new Adapter() });
 
 describe("Tree component", () => {
 
-	it("should render ListItem", () => {
-		const project: TreeItem<Project> = {
-            data: {
-                name: "sample name",
-                createdAt: "2018-06-12",
-                modifiedAt: "2018-06-13",
-                uuid: "uuid",
-                ownerUuid: "ownerUuid",
-                href: "href",
-            },
+    it("should render ListItem", () => {
+        const project: TreeItem<ProjectResource> = {
+            data: mockProjectResource(),
             id: "3",
             open: true,
-			active: true,
-			status: 1,
+            active: true,
+            status: 1,
         };
-		const wrapper = mount(<Tree render={project => <div/>} toggleItem={() => { }} items={[project]}/>)
-		expect(wrapper.find(ListItem)).toHaveLength(1);
-	});
-    
+        const wrapper = mount(<Tree
+            render={project => <div />}
+            toggleItemOpen={jest.fn()}
+            toggleItemActive={jest.fn()}
+            onContextMenu={jest.fn()}
+            items={[project]} />);
+        expect(wrapper.find(ListItem)).toHaveLength(1);
+    });
+
     it("should render arrow", () => {
-		const project: TreeItem<Project> = {
-            data: {
-                name: "sample name",
-                createdAt: "2018-06-12",
-                modifiedAt: "2018-06-13",
-                uuid: "uuid",
-                ownerUuid: "ownerUuid",
-                href: "href",
-            },
+        const project: TreeItem<ProjectResource> = {
+            data: mockProjectResource(),
             id: "3",
             open: true,
-			active: true,
-			status: 1,
+            active: true,
+            status: 1,
         };
-		const wrapper = mount(<Tree render={project => <div/>} toggleItem={() => { }} items={[project]}/>)
-		expect(wrapper.find('i')).toHaveLength(1);
-	});
+        const wrapper = mount(<Tree
+            render={project => <div />}
+            toggleItemOpen={jest.fn()}
+            toggleItemActive={jest.fn()}
+            onContextMenu={jest.fn()}
+            items={[project]} />);
+        expect(wrapper.find('i')).toHaveLength(1);
+    });
 });
