tests for remove() and delete()

ljharb · Jun 6, 2011 · f5d429a · f5d429a
1 parent 7010fe2
commit f5d429a
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 1 deletion.
diff --git a/test/circular.js b/test/circular.js
@@ -108,7 +108,7 @@ exports.circMapScrub = function () {
         Object.keys(scrubbed).sort(),
         [ 'a', 'b' ]
     );
-    Traverse.deepEqual(scrubbed, { a : 1, b : 2 });
+    assert.ok(Traverse.deepEqual(scrubbed, { a : 1, b : 2 }));
 
     assert.equal(obj.c, obj);
 };
diff --git a/test/mutability.js b/test/mutability.js
@@ -105,3 +105,31 @@ exports.removeMap = function () {
     assert.deepEqual(obj, { a : 1, b : 2, c : [ 3, 4 ] });
     assert.deepEqual(res, { a : 1, c : [ 3 ] });
 };
+
+exports.delete = function () {
+    var obj = { a : 1, b : 2, c : [ 3, 4 ] };
+    Traverse(obj).forEach(function (x) {
+        if (this.isLeaf && x % 2 == 0) this.delete();
+    });
+
+    assert.ok(Traverse.deepEqual(
+        obj,
+        { a : 1, c : [ 3, undefined ] }
+    ));
+};
+
+exports.deleteMap = function () {
+    var obj = { a : 1, b : 2, c : [ 3, 4 ] };
+    var res = Traverse(obj).map(function (x) {
+        if (this.isLeaf && x % 2 == 0) this.delete();
+    });
+
+    assert.ok(Traverse.deepEqual(
+        res,
+        { a : 1, b : 2, c : [ 3, 4 ] }
+    ));
+    assert.ok(Traverse.deepEqual(
+        res,
+        { a : 1, c : [ 3, undefined ] }
+    ));
+};