Missing await on variant transition call

Author: clbond

package.json

@@ -1,6 +1,6 @@
 {
   "name": "angular-ssr",
-  "version": "0.1.52",
+  "version": "0.1.53",
   "description": "Angular server-side rendering implementation",
   "main": "build/index.js",
   "typings": "build/index.d.ts",

source/exception.ts

@@ -3,13 +3,13 @@ import chalk = require('chalk');
 
 export class Exception extends Error {
   constructor(msg: string, public innerException?: Error) {
-    super(innerException ? `${msg} -> ${innerException.stack}` : chalk.red(msg));
+    super(innerException ? `${chalk.red(msg)} -> ${chalk.red(innerException.stack)}` : chalk.red(msg));
  }
 }
 
 export class AggregateException extends Exception {
   constructor(public exceptions: Array<Exception | Error>) {
-    super(chalk.red(`Multiple exceptions occurred (${exceptions.length}) [${exceptions.map(e => e.stack).join(', ')}]`));
+    super(`Multiple exceptions occurred (${exceptions.length}) ${chalk.red(`[${exceptions.map(e => e.stack).join(', ')}]`)}`);
   }
 
   get stack(): string {

source/output/interprocess.ts

@@ -3,7 +3,7 @@ import {OutputProducer} from './producer';
 import {Snapshot} from '../snapshot';
 
 export class InterprocessOutput implements OutputProducer {
-  initialize(): Promise<void> {
+  initialize() {
     throw new NotImplementedException();
   }
 

source/output/producer.ts

@@ -1,7 +1,7 @@
 import {Snapshot} from '../snapshot';
 
 export interface OutputProducer {
-  initialize(): Promise<void>;
+  initialize(): void;
 
   write<V>(snapshot: Snapshot<V>): Promise<void>;
 

source/snapshot/creator/bootstrap.ts

@@ -16,7 +16,7 @@ export const executeBootstrap = async <M>(moduleRef: NgModuleRef<M>, bootstrappe
   await bootstrap(moduleRef.injector);
 
   if (typeof transition === 'function') {
-    transition(moduleRef.injector);
+    await Promise.resolve(transition(moduleRef.injector));
   }
 };