From 45509a7cf9962be256be1e30fc3a732852b7212a Mon Sep 17 00:00:00 2001
From: Brian Ingles <brianingles@deephaven.io>
Date: Fri, 22 Mar 2024 12:23:07 -0500
Subject: [PATCH] PickerUtils tests (#1858)

---
 .../src/spectrum/Picker/PickerUtils.test.tsx  | 49 +++++++++++++++++++
 1 file changed, 49 insertions(+)
 create mode 100644 packages/jsapi-components/src/spectrum/Picker/PickerUtils.test.tsx

diff --git a/packages/jsapi-components/src/spectrum/Picker/PickerUtils.test.tsx b/packages/jsapi-components/src/spectrum/Picker/PickerUtils.test.tsx
new file mode 100644
index 0000000000..a0f4a43735
--- /dev/null
+++ b/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);
+    }
+  );
+});