test(tables): add updateRow partial merge tests

Covers the bug where partial updates wiped unmentioned columns — verifies
that fields not in the update payload are preserved, nulling a field works,
full-row updates are idempotent, and missing rows throw correctly.

Author: waleedlatif1

apps/sim/lib/table/__tests__/update-row.test.ts

@@ -0,0 +1,120 @@
+/**
+ * @vitest-environment node
+ */
+import { databaseMock } from '@sim/testing'
+import { beforeEach, describe, expect, it, vi } from 'vitest'
+import { updateRow } from '../service'
+import type { TableDefinition } from '../types'
+
+const EXISTING_ROW = {
+  id: 'row-1',
+  tableId: 'tbl-1',
+  workspaceId: 'ws-1',
+  data: { name: 'Alice', age: 30 },
+  position: 1,
+  createdAt: new Date('2024-01-01'),
+  updatedAt: new Date('2024-01-01'),
+}
+
+const TABLE: TableDefinition = {
+  id: 'tbl-1',
+  name: 'People',
+  description: null,
+  schema: {
+    columns: [
+      { name: 'name', type: 'string' },
+      { name: 'age', type: 'number' },
+    ],
+  },
+  metadata: null,
+  rowCount: 0,
+  maxRows: 1000,
+  workspaceId: 'ws-1',
+  createdBy: 'user-1',
+  archivedAt: null,
+  createdAt: new Date('2024-01-01'),
+  updatedAt: new Date('2024-01-01'),
+}
+
+describe('updateRow — partial merge', () => {
+  let mockSet: ReturnType<typeof vi.fn>
+
+  beforeEach(() => {
+    vi.clearAllMocks()
+
+    // db.select() → used by getRowById to fetch the existing row
+    const mockLimit = vi.fn().mockResolvedValue([EXISTING_ROW])
+    const mockSelectWhere = vi.fn().mockReturnValue({ limit: mockLimit })
+    const mockFrom = vi.fn().mockReturnValue({ where: mockSelectWhere })
+    databaseMock.db.select.mockReturnValue({ from: mockFrom })
+
+    // db.update() → captures what merged data gets written
+    const mockUpdateWhere = vi.fn().mockResolvedValue([])
+    mockSet = vi.fn().mockReturnValue({ where: mockUpdateWhere })
+    databaseMock.db.update.mockReturnValue({ set: mockSet })
+  })
+
+  it('preserves columns not included in the partial update', async () => {
+    const result = await updateRow(
+      { tableId: 'tbl-1', rowId: 'row-1', data: { age: 31 }, workspaceId: 'ws-1' },
+      TABLE,
+      'req-1'
+    )
+
+    expect(result.data).toEqual({ name: 'Alice', age: 31 })
+    expect(mockSet).toHaveBeenCalledWith(
+      expect.objectContaining({ data: { name: 'Alice', age: 31 } })
+    )
+  })
+
+  it('allows updating a single column without affecting others', async () => {
+    const result = await updateRow(
+      { tableId: 'tbl-1', rowId: 'row-1', data: { name: 'Bob' }, workspaceId: 'ws-1' },
+      TABLE,
+      'req-1'
+    )
+
+    expect(result.data).toEqual({ name: 'Bob', age: 30 })
+    expect(mockSet).toHaveBeenCalledWith(
+      expect.objectContaining({ data: { name: 'Bob', age: 30 } })
+    )
+  })
+
+  it('allows explicitly nulling a field while preserving others', async () => {
+    const result = await updateRow(
+      { tableId: 'tbl-1', rowId: 'row-1', data: { age: null }, workspaceId: 'ws-1' },
+      TABLE,
+      'req-1'
+    )
+
+    expect(result.data).toEqual({ name: 'Alice', age: null })
+    expect(mockSet).toHaveBeenCalledWith(
+      expect.objectContaining({ data: { name: 'Alice', age: null } })
+    )
+  })
+
+  it('handles a full-row update correctly (idempotent merge)', async () => {
+    const result = await updateRow(
+      { tableId: 'tbl-1', rowId: 'row-1', data: { name: 'Bob', age: 25 }, workspaceId: 'ws-1' },
+      TABLE,
+      'req-1'
+    )
+
+    expect(result.data).toEqual({ name: 'Bob', age: 25 })
+  })
+
+  it('throws when the row does not exist', async () => {
+    const mockLimit = vi.fn().mockResolvedValue([])
+    const mockSelectWhere = vi.fn().mockReturnValue({ limit: mockLimit })
+    const mockFrom = vi.fn().mockReturnValue({ where: mockSelectWhere })
+    databaseMock.db.select.mockReturnValue({ from: mockFrom })
+
+    await expect(
+      updateRow(
+        { tableId: 'tbl-1', rowId: 'row-missing', data: { age: 31 }, workspaceId: 'ws-1' },
+        TABLE,
+        'req-1'
+      )
+    ).rejects.toThrow('Row not found')
+  })
+})