|
1 | 1 | import type { Column } from '../../../src'; |
2 | 2 | import { cellClassname } from '../../../src/style/cell'; |
3 | | -import { getCells, setup } from '../utils'; |
| 3 | +import { getCellsNew, setupNew } from '../utils'; |
4 | 4 |
|
5 | 5 | interface Row { |
6 | 6 | id: number; |
7 | 7 | } |
8 | 8 |
|
9 | 9 | const rows: readonly Row[] = [{ id: 0 }, { id: 1 }]; |
10 | 10 |
|
11 | | -test('cellClass is undefined', () => { |
| 11 | +test('cellClass is undefined', async () => { |
12 | 12 | const columns: readonly Column<Row>[] = [ |
13 | 13 | { |
14 | 14 | key: 'id', |
15 | 15 | name: 'ID' |
16 | 16 | } |
17 | 17 | ]; |
18 | | - setup({ columns, rows }); |
19 | | - const [cell1, cell2] = getCells(); |
20 | | - expect(cell1).toHaveClass(cellClassname, { exact: true }); |
21 | | - expect(cell2).toHaveClass(cellClassname, { exact: true }); |
| 18 | + setupNew({ columns, rows }); |
| 19 | + const [cell1, cell2] = getCellsNew(); |
| 20 | + await expect.element(cell1).toHaveClass(cellClassname, { exact: true }); |
| 21 | + await expect.element(cell2).toHaveClass(cellClassname, { exact: true }); |
22 | 22 | }); |
23 | 23 |
|
24 | | -test('cellClass is a string', () => { |
| 24 | +test('cellClass is a string', async () => { |
25 | 25 | const columns: readonly Column<Row>[] = [ |
26 | 26 | { |
27 | 27 | key: 'id', |
28 | 28 | name: 'ID', |
29 | 29 | cellClass: 'my-cell' |
30 | 30 | } |
31 | 31 | ]; |
32 | | - setup({ columns, rows }); |
33 | | - const [cell1, cell2] = getCells(); |
34 | | - expect(cell1).toHaveClass(`${cellClassname} my-cell`, { exact: true }); |
35 | | - expect(cell2).toHaveClass(`${cellClassname} my-cell`, { exact: true }); |
| 32 | + setupNew({ columns, rows }); |
| 33 | + const [cell1, cell2] = getCellsNew(); |
| 34 | + await expect.element(cell1).toHaveClass(`${cellClassname} my-cell`, { exact: true }); |
| 35 | + await expect.element(cell2).toHaveClass(`${cellClassname} my-cell`, { exact: true }); |
36 | 36 | }); |
37 | 37 |
|
38 | | -test('cellClass returns a string', () => { |
| 38 | +test('cellClass returns a string', async () => { |
39 | 39 | const columns: readonly Column<Row>[] = [ |
40 | 40 | { |
41 | 41 | key: 'id', |
42 | 42 | name: 'ID', |
43 | 43 | cellClass: (row) => `my-cell-${row.id}` |
44 | 44 | } |
45 | 45 | ]; |
46 | | - setup({ columns, rows }); |
47 | | - const [cell1, cell2] = getCells(); |
48 | | - expect(cell1).toHaveClass(`${cellClassname} my-cell-0`, { exact: true }); |
49 | | - expect(cell2).toHaveClass(`${cellClassname} my-cell-1`, { exact: true }); |
| 46 | + setupNew({ columns, rows }); |
| 47 | + const [cell1, cell2] = getCellsNew(); |
| 48 | + await expect.element(cell1).toHaveClass(`${cellClassname} my-cell-0`, { exact: true }); |
| 49 | + await expect.element(cell2).toHaveClass(`${cellClassname} my-cell-1`, { exact: true }); |
50 | 50 | }); |
51 | 51 |
|
52 | | -test('cellClass returns undefined', () => { |
| 52 | +test('cellClass returns undefined', async () => { |
53 | 53 | const columns: readonly Column<Row>[] = [ |
54 | 54 | { |
55 | 55 | key: 'id', |
56 | 56 | name: 'ID', |
57 | 57 | cellClass: () => undefined |
58 | 58 | } |
59 | 59 | ]; |
60 | | - setup({ columns, rows }); |
61 | | - const [cell1, cell2] = getCells(); |
62 | | - expect(cell1).toHaveClass(cellClassname, { exact: true }); |
63 | | - expect(cell2).toHaveClass(cellClassname, { exact: true }); |
| 60 | + setupNew({ columns, rows }); |
| 61 | + const [cell1, cell2] = getCellsNew(); |
| 62 | + await expect.element(cell1).toHaveClass(cellClassname, { exact: true }); |
| 63 | + await expect.element(cell2).toHaveClass(cellClassname, { exact: true }); |
64 | 64 | }); |
0 commit comments