make DelayedFunctionScheduler update the mockDate

This commit is contained in:
Andre Wiggins
2015-11-20 03:45:57 -08:00
parent b3f69852c1
commit c8b212f278
4 changed files with 53 additions and 6 deletions

View File

@@ -652,4 +652,23 @@ describe("Clock (acceptance)", function() {
expect(timeoutDate).toEqual(baseTime.getTime() + 150);
});
it("mocks the Date object and updates the date per delayed function", function () {
var delayedFunctionScheduler = new j$.DelayedFunctionScheduler(),
global = {Date: Date},
mockDate = new j$.MockDate(global),
clock = new j$.Clock({setTimeout: setTimeout}, function () { return delayedFunctionScheduler; }, mockDate),
baseTime = new Date();
clock.install().mockDate(baseTime);
var actualTimes = [];
var pushCurrentTime = function() { actualTimes.push(global.Date().getTime()); };
delayedFunctionScheduler.scheduleFunction(pushCurrentTime);
delayedFunctionScheduler.scheduleFunction(pushCurrentTime, 1);
clock.tick(1);
expect(actualTimes).toEqual([baseTime.getTime(), baseTime.getTime() + 1]);
})
});

View File

@@ -251,5 +251,23 @@ describe("DelayedFunctionScheduler", function() {
expect(fn).toHaveBeenCalled();
expect(fn.calls.count()).toBe(1);
});
it("updates the mockDate per scheduled time", function () {
var scheduler = new j$.DelayedFunctionScheduler(),
fakeGlobal = { Date: Date },
mockDate = new j$.MockDate(fakeGlobal),
baseTime = new Date();
mockDate.install(baseTime);
var actualTimes = [];
var pushCurrentTime = function() { actualTimes.push(fakeGlobal.Date().getTime()); };
scheduler.scheduleFunction(pushCurrentTime);
scheduler.scheduleFunction(pushCurrentTime, 1);
scheduler.tick(1, mockDate);
expect(actualTimes).toEqual([baseTime.getTime(), baseTime.getTime() + 1]);
})
});

View File

@@ -82,8 +82,7 @@ getJasmineRequireObj().Clock = function() {
self.tick = function(millis) {
if (installed) {
mockDate.tick(millis);
delayedFunctionScheduler.tick(millis);
delayedFunctionScheduler.tick(millis, mockDate);
} else {
throw new Error('Mock clock is not installed, use jasmine.clock().install()');
}

View File

@@ -6,11 +6,11 @@ getJasmineRequireObj().DelayedFunctionScheduler = function() {
var currentTime = 0;
var delayedFnCount = 0;
self.tick = function(millis) {
self.tick = function(millis, mockDate) {
millis = millis || 0;
var endTime = currentTime + millis;
runScheduledFunctions(endTime);
runScheduledFunctions(endTime, mockDate);
currentTime = endTime;
};
@@ -113,13 +113,17 @@ getJasmineRequireObj().DelayedFunctionScheduler = function() {
}
}
function runScheduledFunctions(endTime) {
function runScheduledFunctions(endTime, mockDate) {
if (scheduledLookup.length === 0 || scheduledLookup[0] > endTime) {
updateMockDate(mockDate, endTime);
return;
}
do {
currentTime = scheduledLookup.shift();
var newCurrentTime = scheduledLookup.shift();
updateMockDate(mockDate, newCurrentTime);
currentTime = newCurrentTime;
var funcsToRun = scheduledFunctions[currentTime];
delete scheduledFunctions[currentTime];
@@ -139,6 +143,13 @@ getJasmineRequireObj().DelayedFunctionScheduler = function() {
currentTime !== endTime &&
scheduledLookup[0] <= endTime);
}
function updateMockDate(mockDate, newCurrentTime) {
if (mockDate) {
var millis = newCurrentTime - currentTime;
mockDate.tick(millis);
}
}
}
return DelayedFunctionScheduler;