Bug 643187 Add support for generators to mochitests r=robcee

This commit is contained in:
Benjamin Stover 2011-03-29 16:41:29 -07:00
parent 24bc5d4c25
commit 6dc7507c52

View File

@ -242,10 +242,21 @@ Tester.prototype = {
// Run the test
this.lastStartTime = Date.now();
this.currentTest.scope.test();
if ("generatorTest" in this.currentTest.scope) {
if ("test" in this.currentTest.scope)
throw "Cannot run both a generator test and a normal test at the same time.";
// This test is a generator. It will not finish immediately.
this.currentTest.scope.waitForExplicitFinish();
var result = this.currentTest.scope.generatorTest();
this.currentTest.scope.__generator = result;
result.next();
} else {
this.currentTest.scope.test();
}
} catch (ex) {
this.currentTest.addResult(new testResult(false, "Exception thrown", ex, false));
this.currentTest.scope.__done = true;
this.currentTest.scope.finish();
}
// If the test ran synchronously, move to the next test, otherwise the test
@ -354,6 +365,29 @@ function testScope(aTester, aTest) {
}, Ci.nsIThread.DISPATCH_NORMAL);
};
this.nextStep = function test_nextStep(arg) {
if (self.__done) {
aTest.addResult(new testResult(false, "nextStep was called too many times", "", false));
return;
}
if (!self.__generator) {
aTest.addResult(new testResult(false, "nextStep called with no generator", "", false));
self.finish();
return;
}
try {
self.__generator.send(arg);
} catch (ex if ex instanceof StopIteration) {
// StopIteration means test is finished.
self.finish();
} catch (ex) {
aTest.addResult(new testResult(false, "Exception thrown", ex, false));
self.finish();
}
};
this.waitForExplicitFinish = function test_waitForExplicitFinish() {
self.__done = false;
};
@ -393,6 +427,7 @@ function testScope(aTester, aTest) {
}
testScope.prototype = {
__done: true,
__generator: null,
__waitTimer: null,
__cleanupFunctions: [],
__timeoutFactor: 1,