Function setup

test/browser/utils.tsx:7–32

(
  props: DataGridProps<R, SR, K>,
  renderBeforeAfterButtons = false
)

import type { DataGridProps } from '../../src';

export function setup<R, SR, K extends React.Key = React.Key>(
  props: DataGridProps<R, SR, K>,
  renderBeforeAfterButtons = false
) {
  const grid = (
    <DataGrid
      {...props}
      className={css`
        block-size: 1080px;
        scrollbar-width: none;
      `}
    />
  );

  if (renderBeforeAfterButtons) {
    return page.render(
      <>
        <button type="button">Before</button>
        {grid}
        <br />
        <button type="button">After</button>
      </>
    );
  }
  return page.render(grid);
}

export function getGrid() {
  return page.getByRole('grid');