Handle arrays with properties in assert.deepEqual

Author: Andy Hanson

src/harness/harness.ts

@@ -32,9 +32,38 @@
 // this will work in the browser via browserify
 var _chai: typeof chai = require("chai");
 var assert: typeof _chai.assert = _chai.assert;
-// chai's builtin `assert.isFalse` is featureful but slow - we don't use those features,
-// so we'll just overwrite it as an alterative to migrating a bunch of code off of chai
-assert.isFalse = (expr, msg) => { if (expr as any as boolean !== false) throw new Error(msg); };
+{
+    // chai's builtin `assert.isFalse` is featureful but slow - we don't use those features,
+    // so we'll just overwrite it as an alterative to migrating a bunch of code off of chai
+    assert.isFalse = (expr, msg) => { if (expr as any as boolean !== false) throw new Error(msg); };
+
+    const assertDeepImpl = assert.deepEqual;
+    assert.deepEqual = (a, b, msg) => {
+        if (ts.isArray(a) && ts.isArray(b)) {
+            assertDeepImpl(arrayExtraKeysObject(a), arrayExtraKeysObject(b), "Array extra keys differ");
+        }
+        assertDeepImpl(a, b, msg);
+
+        function arrayExtraKeysObject(a: ReadonlyArray<{} | null | undefined>): object {
+            const obj: { [key: string]: {} | null | undefined } = {};
+            for (const key in a) {
+                if (Number.isNaN(Number(key))) {
+                    obj[key] = a[key];
+                }
+            }
+            return obj;
+        }
+    };
+
+    describe("assert", () => {
+        it("deepEqual", () => {
+            assert.throws(() => assert.deepEqual(ts.createNodeArray([ts.createIdentifier("A")]), ts.createNodeArray([ts.createIdentifier("B")])));
+            assert.throws(() => assert.deepEqual(ts.createNodeArray([], /*hasTrailingComma*/ true), ts.createNodeArray([], /*hasTrailingComma*/ false)));
+            assert.deepEqual(ts.createNodeArray([ts.createIdentifier("A")], /*hasTrailingComma*/ true), ts.createNodeArray([ts.createIdentifier("A")], /*hasTrailingComma*/ true));
+        });
+    });
+}
+
 declare var __dirname: string; // Node-specific
 var global: NodeJS.Global = <any>Function("return this").call(undefined);