fix: module exports is now all declarations

help-me-mom · Feb 2, 2018 · fbb0e73 · fbb0e73
1 parent 610cbdc
commit fbb0e73
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/lib/mock-module.ts b/lib/mock-module.ts
@@ -18,28 +18,27 @@ const mockProvider = (provider: any) => ({
   provide: provider, useValue: {}
 });
 
-class MockedModule {}
-
 export function MockModule<TModule>(module: Type<TModule>): any {
-  return NgModule(MockIt(module))(MockedModule);
+  return NgModule(MockIt(module))(class MockedModule {});
 }
 
 function MockIt(module: any): any {
   const mockedModule = { declarations: [] as any[],
                          exports: [] as any[],
                          providers: [] as any[] };
   const declarations = (module as any).__annotations__[0].declarations || [];
-  const exports = (module as any).__annotations__[0].exports || [];
   const imports = (module as any).__annotations__[0].imports || [];
   const providers = (module as any).__annotations__[0].providers || [];
-  mockedModule.declarations = [...declarations.map(mockDeclaration)];
-  mockedModule.exports = exports;
+
+  mockedModule.exports = mockedModule.declarations = [...declarations.map(mockDeclaration)];
   mockedModule.providers = providers.map(mockProvider);
-  imports.reduce((acc: any, im: any) => {
-    const result = MockIt(im);
+
+  imports.reduce((acc: any, imPort: any) => {
+    const result = MockIt(imPort);
     acc.declarations.push(...result.declarations);
     acc.providers.push(...result.providers);
     acc.exports.push(...result.declarations);
   }, mockedModule);
+
   return mockedModule;
 }