adding some additional test cases

Author: LFDanLu

packages/@react-stately/data/test/useListData.test.js

@@ -55,6 +55,26 @@ describe('useListData', function () {
       expect(result.current.selectedKeys).not.toBe(initialResult.selectedKeys);
       expect(result.current.selectedKeys).toEqual(new Set(['Sam', 'Julia']));
     });
+
+    it('should support adding "all" to selected keys', function () {
+      let {result} = renderHook(() => useListData({initialItems: initial, getKey, initialSelectedKeys: ['Sam']}));
+      let initialResult = result.current;
+
+      act(() => {
+        result.current.addKeysToSelection('all');
+      });
+      expect(result.current.selectedKeys).not.toBe(initialResult.selectedKeys);
+      expect(result.current.selectedKeys).toEqual('all');
+    });
+
+    it('should still return "all" if selected keys was already "all"', function () {
+      let {result} = renderHook(() => useListData({initialItems: initial, getKey, initialSelectedKeys: 'all'}));
+
+      act(() => {
+        result.current.addKeysToSelection(['Same']);
+      });
+      expect(result.current.selectedKeys).toEqual('all');
+    });
   });
 
   describe('removeKeysFromSelection', function () {
@@ -79,6 +99,28 @@ describe('useListData', function () {
       expect(result.current.selectedKeys).not.toBe(initialResult.selectedKeys);
       expect(result.current.selectedKeys).toEqual(new Set(['Julia']));
     });
+
+    it('should remove the selected keys from an "all" set', function () {
+      let {result} = renderHook(() => useListData({initialItems: initial, getKey, initialSelectedKeys: 'all'}));
+      let initialResult = result.current;
+
+      act(() => {
+        result.current.removeKeysFromSelection(['Sam', 'David']);
+      });
+      expect(result.current.selectedKeys).not.toBe(initialResult.selectedKeys);
+      expect(result.current.selectedKeys).toEqual(new Set(['Julia']));
+    });
+
+    it('should support removing "all"', function () {
+      let {result} = renderHook(() => useListData({initialItems: initial, getKey, initialSelectedKeys: ['Sam', 'Julia']}));
+      let initialResult = result.current;
+
+      act(() => {
+        result.current.removeKeysFromSelection('all');
+      });
+      expect(result.current.selectedKeys).not.toBe(initialResult.selectedKeys);
+      expect(result.current.selectedKeys).toEqual(new Set([]));
+    });
   });
 
   it('should get an item by key', function () {