From 76d1190ff8a77ff600934e225628be41f24694d9 Mon Sep 17 00:00:00 2001
From: XieZongChen <46394163+amadeus711@users.noreply.github.com>
Date: Fri, 1 Oct 2021 21:34:37 -0500
Subject: [PATCH] test(data-table): update test (#1273)

---
 src/data-table/tests/DataTable.spec.tsx | 29 +++++++++++++++++++++++++
 1 file changed, 29 insertions(+)

diff --git a/src/data-table/tests/DataTable.spec.tsx b/src/data-table/tests/DataTable.spec.tsx
index 189a2a427..5cc87c2e7 100644
--- a/src/data-table/tests/DataTable.spec.tsx
+++ b/src/data-table/tests/DataTable.spec.tsx
@@ -558,4 +558,33 @@ describe('n-data-table', () => {
     ))
     expect(wrapper.find('tbody').element.children.length).toBe(0)
   })
+
+  it('should work with `on-update:checked-row-keys` prop', async () => {
+    const handleCheck = jest.fn()
+    const columns: DataTableColumns = [
+      {
+        type: 'selection'
+      },
+      {
+        title: 'Name',
+        key: 'name'
+      }
+    ]
+    const data = new Array(2).fill(0).map((_, index) => {
+      return {
+        name: index
+      }
+    })
+    const rowKey = (row: any): number => row.name
+    const wrapper = mount(() => (
+      <NDataTable
+        columns={columns}
+        data={data}
+        row-key={rowKey}
+        onUpdateCheckedRowKeys={handleCheck}
+      />
+    ))
+    await wrapper.find('.n-checkbox').trigger('click')
+    expect(handleCheck).toHaveBeenCalled()
+  })
 })