refactor(schematics): update schematics to use the new get token signature (#1969)

Closes #1968

Author: itayod

modules/effects/schematics/ng-add/files/__name@dasherize@if-flat__/__name@dasherize__.effects.spec.ts.template

@@ -16,7 +16,7 @@ describe('<%= classify(name) %>Effects', () => {
       ]
     });
 
-    effects = TestBed.get(<%= classify(name) %>Effects);
+    effects = TestBed.get<<%= classify(name) %>Effects>(<%= classify(name) %>Effects);
   });
 
   it('should be created', () => {

modules/effects/schematics/ng-add/index.spec.ts

@@ -180,4 +180,18 @@ describe('Effect ng-add Schematic', () => {
       files.indexOf(`${projectPath}/src/app/effects/foo.effects.ts`)
     ).toBeGreaterThanOrEqual(0);
   });
+
+  it('should inject the effect service correctly', async () => {
+    const options = { ...defaultOptions, spec: true };
+    const tree = await schematicRunner
+      .runSchematicAsync('ng-add', options, appTree)
+      .toPromise();
+    const content = tree.readContent(
+      `${projectPath}/src/app/foo/foo.effects.spec.ts`
+    );
+
+    expect(content).toMatch(
+      /effects = TestBed\.get<FooEffects>\(FooEffects\);/
+    );
+  });
 });

modules/schematics/src/container/files/__name@dasherize@if-flat__/__name@dasherize__.component.spec.ts.template

@@ -20,7 +20,7 @@ describe('<%= classify(name) %>Component', () => {
   beforeEach(() => {
     fixture = TestBed.createComponent(<%= classify(name) %>Component);
     component = fixture.componentInstance;
-    store = TestBed.get(Store);
+    store = TestBed.get<Store>(Store);
 
     spyOn(store, 'dispatch').and.callThrough();
     fixture.detectChanges();

modules/schematics/src/container/index.spec.ts

@@ -121,4 +121,15 @@ describe('Container Schematic', () => {
       /import { Store, StoreModule } from '@ngrx\/store';/
     );
   });
+
+  it('should inject Store correctly', async () => {
+    const options = { ...defaultOptions, spec: true };
+    const tree = await schematicRunner
+      .runSchematicAsync('container', options, appTree)
+      .toPromise();
+    const content = tree.readContent(
+      `${projectPath}/src/app/foo/foo.component.spec.ts`
+    );
+    expect(content).toMatch(/store = TestBed\.get<Store>\(Store\);/);
+  });
 });

modules/schematics/src/effect/files/__name@dasherize@if-flat__/__name@dasherize__.effects.spec.ts.template

@@ -16,7 +16,7 @@ describe('<%= classify(name) %>Effects', () => {
       ]
     });
 
-    effects = TestBed.get(<%= classify(name) %>Effects);
+    effects = TestBed.get<<%= classify(name) %>Effects>(<%= classify(name) %>Effects);
   });
 
   it('should be created', () => {

modules/schematics/src/effect/index.spec.ts

@@ -370,4 +370,18 @@ describe('Effect Schematic', () => {
       /loadFoos\$ = createEffect\(\(\) => this.actions\$.pipe\(/
     );
   });
+
+  it('should inject the effect service correctly', async () => {
+    const options = { ...defaultOptions, spec: true };
+    const tree = await schematicRunner
+      .runSchematicAsync('effect', options, appTree)
+      .toPromise();
+    const content = tree.readContent(
+      `${projectPath}/src/app/foo/foo.effects.spec.ts`
+    );
+
+    expect(content).toMatch(
+      /effects = TestBed\.get<FooEffects>\(FooEffects\);/
+    );
+  });
 });

projects/ngrx.io/content/examples/testing-store/src/app/auth-guard.service.spec.ts

@@ -15,8 +15,8 @@ describe('Auth Guard', () => {
       providers: [AuthGuard, provideMockStore()],
     });
 
-    store = TestBed.get(Store);
-    guard = TestBed.get(AuthGuard);
+    store = TestBed.get<Store>(Store);
+    guard = TestBed.get<AuthGuard>(AuthGuard);
 
     loggedIn = store.overrideSelector(fromAuth.getLoggedIn, false);
   });
@@ -34,4 +34,4 @@ describe('Auth Guard', () => {
 
     expect(guard.canActivate()).toBeObservable(expected);
   });
-});
+});

projects/ngrx.io/content/guide/effects/testing.md

@@ -34,7 +34,7 @@ describe('My Effects', () => {
       ],
     });
 
-    effects = TestBed.get(MyEffects);
+    effects = TestBed.get<MyEffects>(MyEffects);
   });
 
   it('should work', () => {
@@ -76,7 +76,7 @@ describe('My Effects', () => {
       ],
     });
 
-    effects = TestBed.get(MyEffects);
+    effects = TestBed.get<MyEffects>(MyEffects);
   });
 
   it('should work', () => {
@@ -114,7 +114,7 @@ describe('My Effects', () => {
       ],
     });
 
-    effects = TestBed.get(MyEffects);
+    effects = TestBed.get<MyEffects>(MyEffects);
     metadata = getEffectsMetadata(effects);
   });
 
@@ -258,9 +258,9 @@ describe('CollectionEffects', () => {
       ],
     });
 
-    effects = TestBed.get(CollectionEffects);
-    actions$ = TestBed.get(Actions);
-    store = TestBed.get(Store);
+    effects = TestBed.get<CollectionEffects>(CollectionEffects);
+    actions$ = TestBed.get<Actions>(Actions);
+    store = TestBed.get<Store>(Store);
   });
 
   describe('addBookToCollectionSuccess$', () => {

projects/ngrx.io/content/guide/store/testing.md

@@ -38,8 +38,8 @@ describe('Auth Guard', () => {
       ],
     });
 
-    guard = TestBed.get(AuthGuard);
-    store = TestBed.get(Store);
+    store = TestBed.get<Store&gt(Store);
+    guard = TestBed.get<AuthGuard>(AuthGuard);
   });
 
   it('should return false if the user state is not logged in', () => {
@@ -144,7 +144,7 @@ describe('My Component', () => {
       ],
     });
 
-    store = TestBed.get(Store);
+    store = TestBed.get<Store&gt(Store);
 
     spyOn(store, 'dispatch').and.callThrough();