@onewelcome/react-lib-components 0.1.5-alpha → 0.1.8-alpha

package/src/ContextMenu/ContextMenu.test.tsx

@@ -1,6 +1,6 @@
 import React, { useEffect, useRef } from 'react';
 import { ContextMenu, Props } from './ContextMenu';
-import { render, getByRole } from '@testing-library/react';
+import { render } from '@testing-library/react';
 import { Button } from '../Button/Button';
 import { ContextMenuItem } from './ContextMenuItem';
 import userEvent from '@testing-library/user-event';
@@ -15,8 +15,12 @@ const defaultParams: Props = {
     <ContextMenuItem onClick={onClick} data-testid="contextmenuitem" key="1">
       Example item 1
     </ContextMenuItem>,
-    <ContextMenuItem key="2">Example item 2</ContextMenuItem>,
-    <ContextMenuItem key="3">Example item 3</ContextMenuItem>,
+    <ContextMenuItem onClick={onClick} data-testid="contextmenuitem2" key="2">
+      Example item 2
+    </ContextMenuItem>,
+    <ContextMenuItem onClick={onClick} data-testid="contextmenuitem3" key="3">
+      Example item 3
+    </ContextMenuItem>,
   ],
   show: false,
   onShow: onShow,
@@ -61,12 +65,10 @@ describe('ContextMenu should render', () => {
       show: true,
     }));
 
-    const contextmenuitem = getByTestId('contextmenuitem');
-    const button = getByRole(contextmenuitem, 'button');
+    const button = getByTestId('contextmenuitem');
 
     userEvent.click(button);
 
-    expect(contextmenuitem).toBeTruthy();
     expect(onClick).toHaveBeenCalled();
   });
 
@@ -117,3 +119,116 @@ describe('ref should work', () => {
     render(<ExampleComponent propagateRef={refCheck} />);
   });
 });
+
+describe('accessibility controls', () => {
+  it('opening works correctly with arrow key down, then navigation should work with arrow keys', () => {
+    const { getByTestId, trigger } = createContextMenu();
+    const firstContextMenuItem = getByTestId('contextmenuitem');
+    const secondContextMenuItem = getByTestId('contextmenuitem2');
+    const thirdContextMenuItem = getByTestId('contextmenuitem3');
+
+    userEvent.keyboard('{arrowdown}');
+    userEvent.keyboard('{arrowdown}');
+    expect(trigger).toHaveAttribute('aria-expanded', 'true');
+    expect(firstContextMenuItem).toHaveFocus();
+
+    userEvent.keyboard('{arrowdown}');
+    expect(secondContextMenuItem).toHaveFocus();
+
+    userEvent.keyboard('{arrowdown}');
+    expect(thirdContextMenuItem).toHaveFocus();
+
+    userEvent.keyboard('{arrowdown}');
+    expect(firstContextMenuItem).toHaveFocus();
+
+    userEvent.keyboard('{arrowup}');
+    expect(thirdContextMenuItem).toHaveFocus();
+
+    userEvent.keyboard('{arrowup}');
+    expect(secondContextMenuItem).toHaveFocus();
+  });
+
+  it('opens correctly with enter key, closing works with escape key.', async () => {
+    const { trigger } = createContextMenu();
+
+    userEvent.keyboard('{enter}');
+
+    expect(trigger).toHaveAttribute('aria-expanded', 'true');
+
+    userEvent.keyboard('{escape}');
+
+    expect(trigger).toHaveAttribute('aria-expanded', 'false');
+  });
+
+  it('opens correctly with space, home and end buttons work', () => {
+    const { trigger, getByTestId } = createContextMenu();
+    const firstContextMenuItem = getByTestId('contextmenuitem');
+    const thirdContextMenuItem = getByTestId('contextmenuitem3');
+
+    userEvent.keyboard('{space}');
+
+    expect(trigger).toHaveAttribute('aria-expanded', 'true');
+
+    userEvent.keyboard('{end}');
+
+    expect(thirdContextMenuItem).toHaveFocus();
+
+    userEvent.keyboard('{home}');
+
+    expect(firstContextMenuItem).toHaveFocus();
+  });
+
+  it('opens correctly with space, navigate with arrow keys, select with enter', () => {
+    onClick.mockImplementation((e) => {
+      expect(e.target.getAttribute('data-testid')).toBe('contextmenuitem3');
+    });
+
+    const { trigger, getByTestId } = createContextMenu((defaultParams) => ({
+      ...defaultParams,
+    }));
+    const thirdContextMenuItem = getByTestId('contextmenuitem3');
+
+    userEvent.keyboard('{space}');
+
+    expect(trigger).toHaveAttribute('aria-expanded', 'true');
+
+    userEvent.keyboard('{arrowdown}');
+    userEvent.keyboard('{arrowdown}');
+    userEvent.keyboard('{arrowdown}');
+
+    expect(thirdContextMenuItem).toHaveFocus();
+
+    userEvent.keyboard('{enter}');
+
+    expect(onClick).toHaveBeenCalled();
+  });
+
+  it('opens correctly with enter, navigate with arrow keys, select with space', () => {
+    onClick.mockImplementation((e) => {
+      expect(e.target.getAttribute('data-testid')).toBe('contextmenuitem3');
+    });
+
+    const { trigger, getByTestId } = createContextMenu((defaultParams) => ({
+      ...defaultParams,
+    }));
+    const thirdContextMenuItem = getByTestId('contextmenuitem3');
+
+    userEvent.keyboard('{enter}');
+
+    expect(trigger).toHaveAttribute('aria-expanded', 'true');
+
+    userEvent.keyboard('{arrowdown}');
+    userEvent.keyboard('{arrowdown}');
+    userEvent.keyboard('{arrowdown}');
+
+    expect(thirdContextMenuItem).toHaveFocus();
+
+    userEvent.keyboard('{space}');
+
+    expect(onClick).toHaveBeenCalled();
+
+    userEvent.keyboard('{space}');
+
+    expect(thirdContextMenuItem).toHaveFocus();
+  });
+});

package/src/ContextMenu/ContextMenu.tsx

@@ -1,16 +1,22 @@
-import React, { ComponentPropsWithRef, ReactElement, useEffect, useRef, useState } from 'react';
+import React, {
+  ComponentPropsWithRef,
+  ReactElement,
+  ReactNode,
+  useEffect,
+  useRef,
+  useState,
+} from 'react';
 import { Props as ButtonProps } from '../Button/Button';
 import { Props as IconButtonProps } from '../Button/IconButton';
 import { Popover } from '../Popover/Popover';
 import { Placement, Offset } from '../hooks/usePosition';
-import { Props as ContextMenuItemProps } from './ContextMenuItem';
 import classes from './ContextMenu.module.scss';
 import { useBodyClick } from '../hooks/useBodyClick';
 import { createPortal } from 'react-dom';
 
 export interface Props extends ComponentPropsWithRef<'div'> {
   trigger: ReactElement<ButtonProps> | ReactElement<IconButtonProps>;
-  children: ReactElement<ContextMenuItemProps> | ReactElement<ContextMenuItemProps>[];
+  children: ReactNode;
   placement?: Placement;
   transformOrigin?: Placement;
   offset?: Offset;
@@ -40,11 +46,78 @@ export const ContextMenu = React.forwardRef<HTMLDivElement, Props>(
   ) => {
     const anchorEl = useRef<HTMLButtonElement>(null);
     const [showContextMenu, setShowContextMenu] = useState(show);
+    const [selectedContextMenuItem, setSelectedContextMenuItem] = useState(-1);
+    const [focusedContextMenuItem, setFocusedContextMenuItem] = useState(-1);
+    const [shouldClick, setShouldClick] =
+      useState(
+        false
+      ); /** We need this, because whenever we use the arrow keys to select the contextmenu item, and we focus the currently selected item it fires the "click" listener in ContextMenuItem component. Instead, we only want this to fire if we press "enter" or "spacebar" so we set this to true whenever that is the case, and back to false when it has been executed. */
+    const [childrenCount] = useState(React.Children.count(children));
 
     if (!id) {
       throw new Error('You need to provide an ID to the context menu');
     }
 
+    const onArrowNavigation = (event: React.KeyboardEvent) => {
+      if (focusedContextMenuItem === -1 && selectedContextMenuItem !== -1) {
+        setFocusedContextMenuItem(selectedContextMenuItem);
+      }
+
+      const codesToPrevenDefault = [
+        'ArrowDown',
+        'ArrowUp',
+        'ArrowLeft',
+        'ArrowRight',
+        'Enter',
+        'Space',
+        'Escape',
+        'End',
+        'Home',
+      ];
+
+      if (codesToPrevenDefault.includes(event.code)) {
+        event.preventDefault();
+      }
+
+      switch (event.code) {
+        case 'ArrowDown':
+          if (!showContextMenu) {
+            setShowContextMenu(true);
+            return;
+          }
+          setFocusedContextMenuItem((prevState) => {
+            return prevState + 1 > childrenCount - 1 ? 0 : prevState + 1;
+          });
+          return;
+        case 'ArrowUp':
+          setFocusedContextMenuItem((prevState) => {
+            return prevState - 1 < 0 ? childrenCount - 1 : prevState - 1;
+          });
+          return;
+        case 'Enter':
+        case 'Space':
+          if (!showContextMenu) {
+            setShowContextMenu(true);
+            return;
+          }
+
+          setShouldClick(true);
+          setSelectedContextMenuItem(focusedContextMenuItem);
+          setShowContextMenu(false);
+          return;
+        case 'Tab':
+        case 'Escape':
+          setShowContextMenu(false);
+          return;
+        case 'End':
+          setFocusedContextMenuItem(childrenCount - 1);
+          return;
+        case 'Home':
+          setFocusedContextMenuItem(0);
+          return;
+      }
+    };
+
     useBodyClick(
       (event) => {
         return showContextMenu && anchorEl.current !== event.target;
@@ -60,6 +133,8 @@ export const ContextMenu = React.forwardRef<HTMLDivElement, Props>(
         onShow && onShow();
       } else {
         onClose && onClose();
+        setFocusedContextMenuItem(-1);
+        anchorEl.current && anchorEl.current.focus();
       }
     }, [showContextMenu]);
 
@@ -68,13 +143,31 @@ export const ContextMenu = React.forwardRef<HTMLDivElement, Props>(
         id: id,
         'aria-haspopup': 'true',
         'aria-controls': `${id}-menu`,
-        'aria-expanded': show,
+        'aria-expanded': showContextMenu,
         onClick: () => setShowContextMenu(!showContextMenu),
+        tabIndex: 0,
         ref: anchorEl,
       });
 
+    const renderChildren = () => {
+      return React.Children.map(children, (child, index) => {
+        return React.cloneElement(child as ReactElement, {
+          onFocusChange: (childIndex: number) => setFocusedContextMenuItem(childIndex),
+          onSelectedChange: (childIndex: number) => {
+            setSelectedContextMenuItem(childIndex);
+            setShouldClick(false);
+          },
+          childIndex: index,
+          hasFocus: focusedContextMenuItem === index,
+          isSelected: selectedContextMenuItem === index,
+          contextMenuOpened: showContextMenu,
+          shouldClick: shouldClick,
+        });
+      });
+    };
+
     return (
-      <div {...rest} ref={ref} className={classes['context-menu']}>
+      <div {...rest} ref={ref} onKeyDown={onArrowNavigation} className={classes['context-menu']}>
         {renderTrigger()}
         {createPortal(
           <Popover
@@ -85,7 +178,7 @@ export const ContextMenu = React.forwardRef<HTMLDivElement, Props>(
             show={showContextMenu}
           >
             <ul className={classes.menu} id={`${id}-menu`} aria-describedby={id} role="menu">
-              {children}
+              {renderChildren()}
             </ul>
           </Popover>,
           domRoot

package/src/ContextMenu/ContextMenuItem.tsx

@@ -1,15 +1,63 @@
-import React, { HTMLProps } from 'react';
+import React, { ComponentPropsWithRef, createRef, RefObject, useEffect } from 'react';
 import classes from './ContextMenuItem.module.scss';
 
-export interface Props extends Omit<HTMLProps<HTMLLIElement>, 'onClick'> {
+export interface Props extends Omit<ComponentPropsWithRef<'button'>, 'onClick'> {
   children?: string;
+  hasFocus?: boolean;
+  isSelected?: boolean;
+  childIndex?: number;
+  shouldClick?: boolean;
+  contextMenuOpened?: boolean;
   onClick?: (event: React.MouseEvent<HTMLButtonElement>) => void;
+  onFocusChange?: (childIndex: number) => void;
+  onSelectedChange?: (childIndex: number) => void;
 }
 
-export const ContextMenuItem = ({ children, onClick, ...rest }: Props) => {
-  return (
-    <li {...rest} role="menuitem" className={classes['context-menu-item']}>
-      <button onClick={(event) => onClick && onClick(event)}>{children}</button>
-    </li>
-  );
-};
+export const ContextMenuItem = React.forwardRef<HTMLButtonElement, Props>(
+  (
+    {
+      children,
+      onClick,
+      onFocusChange,
+      onSelectedChange,
+      hasFocus,
+      isSelected,
+      childIndex,
+      contextMenuOpened,
+      shouldClick,
+      ...rest
+    }: Props,
+    ref
+  ) => {
+    let innerButtonRef = (ref as RefObject<HTMLButtonElement>) || createRef<HTMLButtonElement>();
+
+    useEffect(() => {
+      if (isSelected && innerButtonRef.current && shouldClick) {
+        innerButtonRef.current.click();
+      }
+    }, [isSelected, shouldClick]);
+
+    useEffect(() => {
+      if (innerButtonRef.current && hasFocus && contextMenuOpened) {
+        onFocusChange && childIndex && onFocusChange(childIndex);
+        innerButtonRef.current.focus();
+      }
+    }, [hasFocus, innerButtonRef, contextMenuOpened]);
+
+    return (
+      <li role="menuitem" className={classes['context-menu-item']}>
+        <button
+          {...rest}
+          ref={innerButtonRef}
+          data-focus={hasFocus}
+          onClick={(event) => {
+            onClick && onClick(event);
+            onSelectedChange && childIndex && onSelectedChange(childIndex);
+          }}
+        >
+          {children}
+        </button>
+      </li>
+    );
+  }
+);

package/src/DataGrid/DataGrid.module.scss

@@ -0,0 +1,21 @@
+.table-wrapper {
+  overflow-x: auto;
+}
+
+.table {
+  border-collapse: collapse;
+  padding-top: 0.5rem;
+  width: 100%;
+  position: relative;
+}
+
+.pagination {
+  margin-top: 1rem;
+  padding: 0 1rem;
+}
+
+@media only screen and (min-width: 50em) {
+  .pagination {
+    padding: 0 1.25rem;
+  }
+}

package/src/DataGrid/DataGrid.test.tsx

@@ -0,0 +1,276 @@
+import React, { useEffect, useRef } from 'react';
+import { DataGrid, Props } from './DataGrid';
+import { getAllByRole, render, queryAllByRole, getByRole } from '@testing-library/react';
+import { DataGridRow } from './DataGridBody/DataGridRow';
+import { DataGridCell } from './DataGridBody/DataGridCell';
+import { ContextMenu } from '../ContextMenu/ContextMenu';
+import { IconButton } from '../Button/IconButton';
+import { Icon, Icons } from '../Icon/Icon';
+import { ContextMenuItem } from '../ContextMenu/ContextMenuItem';
+import userEvent from '@testing-library/user-event';
+
+type DataType = { firstName: string; lastName: string; date: string };
+
+const defaultParams: Props<DataType> = {
+  children: ({ item }) => (
+    <DataGridRow key={item.firstName}>
+      <DataGridCell>{item.firstName}</DataGridCell>
+      <DataGridCell>{item.lastName}</DataGridCell>
+      <DataGridCell>{item.date}</DataGridCell>
+      <DataGridCell>
+        <ContextMenu
+          id={`consent_menu_${item.firstName}`}
+          trigger={
+            <IconButton title={`Actions for ${item.firstName}`} color="default">
+              <Icon icon={Icons.EllipsisAlt} />
+            </IconButton>
+          }
+        >
+          <ContextMenuItem>Item 1</ContextMenuItem>
+        </ContextMenu>
+      </DataGridCell>
+    </DataGridRow>
+  ),
+  headers: [
+    { name: 'firstName', headline: 'First name' },
+    { name: 'lastName', headline: 'Last name' },
+    { name: 'date', headline: 'Date' },
+  ],
+  enableMultiSorting: true,
+  data: [
+    { firstName: 'Paweł', lastName: 'Napieracz', date: '12.12.1990' },
+    { firstName: 'Michał', lastName: 'Górski', date: '12.12.1994' },
+    { firstName: 'Daniel', lastName: 'Velden', date: '12.12.199x' },
+    { firstName: 'Jasha', lastName: 'Joachimsthal', date: '12.12.198x' },
+  ],
+};
+
+const createDataGrid = (params?: (defaultParams: Props<DataType>) => Props<DataType>) => {
+  let parameters = defaultParams;
+  if (params) {
+    parameters = params(defaultParams);
+  }
+  const queries = render(<DataGrid {...parameters} data-testid="dataGrid" />);
+  const dataGrid = queries.getByTestId('dataGrid');
+
+  return {
+    ...queries,
+    dataGrid,
+  };
+};
+
+describe('DataGrid should render', () => {
+  it('renders without crashing', () => {
+    const rowsCount = defaultParams.data?.length!;
+    const colsCount = defaultParams.headers?.length!;
+    const contextMenuCellCount = 1;
+    const { dataGrid, ...queries } = createDataGrid();
+
+    expect(dataGrid).toBeDefined();
+
+    expect(dataGrid.querySelector('.actions')).toBeNull();
+
+    const [thead, tbody] = queries.getAllByRole('rowgroup');
+
+    expect(queryAllByRole(thead, 'columnheader')).toHaveLength(3);
+    expect(queryAllByRole(thead, 'cell')).toHaveLength(contextMenuCellCount);
+    expect(queryAllByRole(thead, 'button')).toHaveLength(0); //sorting should be disabled by default
+
+    expect(queryAllByRole(tbody, 'cell')).toHaveLength(
+      rowsCount * (colsCount + contextMenuCellCount)
+    );
+  });
+
+  it('renders actions buttons', () => {
+    const { dataGrid } = createDataGrid((params) => ({
+      ...params,
+      actions: {
+        enableAddBtn: true,
+        enableColumnsBtn: true,
+        enableSearchBtn: true,
+      },
+    }));
+    const [addBtn, desktopColumnsBtn, mobileColumnsBtn, desktopSearchBtn, mobileSearchBtn] =
+      getAllByRole(dataGrid.querySelector('.actions')!, 'button');
+    expect(addBtn).toHaveTextContent('Add item');
+    expect(desktopColumnsBtn).toHaveTextContent('Columns');
+    expect(mobileColumnsBtn).toHaveTextContent('Show/hide columns');
+    expect(desktopSearchBtn).toHaveTextContent('Search');
+    expect(mobileSearchBtn).toHaveTextContent('Search');
+  });
+
+  it('renders table with initialized sorting indicators', () => {
+    const { getAllByRole } = createDataGrid((params) => ({
+      ...params,
+      onSort: jest.fn(),
+      initialSort: [
+        { name: defaultParams.headers[0].name, direction: 'DESC' },
+        { name: defaultParams.headers[2].name, direction: 'ASC' },
+      ],
+    }));
+
+    const [firstNameHeader, lastNameHeader, dateHeader] = getAllByRole('columnheader');
+    expect(firstNameHeader).toHaveAttribute('aria-sort', 'descending');
+    expect(lastNameHeader).not.toHaveAttribute('aria-sort');
+    expect(dateHeader).toHaveAttribute('aria-sort', 'ascending');
+  });
+
+  it('renders table without context menu', () => {
+    const rowsCount = defaultParams.data?.length!;
+    const colsCount = defaultParams.headers?.length!;
+    const { getAllByRole } = createDataGrid((params) => ({
+      ...params,
+      disableContextMenuColumn: true,
+      children: ({ item }) => (
+        <DataGridRow key={item.firstName}>
+          <DataGridCell>{item.firstName}</DataGridCell>
+          <DataGridCell>{item.lastName}</DataGridCell>
+          <DataGridCell>{item.date}</DataGridCell>
+        </DataGridRow>
+      ),
+    }));
+
+    expect(getAllByRole('columnheader')).toHaveLength(3);
+    expect(getAllByRole('cell')).toHaveLength(rowsCount * colsCount);
+  });
+
+  it('renders loading state with skeleton loading animation on data cells and empty row on content menu cells', () => {
+    const { ...queries } = createDataGrid((params) => ({ ...params, isLoading: true }));
+
+    const [_thead, tbody] = queries.getAllByRole('rowgroup');
+    expect(
+      getAllByRole(tbody, 'cell').find((cell, index) => {
+        const contextMenuCell = index % 4 === 3;
+        return (
+          !contextMenuCell && (cell.firstChild as HTMLElement).getAttribute('class') !== 'loading'
+        );
+      })
+    ).toBeUndefined();
+  });
+
+  it('renders empty state', () => {
+    const emptyLabel = 'emptyLabel';
+    const { ...queries } = createDataGrid((params) => ({ ...params, emptyLabel, data: [] }));
+
+    const [_thead, tbody] = queries.getAllByRole('rowgroup');
+    const rows = getAllByRole(tbody, 'row');
+    const firstRowCells = getAllByRole(rows[0], 'cell');
+    expect(firstRowCells).toHaveLength(1);
+    expect(firstRowCells[0]).toHaveAttribute('colspan', `${defaultParams.headers.length + 1}`);
+    expect(firstRowCells[0]).toHaveTextContent(emptyLabel);
+  });
+
+  it('renders pagination', () => {
+    const { getByRole } = createDataGrid((params) => ({
+      ...params,
+      paginationProps: {
+        totalElements: 105,
+        pageSize: 10,
+        currentPage: 2,
+        onPageChange: jest.fn(),
+        onPageSizeChange: jest.fn(),
+      },
+    }));
+
+    expect(getByRole('button', { name: 'previous' })).toBeDefined();
+    expect(getByRole('button', { name: 'next' })).toBeDefined();
+    expect(getByRole('button', { name: 'last' })).toBeDefined();
+  });
+});
+
+describe('DataGrid should have interactive table header', () => {
+  it('clicking table headers should call sort callback', () => {
+    const onSort = jest.fn();
+    const { getAllByRole } = createDataGrid((params) => ({
+      ...params,
+      onSort,
+    }));
+
+    const [firstNameHeader, lastNameHeader, dateHeader] = getAllByRole('columnheader');
+
+    userEvent.click(getByRole(firstNameHeader, 'button'));
+    expect(onSort).toBeCalledWith([{ name: defaultParams.headers[0].name, direction: 'ASC' }]);
+
+    userEvent.click(getByRole(lastNameHeader, 'button'));
+    userEvent.click(getByRole(dateHeader, 'button'));
+    expect(onSort).toBeCalledWith([
+      { name: defaultParams.headers[2].name, direction: 'ASC' },
+      { name: defaultParams.headers[1].name, direction: 'ASC' },
+      { name: defaultParams.headers[0].name, direction: 'ASC' },
+    ]);
+
+    userEvent.click(getByRole(firstNameHeader, 'button'));
+    expect(onSort).toBeCalledWith([
+      { name: defaultParams.headers[0].name, direction: 'DESC' },
+      { name: defaultParams.headers[2].name, direction: 'ASC' },
+      { name: defaultParams.headers[1].name, direction: 'ASC' },
+    ]);
+
+    userEvent.click(getByRole(dateHeader, 'button'));
+    expect(onSort).toBeCalledWith([
+      { name: defaultParams.headers[2].name, direction: 'DESC' },
+      { name: defaultParams.headers[0].name, direction: 'DESC' },
+      { name: defaultParams.headers[1].name, direction: 'ASC' },
+    ]);
+
+    userEvent.click(getByRole(dateHeader, 'button'));
+    userEvent.click(getByRole(firstNameHeader, 'button'));
+    userEvent.click(getByRole(lastNameHeader, 'button'));
+    expect(onSort).toBeCalledWith([{ name: defaultParams.headers[1].name, direction: 'DESC' }]);
+    expect(onSort).toBeCalledTimes(8);
+  });
+
+  it('clicking on show/hide columns popover should show/hide columns', async () => {
+    const { getByRole, findByLabelText, ...queries } = createDataGrid((params) => ({
+      ...params,
+      actions: { enableColumnsBtn: true },
+    }));
+    expect(queries.getAllByRole('columnheader')).toHaveLength(3);
+
+    userEvent.click(getByRole('button', { name: 'Columns' }));
+
+    const nameToggle = await findByLabelText(defaultParams.headers[1].headline);
+    expect(nameToggle).toBeChecked();
+    userEvent.click(nameToggle);
+    expect(nameToggle).not.toBeChecked();
+    expect(getByRole('button', { name: 'Close show columns dialog' })).toBeDefined();
+
+    const columns = queries.getAllByRole('columnheader');
+    expect(columns).toHaveLength(2);
+    expect(columns[0]).toHaveTextContent(defaultParams.headers[0].headline);
+    expect(columns[1]).toHaveTextContent(defaultParams.headers[2].headline);
+
+    const rowsCount = defaultParams.data?.length!;
+    const colsCount = defaultParams.headers.length;
+    const hiddenColsCount = defaultParams.headers.length;
+    const contextMenuColsCount = defaultParams.headers.length;
+    const [_thead, tbody] = queries.getAllByRole('rowgroup');
+    expect(getAllByRole(tbody, 'cell')).toHaveLength(
+      rowsCount * (colsCount - hiddenColsCount + contextMenuColsCount)
+    );
+  });
+});
+
+describe('ref should work', () => {
+  it('should give back the proper data prop, this also checks if the component propagates ...rest properly', () => {
+    const ExampleComponent = ({
+      propagateRef,
+    }: {
+      propagateRef: (ref: React.RefObject<HTMLElement>) => void;
+    }) => {
+      const ref = useRef(null);
+
+      useEffect(() => {
+        propagateRef(ref);
+      }, [ref]);
+
+      return <DataGrid {...defaultParams} data-ref="testing" ref={ref} />;
+    };
+
+    const refCheck = (ref: React.RefObject<HTMLElement>) => {
+      expect(ref.current).toHaveAttribute('data-ref', 'testing');
+    };
+
+    render(<ExampleComponent propagateRef={refCheck} />);
+  });
+});