PickerUtils tests (#1858)

packages/jsapi-components/src/spectrum/Picker/PickerUtils.test.tsx

@@ -0,0 +1,49 @@
+import { type dh } from '@deephaven/jsapi-types';
+import { TestUtils } from '@deephaven/utils';
+import { getPickerKeyColumn, getPickerLabelColumn } from './PickerUtils';
+
+const { createMockProxy } = TestUtils;
+
+const keyColumn = createMockProxy<dh.Column>({ name: 'keyColumn' });
+const labelColumn = createMockProxy<dh.Column>({ name: 'labelColumn' });
+const otherColumn = createMockProxy<dh.Column>({ name: 'otherColumn' });
+
+const columns = [keyColumn, labelColumn, otherColumn];
+
+const table = createMockProxy<dh.Table>({
+  columns,
+  findColumn: jest.fn(
+    columnName => columns.find(column => column.name === columnName)!
+  ),
+});
+
+beforeEach(() => {
+  jest.clearAllMocks();
+  expect.hasAssertions();
+});
+
+describe('getPickerKeyColumn', () => {
+  it.each([
+    ['keyColumn', keyColumn],
+    [undefined, columns[0]],
+  ])(
+    'should return the given key column or fallback to the first column',
+    (keyColumnName, expectedColumn) => {
+      const actual = getPickerKeyColumn(table, keyColumnName);
+      expect(actual).toBe(expectedColumn);
+    }
+  );
+});
+
+describe('getPickerLabelColumn', () => {
+  it.each([
+    ['labelColumn', labelColumn],
+    [undefined, keyColumn],
+  ])(
+    'should return the given label column or fallback to the key column',
+    (labelColumnName, expectedColumn) => {
+      const actual = getPickerLabelColumn(table, keyColumn, labelColumnName);
+      expect(actual).toBe(expectedColumn);
+    }
+  );
+});