Merge pull request #2 from orchestratora/fix-module-config

[Fix] Accept schemas from ngModule config

Author: gund

.vscode/settings.json

@@ -1,3 +1,8 @@
 {
-  "typescript.tsdk": "node_modules\\typescript\\lib"
+  "typescript.tsdk": "node_modules\\typescript\\lib",
+  "workbench.colorCustomizations": {
+    "activityBar.background": "#1D2F40",
+    "titleBar.activeBackground": "#29425A",
+    "titleBar.activeForeground": "#F8FAFC"
+  }
 }

projects/ngx-testing/src/lib/host-generator.service.spec.ts

@@ -1,7 +1,10 @@
 import { ComponentFactoryResolver } from '@angular/core';
+import * as ngCore from '@angular/core';
 import { TestBed } from '@angular/core/testing';
 
+import { spyOnModule } from '../../test/spy-on-module';
 import { HostGeneratorService } from './host-generator.service';
+import * as tplGen from './template-gen';
 import {
   ExtraConfigToken,
   TestModuleToken,
@@ -10,7 +13,6 @@ import {
 } from './tokens';
 import { TestTypeKind } from './types';
 import * as util from './util';
-import * as tplGen from './template-gen';
 
 class TestModuleTokenMock {}
 class TestTypeTokenMock {}
@@ -33,11 +35,74 @@ describe('Service: HostGenerator', () => {
   });
 
   describe('generateModuleFor() method', () => {
+    let ngModule: jasmine.Spy;
+
+    beforeEach(() => {
+      getService();
+      ngModule = spyOnModule(ngCore, 'NgModule');
+    });
+
     it('should return new type', () => {
       expect(getService().generateModuleFor(class {})).toEqual(
         jasmine.any(Function),
       );
     });
+
+    it('should import `this.testModule`', () => {
+      getService().generateModuleFor(class {});
+
+      expect(ngModule).toHaveBeenCalledWith(
+        jasmine.objectContaining({
+          imports: [TestModuleTokenMock],
+        }),
+      );
+    });
+
+    it('should declare `host`', () => {
+      const host = class {};
+      getService().generateModuleFor(host);
+
+      expect(ngModule).toHaveBeenCalledWith(
+        jasmine.objectContaining({
+          declarations: [host],
+        }),
+      );
+    });
+
+    it('should export `host`', () => {
+      const host = class {};
+      getService().generateModuleFor(host);
+
+      expect(ngModule).toHaveBeenCalledWith(
+        jasmine.objectContaining({
+          exports: [host],
+        }),
+      );
+    });
+
+    it('should set `host` as entryComponents', () => {
+      const host = class {};
+      getService().generateModuleFor(host);
+
+      expect(ngModule).toHaveBeenCalledWith(
+        jasmine.objectContaining({
+          entryComponents: [host],
+        }),
+      );
+    });
+
+    it('should set schemas from `this.extraConfig.ngModule.schemas`', () => {
+      const extraConfig = TestBed.get(ExtraConfigToken);
+      extraConfig.ngModule = { schemas: 'schema' };
+
+      getService().generateModuleFor(class {});
+
+      expect(ngModule).toHaveBeenCalledWith(
+        jasmine.objectContaining({
+          schemas: 'schema',
+        }),
+      );
+    });
   });
 
   describe('generate() method', () => {

projects/ngx-testing/src/lib/host-generator.service.ts

@@ -43,13 +43,17 @@ export class HostGeneratorService {
 
   generateModuleFor(host: Type<any>): Type<any> {
     const testModule = this.testModule;
+    const ngModule = this.extraConfig.ngModule;
+
     @NgModule({
       imports: [testModule],
       declarations: [host],
       exports: [host],
       entryComponents: [host],
+      schemas: ngModule ? ngModule.schemas : [],
     })
     class TestModule {}
+
     return TestModule;
   }
 

projects/ngx-testing/src/lib/ngx-testing.module.ts

@@ -68,6 +68,7 @@ function getTestingModuleFor<T>(
     entryComponents: entryType
       ? mergeArrays(extra.entryComponents, [entryType])
       : extra.entryComponents,
+    schemas: extra.schemas,
   })
   class TestingModule<D> {}
   return TestingModule as Type<TestingModule<T>>;

projects/ngx-testing/test/spy-on-module.ts

@@ -0,0 +1,8 @@
+export function spyOnModule<T extends object>(
+  moduleObj: T,
+  exportSymbolName: keyof T,
+): jasmine.Spy {
+  const spy = jasmine.createSpy(`Spy of ${exportSymbolName}`);
+  spyOnProperty(moduleObj, exportSymbolName).and.returnValue(spy);
+  return spy;
+}