--- /dev/null
+import { AsyncAction } from './AsyncAction';
+import { Subscription } from '../Subscription';
+import { AsyncScheduler } from './AsyncScheduler';
+import { SchedulerAction } from '../types';
+
+export class VirtualTimeScheduler extends AsyncScheduler {
+
+ protected static frameTimeFactor: number = 10;
+
+ public frame: number = 0;
+ public index: number = -1;
+
+ constructor(SchedulerAction: typeof AsyncAction = VirtualAction as any,
+ public maxFrames: number = Number.POSITIVE_INFINITY) {
+ super(SchedulerAction, () => this.frame);
+ }
+
+ /**
+ * Prompt the Scheduler to execute all of its queued actions, therefore
+ * clearing its queue.
+ * @return {void}
+ */
+ public flush(): void {
+
+ const {actions, maxFrames} = this;
+ let error: any, action: AsyncAction<any>;
+
+ while ((action = actions[0]) && action.delay <= maxFrames) {
+ actions.shift();
+ this.frame = action.delay;
+
+ if (error = action.execute(action.state, action.delay)) {
+ break;
+ }
+ }
+
+ if (error) {
+ while (action = actions.shift()) {
+ action.unsubscribe();
+ }
+ throw error;
+ }
+ }
+}
+
+/**
+ * We need this JSDoc comment for affecting ESDoc.
+ * @nodoc
+ */
+export class VirtualAction<T> extends AsyncAction<T> {
+
+ protected active: boolean = true;
+
+ constructor(protected scheduler: VirtualTimeScheduler,
+ protected work: (this: SchedulerAction<T>, state?: T) => void,
+ protected index: number = scheduler.index += 1) {
+ super(scheduler, work);
+ this.index = scheduler.index = index;
+ }
+
+ public schedule(state?: T, delay: number = 0): Subscription {
+ if (!this.id) {
+ return super.schedule(state, delay);
+ }
+ this.active = false;
+ // If an action is rescheduled, we save allocations by mutating its state,
+ // pushing it to the end of the scheduler queue, and recycling the action.
+ // But since the VirtualTimeScheduler is used for testing, VirtualActions
+ // must be immutable so they can be inspected later.
+ const action = new VirtualAction(this.scheduler, this.work);
+ this.add(action);
+ return action.schedule(state, delay);
+ }
+
+ protected requestAsyncId(scheduler: VirtualTimeScheduler, id?: any, delay: number = 0): any {
+ this.delay = scheduler.frame + delay;
+ const {actions} = scheduler;
+ actions.push(this);
+ (actions as Array<VirtualAction<T>>).sort(VirtualAction.sortActions);
+ return true;
+ }
+
+ protected recycleAsyncId(scheduler: VirtualTimeScheduler, id?: any, delay: number = 0): any {
+ return undefined;
+ }
+
+ protected _execute(state: T, delay: number): any {
+ if (this.active === true) {
+ return super._execute(state, delay);
+ }
+ }
+
+ public static sortActions<T>(a: VirtualAction<T>, b: VirtualAction<T>) {
+ if (a.delay === b.delay) {
+ if (a.index === b.index) {
+ return 0;
+ } else if (a.index > b.index) {
+ return 1;
+ } else {
+ return -1;
+ }
+ } else if (a.delay > b.delay) {
+ return 1;
+ } else {
+ return -1;
+ }
+ }
+}