fix(Schematics): Add store import to container blueprint

docs/schematics/container.md

@@ -25,7 +25,7 @@ ng generate co ComponentName [options]
 
 Provide the path to your file with an exported state interface
 
-- `--reducers`
+- `--state`
  - Type: `string`
 
 Provide the name of the interface exported for your state interface
@@ -39,5 +39,5 @@ Provide the name of the interface exported for your state interface
 Generate a `UsersPage` container component with your reducers imported and the `Store` typed a custom interface named `MyState`.
 
 ```sh
-ng generate container UsersPage --reducers reducers/index.ts --stateInterface MyState
+ng generate container UsersPage --state reducers/index.ts --stateInterface MyState
 ```

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

@@ -46,6 +46,15 @@ describe('Container Schematic', () => {
  expect(content).toMatch(/import \* as fromStore from '..\/reducers';/);
  });
 
+ it('should import Store into the component', () => {
+ const options = { ...defaultOptions, state: 'reducers' };
+ appTree.create('/src/app/reducers', '');
+ const tree = schematicRunner.runSchematic('container', options, appTree);
+ const content = getFileContent(tree, '/src/app/foo/foo.component.ts');
+ console.log(content);
+ expect(content).toMatch(/import\ {\ Store\ }\ from\ '@ngrx\/store';/);
+ });
+
  it('should update the component constructor if the state path if provided', () => {
  const options = { ...defaultOptions, state: 'reducers' };
  appTree.create('/src/app/reducers', '');

modules/schematics/src/container/index.ts

@@ -61,6 +61,12 @@ function addStateToComponent(options: FeatureOptions) {
  );
 
  const stateImportPath = buildRelativePath(componentPath, statePath);
+ const storeImport = insertImport(
+ source,
+ componentPath,
+ 'Store',
+ '@ngrx/store'
+ );
  const stateImport = options.state
  ? insertImport(
  source,
@@ -94,7 +100,7 @@ function addStateToComponent(options: FeatureOptions) {
  `\n\n ${storeConstructor}`
  );
 
- const changes = [stateImport, constructorUpdate];
+ const changes = [storeImport, stateImport, constructorUpdate];
  const recorder = host.beginUpdate(componentPath);
 
  for (const change of changes) {