test(set-map): update

unadlib · Nov 23, 2024 · 259445c · 259445c
1 parent 9410f7d
commit 259445c
Showing 1 changed file with 10 additions and 22 deletions.
diff --git a/test/immer-non-support.test.ts b/test/immer-non-support.test.ts
@@ -672,17 +672,11 @@ test('CustomSet', () => {
     }
 
     const state = new CustomSet();
-    const newState = create(
-      state,
-      (draft) => {
-        draft.add(1);
-        // @ts-ignore
-        expect(draft.getIdentity()).toBe('CustomSet');
-      },
-      {
-        mark: () => 'immutable',
-      }
-    );
+    const newState = create(state, (draft) => {
+      draft.add(1);
+      // @ts-ignore
+      expect(draft.getIdentity()).toBe('CustomSet');
+    });
     expect(newState instanceof CustomSet).toBeTruthy();
     // @ts-ignore
     expect(newState.getIdentity()).toBe('CustomSet');
@@ -717,17 +711,11 @@ test('CustomMap', () => {
     }
 
     const state = new CustomMap();
-    const newState = create(
-      state,
-      (draft) => {
-        draft.set(1, 1);
-        // @ts-ignore
-        expect(draft.getIdentity()).toBe('CustomMap');
-      },
-      {
-        mark: () => 'immutable',
-      }
-    );
+    const newState = create(state, (draft) => {
+      draft.set(1, 1);
+      // @ts-ignore
+      expect(draft.getIdentity()).toBe('CustomMap');
+    });
     expect(newState instanceof CustomMap).toBeTruthy();
     // @ts-ignore
     expect(newState.getIdentity()).toBe('CustomMap');