Add tests

mui · Feb 26, 2020 · f430c97 · f430c97
1 parent e392b78
commit f430c97
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/packages/material-ui-lab/src/TreeView/TreeView.test.js b/packages/material-ui-lab/src/TreeView/TreeView.test.js
@@ -174,6 +174,27 @@ describe('<TreeView />', () => {
     expect(getByTestId('two')).to.have.focus;
   });
 
+  it('should support conditional rendered tree items', () => {
+    function TestComponent() {
+      const [hide, setState] = React.useState(false);
+
+      return (
+        <React.Fragment>
+          <button type="button" onClick={() => setState(true)}>
+            Hide
+          </button>
+          <TreeView>{!hide && <TreeItem nodeId="test" label="test" />}</TreeView>
+        </React.Fragment>
+      );
+    }
+
+    const { getByText, queryByText } = render(<TestComponent />);
+
+    expect(getByText('test')).to.not.be.null;
+    fireEvent.click(getByText('Hide'));
+    expect(queryByText('test')).to.be.null;
+  });
+
   describe('onNodeToggle', () => {
     it('should be called when a parent node is clicked', () => {
       const handleNodeToggle = spy();