fix(schematics): remove temp path #11198 (#11424)

* fix(schematics): remove temp path #11198

* chore: remove console

* chore: pr nit

Author: amcdnl

src/lib/schematics/update/update.ts

@@ -81,19 +81,15 @@ export default function(): Rule {
     // Delete the temporary schematics directory.
     context.addTask(
         new RunSchematicTask('ng-post-update', {
-          deleteFiles: updateSrcs
-              .map(entry => entry.path.replace(schematicsSrcPath, schematicsTmpPath))
+          deletePath: schematicsTmpPath
         }), [upgradeTask]);
   };
 }
 
 /** Post-update schematic to be called when ng update is finished. */
-export function postUpdate(options: {deleteFiles: string[]}): Rule {
+export function postUpdate(options: {deletePath: string}): Rule {
   return (tree: Tree, context: SchematicContext) => {
-    for (let file of options.deleteFiles) {
-      tree.delete(file);
-    }
-
+    tree.delete(options.deletePath);
     context.addTask(new RunSchematicTask('ng-post-post-update', {}));
   };
 }

src/lib/schematics/update/update_spec.ts

@@ -0,0 +1,22 @@
+import {SchematicTestRunner} from '@angular-devkit/schematics/testing';
+import {join} from 'path';
+import {Tree} from '@angular-devkit/schematics';
+import {createTestApp} from '../utils/testing';
+import {getFileContent} from '@schematics/angular/utility/test';
+
+const collectionPath = join(__dirname, '../collection.json');
+
+describe('material-nav-schematic', () => {
+  let runner: SchematicTestRunner;
+
+  beforeEach(() => {
+    runner = new SchematicTestRunner('schematics', collectionPath);
+  });
+
+  it('should remove the temp directory', () => {
+    const tree = runner.runSchematic('update', {}, createTestApp());
+    const files = tree.files;
+    expect(files.find(file => file.includes('angular_material_schematics-'))).toBeTruthy();
+  });
+
+});