fix(expect): fix immutable.js iterable equality (#5692)

Author: hi-ogawa

packages/expect/src/jest-utils.ts

@@ -268,7 +268,9 @@ export function hasProperty(obj: object | null, property: string): boolean {
 // SENTINEL constants are from https://github.com/facebook/immutable-js
 const IS_KEYED_SENTINEL = '@@__IMMUTABLE_KEYED__@@'
 const IS_SET_SENTINEL = '@@__IMMUTABLE_SET__@@'
+const IS_LIST_SENTINEL = '@@__IMMUTABLE_LIST__@@'
 const IS_ORDERED_SENTINEL = '@@__IMMUTABLE_ORDERED__@@'
+const IS_RECORD_SYMBOL = '@@__IMMUTABLE_RECORD__@@'
 
 export function isImmutableUnorderedKeyed(maybeKeyed: any) {
   return !!(
@@ -286,6 +288,36 @@ export function isImmutableUnorderedSet(maybeSet: any) {
   )
 }
 
+function isObjectLiteral(source: unknown): source is Record<string, unknown> {
+  return source != null && typeof source === 'object' && !Array.isArray(source)
+}
+
+function isImmutableList(source: unknown): boolean {
+  return Boolean(source && isObjectLiteral(source) && source[IS_LIST_SENTINEL])
+}
+
+function isImmutableOrderedKeyed(source: unknown): boolean {
+  return Boolean(
+    source
+    && isObjectLiteral(source)
+    && source[IS_KEYED_SENTINEL]
+    && source[IS_ORDERED_SENTINEL],
+  )
+}
+
+function isImmutableOrderedSet(source: unknown): boolean {
+  return Boolean(
+    source
+    && isObjectLiteral(source)
+    && source[IS_SET_SENTINEL]
+    && source[IS_ORDERED_SENTINEL],
+  )
+}
+
+function isImmutableRecord(source: unknown): boolean {
+  return Boolean(source && isObjectLiteral(source) && source[IS_RECORD_SYMBOL])
+}
+
 /**
  * Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.
  *
@@ -411,10 +443,17 @@ export function iterableEquality(a: any, b: any, customTesters: Array<Tester> =
   if (!bIterator.next().done)
     return false
 
-  const aEntries = Object.entries(a)
-  const bEntries = Object.entries(b)
-  if (!equals(aEntries, bEntries))
-    return false
+  if (
+    !isImmutableList(a)
+    && !isImmutableOrderedKeyed(a)
+    && !isImmutableOrderedSet(a)
+    && !isImmutableRecord(a)
+  ) {
+    const aEntries = Object.entries(a)
+    const bEntries = Object.entries(b)
+    if (!equals(aEntries, bEntries))
+      return false
+  }
 
   // Remove the first value from the stack of traversed values.
   aStack.pop()

pnpm-lock.yaml

@@ -21,6 +21,7 @@
     "@vueuse/integrations": "^10.9.0",
     "axios": "^0.26.1",
     "debug": "^4.3.4",
+    "immutable": "5.0.0-beta.5",
     "strip-ansi": "^7.1.0",
     "sweetalert2": "^11.6.16",
     "tinyspy": "^1.0.2",

test/core/package.json

@@ -0,0 +1,9 @@
+import { expect, test } from 'vitest'
+import im from 'immutable'
+
+test('basic', () => {
+  expect(im.List([{ x: 1 }])).toEqual(im.List([{ x: 1 }]))
+  expect(im.List([{ x: 1 }])).toEqual(im.List([1]).map(i => ({ x: i })))
+  expect(im.List([{ x: 1 }])).not.toEqual(im.List([{ x: 2 }]))
+  expect(im.List([{ x: 1 }])).not.toEqual(im.List([]))
+})