+++ /dev/null
-import { Observable } from '../Observable';
-import { Notification } from '../Notification';
-import { ColdObservable } from './ColdObservable';
-import { HotObservable } from './HotObservable';
-import { SubscriptionLog } from './SubscriptionLog';
-import { VirtualTimeScheduler, VirtualAction } from '../scheduler/VirtualTimeScheduler';
-import { AsyncScheduler } from '../scheduler/AsyncScheduler';
-const defaultMaxFrame = 750;
-export class TestScheduler extends VirtualTimeScheduler {
- constructor(assertDeepEqual) {
- super(VirtualAction, defaultMaxFrame);
- this.assertDeepEqual = assertDeepEqual;
- this.hotObservables = [];
- this.coldObservables = [];
- this.flushTests = [];
- this.runMode = false;
- }
- createTime(marbles) {
- const indexOf = marbles.indexOf('|');
- if (indexOf === -1) {
- throw new Error('marble diagram for time should have a completion marker "|"');
- }
- return indexOf * TestScheduler.frameTimeFactor;
- }
- createColdObservable(marbles, values, error) {
- if (marbles.indexOf('^') !== -1) {
- throw new Error('cold observable cannot have subscription offset "^"');
- }
- if (marbles.indexOf('!') !== -1) {
- throw new Error('cold observable cannot have unsubscription marker "!"');
- }
- const messages = TestScheduler.parseMarbles(marbles, values, error, undefined, this.runMode);
- const cold = new ColdObservable(messages, this);
- this.coldObservables.push(cold);
- return cold;
- }
- createHotObservable(marbles, values, error) {
- if (marbles.indexOf('!') !== -1) {
- throw new Error('hot observable cannot have unsubscription marker "!"');
- }
- const messages = TestScheduler.parseMarbles(marbles, values, error, undefined, this.runMode);
- const subject = new HotObservable(messages, this);
- this.hotObservables.push(subject);
- return subject;
- }
- materializeInnerObservable(observable, outerFrame) {
- const messages = [];
- observable.subscribe((value) => {
- messages.push({ frame: this.frame - outerFrame, notification: Notification.createNext(value) });
- }, (err) => {
- messages.push({ frame: this.frame - outerFrame, notification: Notification.createError(err) });
- }, () => {
- messages.push({ frame: this.frame - outerFrame, notification: Notification.createComplete() });
- });
- return messages;
- }
- expectObservable(observable, subscriptionMarbles = null) {
- const actual = [];
- const flushTest = { actual, ready: false };
- const subscriptionParsed = TestScheduler.parseMarblesAsSubscriptions(subscriptionMarbles, this.runMode);
- const subscriptionFrame = subscriptionParsed.subscribedFrame === Number.POSITIVE_INFINITY ?
- 0 : subscriptionParsed.subscribedFrame;
- const unsubscriptionFrame = subscriptionParsed.unsubscribedFrame;
- let subscription;
- this.schedule(() => {
- subscription = observable.subscribe(x => {
- let value = x;
- if (x instanceof Observable) {
- value = this.materializeInnerObservable(value, this.frame);
- }
- actual.push({ frame: this.frame, notification: Notification.createNext(value) });
- }, (err) => {
- actual.push({ frame: this.frame, notification: Notification.createError(err) });
- }, () => {
- actual.push({ frame: this.frame, notification: Notification.createComplete() });
- });
- }, subscriptionFrame);
- if (unsubscriptionFrame !== Number.POSITIVE_INFINITY) {
- this.schedule(() => subscription.unsubscribe(), unsubscriptionFrame);
- }
- this.flushTests.push(flushTest);
- const { runMode } = this;
- return {
- toBe(marbles, values, errorValue) {
- flushTest.ready = true;
- flushTest.expected = TestScheduler.parseMarbles(marbles, values, errorValue, true, runMode);
- }
- };
- }
- expectSubscriptions(actualSubscriptionLogs) {
- const flushTest = { actual: actualSubscriptionLogs, ready: false };
- this.flushTests.push(flushTest);
- const { runMode } = this;
- return {
- toBe(marbles) {
- const marblesArray = (typeof marbles === 'string') ? [marbles] : marbles;
- flushTest.ready = true;
- flushTest.expected = marblesArray.map(marbles => TestScheduler.parseMarblesAsSubscriptions(marbles, runMode));
- }
- };
- }
- flush() {
- const hotObservables = this.hotObservables;
- while (hotObservables.length > 0) {
- hotObservables.shift().setup();
- }
- super.flush();
- this.flushTests = this.flushTests.filter(test => {
- if (test.ready) {
- this.assertDeepEqual(test.actual, test.expected);
- return false;
- }
- return true;
- });
- }
- static parseMarblesAsSubscriptions(marbles, runMode = false) {
- if (typeof marbles !== 'string') {
- return new SubscriptionLog(Number.POSITIVE_INFINITY);
- }
- const len = marbles.length;
- let groupStart = -1;
- let subscriptionFrame = Number.POSITIVE_INFINITY;
- let unsubscriptionFrame = Number.POSITIVE_INFINITY;
- let frame = 0;
- for (let i = 0; i < len; i++) {
- let nextFrame = frame;
- const advanceFrameBy = (count) => {
- nextFrame += count * this.frameTimeFactor;
- };
- const c = marbles[i];
- switch (c) {
- case ' ':
- if (!runMode) {
- advanceFrameBy(1);
- }
- break;
- case '-':
- advanceFrameBy(1);
- break;
- case '(':
- groupStart = frame;
- advanceFrameBy(1);
- break;
- case ')':
- groupStart = -1;
- advanceFrameBy(1);
- break;
- case '^':
- if (subscriptionFrame !== Number.POSITIVE_INFINITY) {
- throw new Error('found a second subscription point \'^\' in a ' +
- 'subscription marble diagram. There can only be one.');
- }
- subscriptionFrame = groupStart > -1 ? groupStart : frame;
- advanceFrameBy(1);
- break;
- case '!':
- if (unsubscriptionFrame !== Number.POSITIVE_INFINITY) {
- throw new Error('found a second subscription point \'^\' in a ' +
- 'subscription marble diagram. There can only be one.');
- }
- unsubscriptionFrame = groupStart > -1 ? groupStart : frame;
- break;
- default:
- if (runMode && c.match(/^[0-9]$/)) {
- if (i === 0 || marbles[i - 1] === ' ') {
- const buffer = marbles.slice(i);
- const match = buffer.match(/^([0-9]+(?:\.[0-9]+)?)(ms|s|m) /);
- if (match) {
- i += match[0].length - 1;
- const duration = parseFloat(match[1]);
- const unit = match[2];
- let durationInMs;
- switch (unit) {
- case 'ms':
- durationInMs = duration;
- break;
- case 's':
- durationInMs = duration * 1000;
- break;
- case 'm':
- durationInMs = duration * 1000 * 60;
- break;
- default:
- break;
- }
- advanceFrameBy(durationInMs / this.frameTimeFactor);
- break;
- }
- }
- }
- throw new Error('there can only be \'^\' and \'!\' markers in a ' +
- 'subscription marble diagram. Found instead \'' + c + '\'.');
- }
- frame = nextFrame;
- }
- if (unsubscriptionFrame < 0) {
- return new SubscriptionLog(subscriptionFrame);
- }
- else {
- return new SubscriptionLog(subscriptionFrame, unsubscriptionFrame);
- }
- }
- static parseMarbles(marbles, values, errorValue, materializeInnerObservables = false, runMode = false) {
- if (marbles.indexOf('!') !== -1) {
- throw new Error('conventional marble diagrams cannot have the ' +
- 'unsubscription marker "!"');
- }
- const len = marbles.length;
- const testMessages = [];
- const subIndex = runMode ? marbles.replace(/^[ ]+/, '').indexOf('^') : marbles.indexOf('^');
- let frame = subIndex === -1 ? 0 : (subIndex * -this.frameTimeFactor);
- const getValue = typeof values !== 'object' ?
- (x) => x :
- (x) => {
- if (materializeInnerObservables && values[x] instanceof ColdObservable) {
- return values[x].messages;
- }
- return values[x];
- };
- let groupStart = -1;
- for (let i = 0; i < len; i++) {
- let nextFrame = frame;
- const advanceFrameBy = (count) => {
- nextFrame += count * this.frameTimeFactor;
- };
- let notification;
- const c = marbles[i];
- switch (c) {
- case ' ':
- if (!runMode) {
- advanceFrameBy(1);
- }
- break;
- case '-':
- advanceFrameBy(1);
- break;
- case '(':
- groupStart = frame;
- advanceFrameBy(1);
- break;
- case ')':
- groupStart = -1;
- advanceFrameBy(1);
- break;
- case '|':
- notification = Notification.createComplete();
- advanceFrameBy(1);
- break;
- case '^':
- advanceFrameBy(1);
- break;
- case '#':
- notification = Notification.createError(errorValue || 'error');
- advanceFrameBy(1);
- break;
- default:
- if (runMode && c.match(/^[0-9]$/)) {
- if (i === 0 || marbles[i - 1] === ' ') {
- const buffer = marbles.slice(i);
- const match = buffer.match(/^([0-9]+(?:\.[0-9]+)?)(ms|s|m) /);
- if (match) {
- i += match[0].length - 1;
- const duration = parseFloat(match[1]);
- const unit = match[2];
- let durationInMs;
- switch (unit) {
- case 'ms':
- durationInMs = duration;
- break;
- case 's':
- durationInMs = duration * 1000;
- break;
- case 'm':
- durationInMs = duration * 1000 * 60;
- break;
- default:
- break;
- }
- advanceFrameBy(durationInMs / this.frameTimeFactor);
- break;
- }
- }
- }
- notification = Notification.createNext(getValue(c));
- advanceFrameBy(1);
- break;
- }
- if (notification) {
- testMessages.push({ frame: groupStart > -1 ? groupStart : frame, notification });
- }
- frame = nextFrame;
- }
- return testMessages;
- }
- run(callback) {
- const prevFrameTimeFactor = TestScheduler.frameTimeFactor;
- const prevMaxFrames = this.maxFrames;
- TestScheduler.frameTimeFactor = 1;
- this.maxFrames = Number.POSITIVE_INFINITY;
- this.runMode = true;
- AsyncScheduler.delegate = this;
- const helpers = {
- cold: this.createColdObservable.bind(this),
- hot: this.createHotObservable.bind(this),
- flush: this.flush.bind(this),
- expectObservable: this.expectObservable.bind(this),
- expectSubscriptions: this.expectSubscriptions.bind(this),
- };
- try {
- const ret = callback(helpers);
- this.flush();
- return ret;
- }
- finally {
- TestScheduler.frameTimeFactor = prevFrameTimeFactor;
- this.maxFrames = prevMaxFrames;
- this.runMode = false;
- AsyncScheduler.delegate = undefined;
- }
- }
-}
-//# sourceMappingURL=TestScheduler.js.map
\ No newline at end of file
projects / dotfiles / .git / blobdiff