fix(@schematics/angular): Add budget configuration to update logic

fixes #10616

Author: Brocco

packages/schematics/angular/migrations/update-6/index.ts

@@ -6,6 +6,7 @@
  * found in the LICENSE file at https://angular.io/license
  */
 import {
+ JsonArray,
  JsonObject,
  JsonParseMode,
  Path,
@@ -327,6 +328,7 @@ function extractProjectsConfig(config: CliConfig, tree: Tree): JsonObject {
  : {}
  ),
  ...(isProduction && swConfig ? swConfig : {}),
+ ...(isProduction && app.budgets ? { budgets: app.budgets as JsonArray } : {}),
  fileReplacements: [
  {
  replace: `${app.root}/${source}`,

packages/schematics/angular/migrations/update-6/index_spec.ts

@@ -602,6 +602,23 @@ describe('Migration to v6', () => {
  .toEqual(['src/tsconfig.app.json', 'src/tsconfig.spec.json']);
  expect(tslint.options.exclude).toEqual([ '**/node_modules/**' ]);
  });
+
+ it('should set the budgets configuration', () => {
+ baseConfig.apps[0].budgets = [{
+ type: 'bundle',
+ name: 'main',
+ error: '123kb',
+ }];
+
+ tree.create(oldConfigPath, JSON.stringify(baseConfig, null, 2));
+ tree = schematicRunner.runSchematic('migration-01', defaultOptions, tree);
+ const config = getConfig(tree);
+ const budgets = config.projects.foo.architect.build.configurations.production.budgets;
+ expect(budgets.length).toEqual(1);
+ expect(budgets[0].type).toEqual('bundle');
+ expect(budgets[0].name).toEqual('main');
+ expect(budgets[0].error).toEqual('123kb');
+ });
  });
 
  describe('e2e projects', () => {

packages/schematics/angular/utility/config.ts

@@ -126,6 +126,44 @@ export interface AppConfig {
  app: string;
  route: string;
  };
+ budgets?: {
+ /**
+ * The type of budget
+ */
+ type?: ('bundle' | 'initial' | 'allScript' | 'all' | 'anyScript' | 'any');
+ /**
+ * The name of the bundle
+ */
+ name?: string;
+ /**
+ * The baseline size for comparison.
+ */
+ baseline?: string;
+ /**
+ * The maximum threshold for warning relative to the baseline.
+ */
+ maximumWarning?: string;
+ /**
+ * The maximum threshold for error relative to the baseline.
+ */
+ maximumError?: string;
+ /**
+ * The minimum threshold for warning relative to the baseline.
+ */
+ minimumWarning?: string;
+ /**
+ * The minimum threshold for error relative to the baseline.
+ */
+ minimumError?: string;
+ /**
+ * The threshold for warning relative to the baseline (min & max).
+ */
+ warning?: string;
+ /**
+ * The threshold for error relative to the baseline (min & max).
+ */
+ error?: string;
+ }[];
 }
 
 export interface CliConfig {