Tests: Allow async domEqual() tests.

1 files changed, 14 insertions, 3 deletions

diff --git a/tests/unit/testsuite.js b/tests/unit/testsuite.js
index f3c0982fb..42fdf4f9e 100644
--- a/tests/unit/testsuite.js
+++ b/tests/unit/testsuite.js
@@ -271,11 +271,22 @@ window.domEqual = function( selector, modifier, message ) {
 		}
 		return result;
 	}
+
+	function done() {
+		actual = extract( $( selector ) );
+		QUnit.push( QUnit.equiv(actual, expected), actual, expected, message );
+	}
+
+	// Get current state prior to modifier
 	expected = extract( $( selector ) );
-	modifier( $( selector ) );
 
-	actual = extract( $( selector ) );
-	QUnit.push( QUnit.equiv(actual, expected), actual, expected, message );
+	// Run modifier (async or sync), then compare state via done()
+	if ( modifier.length ) {
+		modifier( done );
+	} else {
+		modifier();
+		done();
+	}
 };
 
 }( jQuery ));