Fix $DONE usage

test/built-ins/Promise/create-resolving-functions-reject.js

@@ -15,13 +15,14 @@ flags: [async]
 ---*/
 
 Promise.resolve(1).then(function() {
-  return Promise.resolve(2);
-});
+  return Promise.resolve();
+}).then($DONE, $DONE);
 
-Promise.prototype.then = function(_resolve, reject) {
+var then = Promise.prototype.then;
+Promise.prototype.then = function(resolve, reject) {
   assert(!isConstructor(reject));
   assert.sameValue(reject.length, 1);
   assert.sameValue(reject.name, '');
 
-  $DONE();
+  return then.call(this, resolve, reject);
 };

test/built-ins/Promise/create-resolving-functions-resolve.js

@@ -15,13 +15,14 @@ flags: [async]
 ---*/
 
 Promise.resolve(1).then(function() {
-  return Promise.resolve(2);
-});
+  return Promise.resolve();
+}).then($DONE, $DONE);
 
-Promise.prototype.then = function(resolve) {
+var then = Promise.prototype.then;
+Promise.prototype.then = function(resolve, reject) {
   assert(!isConstructor(resolve));
   assert.sameValue(resolve.length, 1);
   assert.sameValue(resolve.name, '');
 
-  $DONE();
+  return then.call(this, resolve, reject);
 };

test/built-ins/Promise/prototype/finally/rejected-observable-then-calls-argument.js

@@ -17,12 +17,16 @@ includes: [isConstructor.js]
 flags: [async]
 ---*/
 
-Promise.reject(new Test262Error()).finally(function() {});
+Promise.reject(new Test262Error())
+  .finally(function() {})
+  .then($DONE, $DONE);
+
+var then = Promise.prototype.then;
 Promise.prototype.then = function(thrower) {
   assert(!isConstructor(thrower));
   assert.sameValue(thrower.length, 1);
   assert.sameValue(thrower.name, '');
   assert.throws(Test262Error, thrower);
 
-  $DONE();
+  return then.call(this, thrower);
 };

test/built-ins/Promise/prototype/finally/resolved-observable-then-calls-argument.js

@@ -19,12 +19,16 @@ flags: [async]
 
 var value = {};
 
-Promise.resolve(value).finally(function() {});
+Promise.resolve(value)
+  .finally(function() {})
+  .then($DONE, $DONE);
+
+var then = Promise.prototype.then;
 Promise.prototype.then = function(valueThunk) {
   assert(!isConstructor(valueThunk));
   assert.sameValue(valueThunk.length, 1);
   assert.sameValue(valueThunk.name, '');
   assert.sameValue(valueThunk(), value);
 
-  $DONE();
+  return then.call(this, valueThunk);
 };