[tree view] Cleanup onKeyDown

mui · Dec 21, 2023 · 0f66d83 · 0f66d83
1 parent d7149ef
commit 0f66d83
Show file tree

Hide file tree

Showing 2 changed files with 196 additions and 144 deletions.
diff --git a/packages/x-tree-view/src/TreeItem/TreeItem.test.tsx b/packages/x-tree-view/src/TreeItem/TreeItem.test.tsx
@@ -1104,6 +1104,36 @@ describe('<TreeItem />', () => {
 
           expect(getByTestId('one')).not.to.have.attribute('aria-selected');
         });
+
+        it('should select a node when Enter is pressed and the node is not selected', () => {
+          const { getByRole, getByTestId } = render(
+            <TreeView>
+              <TreeItem nodeId="one" label="one" data-testid="one" />
+            </TreeView>,
+          );
+
+          act(() => {
+            getByRole('tree').focus();
+          });
+          fireEvent.keyDown(getByRole('tree'), { key: 'Enter' });
+
+          expect(getByTestId('one')).to.have.attribute('aria-selected');
+        });
+
+        it('should not un-select a node when Enter is pressed and the node is selected', () => {
+          const { getByRole, getByTestId } = render(
+            <TreeView defaultSelected="one">
+              <TreeItem nodeId="one" label="one" data-testid="one" />
+            </TreeView>,
+          );
+
+          act(() => {
+            getByRole('tree').focus();
+          });
+          fireEvent.keyDown(getByRole('tree'), { key: 'Enter' });
+
+          expect(getByTestId('one')).to.have.attribute('aria-selected');
+        });
       });
 
       describe('mouse', () => {