2 var __extends = (this && this.__extends) || (function () {
3 var extendStatics = function (d, b) {
4 extendStatics = Object.setPrototypeOf ||
5 ({ __proto__: [] } instanceof Array && function (d, b) { d.__proto__ = b; }) ||
6 function (d, b) { for (var p in b) if (b.hasOwnProperty(p)) d[p] = b[p]; };
7 return extendStatics(d, b);
9 return function (d, b) {
11 function __() { this.constructor = d; }
12 d.prototype = b === null ? Object.create(b) : (__.prototype = b.prototype, new __());
15 Object.defineProperty(exports, "__esModule", { value: true });
16 var Observable_1 = require("../Observable");
17 var Notification_1 = require("../Notification");
18 var ColdObservable_1 = require("./ColdObservable");
19 var HotObservable_1 = require("./HotObservable");
20 var SubscriptionLog_1 = require("./SubscriptionLog");
21 var VirtualTimeScheduler_1 = require("../scheduler/VirtualTimeScheduler");
22 var AsyncScheduler_1 = require("../scheduler/AsyncScheduler");
23 var defaultMaxFrame = 750;
24 var TestScheduler = (function (_super) {
25 __extends(TestScheduler, _super);
26 function TestScheduler(assertDeepEqual) {
27 var _this = _super.call(this, VirtualTimeScheduler_1.VirtualAction, defaultMaxFrame) || this;
28 _this.assertDeepEqual = assertDeepEqual;
29 _this.hotObservables = [];
30 _this.coldObservables = [];
31 _this.flushTests = [];
32 _this.runMode = false;
35 TestScheduler.prototype.createTime = function (marbles) {
36 var indexOf = marbles.indexOf('|');
38 throw new Error('marble diagram for time should have a completion marker "|"');
40 return indexOf * TestScheduler.frameTimeFactor;
42 TestScheduler.prototype.createColdObservable = function (marbles, values, error) {
43 if (marbles.indexOf('^') !== -1) {
44 throw new Error('cold observable cannot have subscription offset "^"');
46 if (marbles.indexOf('!') !== -1) {
47 throw new Error('cold observable cannot have unsubscription marker "!"');
49 var messages = TestScheduler.parseMarbles(marbles, values, error, undefined, this.runMode);
50 var cold = new ColdObservable_1.ColdObservable(messages, this);
51 this.coldObservables.push(cold);
54 TestScheduler.prototype.createHotObservable = function (marbles, values, error) {
55 if (marbles.indexOf('!') !== -1) {
56 throw new Error('hot observable cannot have unsubscription marker "!"');
58 var messages = TestScheduler.parseMarbles(marbles, values, error, undefined, this.runMode);
59 var subject = new HotObservable_1.HotObservable(messages, this);
60 this.hotObservables.push(subject);
63 TestScheduler.prototype.materializeInnerObservable = function (observable, outerFrame) {
66 observable.subscribe(function (value) {
67 messages.push({ frame: _this.frame - outerFrame, notification: Notification_1.Notification.createNext(value) });
69 messages.push({ frame: _this.frame - outerFrame, notification: Notification_1.Notification.createError(err) });
71 messages.push({ frame: _this.frame - outerFrame, notification: Notification_1.Notification.createComplete() });
75 TestScheduler.prototype.expectObservable = function (observable, subscriptionMarbles) {
77 if (subscriptionMarbles === void 0) { subscriptionMarbles = null; }
79 var flushTest = { actual: actual, ready: false };
80 var subscriptionParsed = TestScheduler.parseMarblesAsSubscriptions(subscriptionMarbles, this.runMode);
81 var subscriptionFrame = subscriptionParsed.subscribedFrame === Number.POSITIVE_INFINITY ?
82 0 : subscriptionParsed.subscribedFrame;
83 var unsubscriptionFrame = subscriptionParsed.unsubscribedFrame;
85 this.schedule(function () {
86 subscription = observable.subscribe(function (x) {
88 if (x instanceof Observable_1.Observable) {
89 value = _this.materializeInnerObservable(value, _this.frame);
91 actual.push({ frame: _this.frame, notification: Notification_1.Notification.createNext(value) });
93 actual.push({ frame: _this.frame, notification: Notification_1.Notification.createError(err) });
95 actual.push({ frame: _this.frame, notification: Notification_1.Notification.createComplete() });
97 }, subscriptionFrame);
98 if (unsubscriptionFrame !== Number.POSITIVE_INFINITY) {
99 this.schedule(function () { return subscription.unsubscribe(); }, unsubscriptionFrame);
101 this.flushTests.push(flushTest);
102 var runMode = this.runMode;
104 toBe: function (marbles, values, errorValue) {
105 flushTest.ready = true;
106 flushTest.expected = TestScheduler.parseMarbles(marbles, values, errorValue, true, runMode);
110 TestScheduler.prototype.expectSubscriptions = function (actualSubscriptionLogs) {
111 var flushTest = { actual: actualSubscriptionLogs, ready: false };
112 this.flushTests.push(flushTest);
113 var runMode = this.runMode;
115 toBe: function (marbles) {
116 var marblesArray = (typeof marbles === 'string') ? [marbles] : marbles;
117 flushTest.ready = true;
118 flushTest.expected = marblesArray.map(function (marbles) {
119 return TestScheduler.parseMarblesAsSubscriptions(marbles, runMode);
124 TestScheduler.prototype.flush = function () {
126 var hotObservables = this.hotObservables;
127 while (hotObservables.length > 0) {
128 hotObservables.shift().setup();
130 _super.prototype.flush.call(this);
131 this.flushTests = this.flushTests.filter(function (test) {
133 _this.assertDeepEqual(test.actual, test.expected);
139 TestScheduler.parseMarblesAsSubscriptions = function (marbles, runMode) {
141 if (runMode === void 0) { runMode = false; }
142 if (typeof marbles !== 'string') {
143 return new SubscriptionLog_1.SubscriptionLog(Number.POSITIVE_INFINITY);
145 var len = marbles.length;
147 var subscriptionFrame = Number.POSITIVE_INFINITY;
148 var unsubscriptionFrame = Number.POSITIVE_INFINITY;
150 var _loop_1 = function (i) {
151 var nextFrame = frame;
152 var advanceFrameBy = function (count) {
153 nextFrame += count * _this.frameTimeFactor;
174 if (subscriptionFrame !== Number.POSITIVE_INFINITY) {
175 throw new Error('found a second subscription point \'^\' in a ' +
176 'subscription marble diagram. There can only be one.');
178 subscriptionFrame = groupStart > -1 ? groupStart : frame;
182 if (unsubscriptionFrame !== Number.POSITIVE_INFINITY) {
183 throw new Error('found a second subscription point \'^\' in a ' +
184 'subscription marble diagram. There can only be one.');
186 unsubscriptionFrame = groupStart > -1 ? groupStart : frame;
189 if (runMode && c.match(/^[0-9]$/)) {
190 if (i === 0 || marbles[i - 1] === ' ') {
191 var buffer = marbles.slice(i);
192 var match = buffer.match(/^([0-9]+(?:\.[0-9]+)?)(ms|s|m) /);
194 i += match[0].length - 1;
195 var duration = parseFloat(match[1]);
197 var durationInMs = void 0;
200 durationInMs = duration;
203 durationInMs = duration * 1000;
206 durationInMs = duration * 1000 * 60;
211 advanceFrameBy(durationInMs / this_1.frameTimeFactor);
216 throw new Error('there can only be \'^\' and \'!\' markers in a ' +
217 'subscription marble diagram. Found instead \'' + c + '\'.');
222 var this_1 = this, out_i_1;
223 for (var i = 0; i < len; i++) {
227 if (unsubscriptionFrame < 0) {
228 return new SubscriptionLog_1.SubscriptionLog(subscriptionFrame);
231 return new SubscriptionLog_1.SubscriptionLog(subscriptionFrame, unsubscriptionFrame);
234 TestScheduler.parseMarbles = function (marbles, values, errorValue, materializeInnerObservables, runMode) {
236 if (materializeInnerObservables === void 0) { materializeInnerObservables = false; }
237 if (runMode === void 0) { runMode = false; }
238 if (marbles.indexOf('!') !== -1) {
239 throw new Error('conventional marble diagrams cannot have the ' +
240 'unsubscription marker "!"');
242 var len = marbles.length;
243 var testMessages = [];
244 var subIndex = runMode ? marbles.replace(/^[ ]+/, '').indexOf('^') : marbles.indexOf('^');
245 var frame = subIndex === -1 ? 0 : (subIndex * -this.frameTimeFactor);
246 var getValue = typeof values !== 'object' ?
247 function (x) { return x; } :
249 if (materializeInnerObservables && values[x] instanceof ColdObservable_1.ColdObservable) {
250 return values[x].messages;
255 var _loop_2 = function (i) {
256 var nextFrame = frame;
257 var advanceFrameBy = function (count) {
258 nextFrame += count * _this.frameTimeFactor;
260 var notification = void 0;
280 notification = Notification_1.Notification.createComplete();
287 notification = Notification_1.Notification.createError(errorValue || 'error');
291 if (runMode && c.match(/^[0-9]$/)) {
292 if (i === 0 || marbles[i - 1] === ' ') {
293 var buffer = marbles.slice(i);
294 var match = buffer.match(/^([0-9]+(?:\.[0-9]+)?)(ms|s|m) /);
296 i += match[0].length - 1;
297 var duration = parseFloat(match[1]);
299 var durationInMs = void 0;
302 durationInMs = duration;
305 durationInMs = duration * 1000;
308 durationInMs = duration * 1000 * 60;
313 advanceFrameBy(durationInMs / this_2.frameTimeFactor);
318 notification = Notification_1.Notification.createNext(getValue(c));
323 testMessages.push({ frame: groupStart > -1 ? groupStart : frame, notification: notification });
328 var this_2 = this, out_i_2;
329 for (var i = 0; i < len; i++) {
335 TestScheduler.prototype.run = function (callback) {
336 var prevFrameTimeFactor = TestScheduler.frameTimeFactor;
337 var prevMaxFrames = this.maxFrames;
338 TestScheduler.frameTimeFactor = 1;
339 this.maxFrames = Number.POSITIVE_INFINITY;
341 AsyncScheduler_1.AsyncScheduler.delegate = this;
343 cold: this.createColdObservable.bind(this),
344 hot: this.createHotObservable.bind(this),
345 flush: this.flush.bind(this),
346 expectObservable: this.expectObservable.bind(this),
347 expectSubscriptions: this.expectSubscriptions.bind(this),
350 var ret = callback(helpers);
355 TestScheduler.frameTimeFactor = prevFrameTimeFactor;
356 this.maxFrames = prevMaxFrames;
357 this.runMode = false;
358 AsyncScheduler_1.AsyncScheduler.delegate = undefined;
361 return TestScheduler;
362 }(VirtualTimeScheduler_1.VirtualTimeScheduler));
363 exports.TestScheduler = TestScheduler;
364 //# sourceMappingURL=TestScheduler.js.map