123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208 |
- 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';
- const defaultMaxFrame = 750;
- export class TestScheduler extends VirtualTimeScheduler {
- constructor(assertDeepEqual) {
- super(VirtualAction, defaultMaxFrame);
- this.assertDeepEqual = assertDeepEqual;
- this.hotObservables = [];
- this.coldObservables = [];
- this.flushTests = [];
- }
- 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);
- 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);
- 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, unsubscriptionMarbles = null) {
- const actual = [];
- const flushTest = { actual, ready: false };
- const unsubscriptionFrame = TestScheduler
- .parseMarblesAsSubscriptions(unsubscriptionMarbles).unsubscribedFrame;
- let subscription;
- this.schedule(() => {
- subscription = observable.subscribe(x => {
- let value = x;
- // Support Observable-of-Observables
- 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() });
- });
- }, 0);
- if (unsubscriptionFrame !== Number.POSITIVE_INFINITY) {
- this.schedule(() => subscription.unsubscribe(), unsubscriptionFrame);
- }
- this.flushTests.push(flushTest);
- return {
- toBe(marbles, values, errorValue) {
- flushTest.ready = true;
- flushTest.expected = TestScheduler.parseMarbles(marbles, values, errorValue, true);
- }
- };
- }
- expectSubscriptions(actualSubscriptionLogs) {
- const flushTest = { actual: actualSubscriptionLogs, ready: false };
- this.flushTests.push(flushTest);
- return {
- toBe(marbles) {
- const marblesArray = (typeof marbles === 'string') ? [marbles] : marbles;
- flushTest.ready = true;
- flushTest.expected = marblesArray.map(marbles => TestScheduler.parseMarblesAsSubscriptions(marbles));
- }
- };
- }
- flush() {
- const hotObservables = this.hotObservables;
- while (hotObservables.length > 0) {
- hotObservables.shift().setup();
- }
- super.flush();
- const readyFlushTests = this.flushTests.filter(test => test.ready);
- while (readyFlushTests.length > 0) {
- const test = readyFlushTests.shift();
- this.assertDeepEqual(test.actual, test.expected);
- }
- }
- static parseMarblesAsSubscriptions(marbles) {
- 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;
- for (let i = 0; i < len; i++) {
- const frame = i * this.frameTimeFactor;
- const c = marbles[i];
- switch (c) {
- case '-':
- case ' ':
- break;
- case '(':
- groupStart = frame;
- break;
- case ')':
- groupStart = -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;
- 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:
- throw new Error('there can only be \'^\' and \'!\' markers in a ' +
- 'subscription marble diagram. Found instead \'' + c + '\'.');
- }
- }
- if (unsubscriptionFrame < 0) {
- return new SubscriptionLog(subscriptionFrame);
- }
- else {
- return new SubscriptionLog(subscriptionFrame, unsubscriptionFrame);
- }
- }
- static parseMarbles(marbles, values, errorValue, materializeInnerObservables = false) {
- if (marbles.indexOf('!') !== -1) {
- throw new Error('conventional marble diagrams cannot have the ' +
- 'unsubscription marker "!"');
- }
- const len = marbles.length;
- const testMessages = [];
- const subIndex = marbles.indexOf('^');
- const frameOffset = subIndex === -1 ? 0 : (subIndex * -this.frameTimeFactor);
- const getValue = typeof values !== 'object' ?
- (x) => x :
- (x) => {
- // Support Observable-of-Observables
- if (materializeInnerObservables && values[x] instanceof ColdObservable) {
- return values[x].messages;
- }
- return values[x];
- };
- let groupStart = -1;
- for (let i = 0; i < len; i++) {
- const frame = i * this.frameTimeFactor + frameOffset;
- let notification;
- const c = marbles[i];
- switch (c) {
- case '-':
- case ' ':
- break;
- case '(':
- groupStart = frame;
- break;
- case ')':
- groupStart = -1;
- break;
- case '|':
- notification = Notification.createComplete();
- break;
- case '^':
- break;
- case '#':
- notification = Notification.createError(errorValue || 'error');
- break;
- default:
- notification = Notification.createNext(getValue(c));
- break;
- }
- if (notification) {
- testMessages.push({ frame: groupStart > -1 ? groupStart : frame, notification });
- }
- }
- return testMessages;
- }
- }
- //# sourceMappingURL=TestScheduler.js.map
|