Support overrides in the version solver.

sdk/lib/_internal/pub/test/version_solver_test.dart

 // Copyright (c) 2012, the Dart project authors.  Please see the AUTHORS file
 // for details. All rights reserved. Use of this source code is governed by a
 // BSD-style license that can be found in the LICENSE file.
 
 library pub_upgrade_test;
 
 import 'dart:async';
 
 import 'package:unittest/unittest.dart';
 
@@ skipping 23 matching lines @@
 
   group('basic graph', basicGraph);
   group('with lockfile', withLockFile);
   group('root dependency', rootDependency);
   group('dev dependency', devDependency);
   group('unsolvable', unsolvable);
   group('bad source', badSource);
   group('backtracking', backtracking);
   group('SDK constraint', sdkConstraint);
   group('pre-release', prerelease);
+  group('override', override);
 }
 
 void basicGraph() {
   testResolve('no dependencies', {
     'myapp 0.0.0': {}
   }, result: {
     'myapp from root': '0.0.0'
   });
 
   testResolve('simple dependency tree', {
@@ skipping 794 matching lines @@
     'a 1.0.0': {},
     'a 1.1.0': {},
     'a 2.0.0-dev': {},
     'a 2.0.0': {}
   }, result: {
     'myapp from root': '0.0.0',
     'a': '1.1.0'
   });
 }
 
-testResolve(description, packages, {
-             lockfile, result, FailMatcherBuilder error, int maxTries,
-             bool useBleedingEdgeSdkVersion}) {
-  _testResolve(test, description, packages, lockfile: lockfile, result: result,
-      error: error, maxTries: maxTries,
+void override() {
+  testResolve('chooses best version matching override constraint', {
+    'myapp 0.0.0': {
+      'a': 'any'
+    },
+    'a 1.0.0': {},
+    'a 2.0.0': {},
+    'a 3.0.0': {}
+  }, overrides: {
+    'a': '<3.0.0'
+  }, result: {
+    'myapp from root': '0.0.0',
+    'a': '2.0.0'
+  });
+
+  testResolve('uses override as dependency', {
+    'myapp 0.0.0': {},
+    'a 1.0.0': {},
+    'a 2.0.0': {},
+    'a 3.0.0': {}
+  }, overrides: {
+    'a': '<3.0.0'
+  }, result: {
+    'myapp from root': '0.0.0',
+    'a': '2.0.0'
+  });
+
+  testResolve('ignores other constraints on overridden package', {
+    'myapp 0.0.0': {
+      'b': 'any',
+      'c': 'any'
+    },
+    'a 1.0.0': {},
+    'a 2.0.0': {},
+    'a 3.0.0': {},
+    'b 1.0.0': {
+      'a': '1.0.0'
+    },
+    'c 1.0.0': {
+      'a': '3.0.0'
+    }
+  }, overrides: {
+    'a': '2.0.0'
+  }, result: {
+    'myapp from root': '0.0.0',
+    'a': '2.0.0',
+    'b': '1.0.0',
+    'c': '1.0.0'
+  });
+
+  testResolve('backtracks on overidden package for its constraints', {
+    'myapp 0.0.0': {
+      'shared': '2.0.0'
+    },
+    'a 1.0.0': {
+      'shared': 'any'
+    },
+    'a 2.0.0': {
+      'shared': '1.0.0'
+    },
+    'shared 1.0.0': {},
+    'shared 2.0.0': {}
+  }, overrides: {
+    'a': '<3.0.0'
+  }, result: {
+    'myapp from root': '0.0.0',
+    'a': '1.0.0',
+    'shared': '2.0.0'
+  }, maxTries: 2);
+
+  testResolve('override compatible with locked dependency', {
+    'myapp 0.0.0': {
+      'foo': 'any'
+    },
+    'foo 1.0.0': { 'bar': '1.0.0' },
+    'foo 1.0.1': { 'bar': '1.0.1' },
+    'foo 1.0.2': { 'bar': '1.0.2' },
+    'bar 1.0.0': {},
+    'bar 1.0.1': {},
+    'bar 1.0.2': {}
+  }, lockfile: {
+    'foo': '1.0.1'
+  }, overrides: {
+    'foo': '<1.0.2'
+  }, result: {
+    'myapp from root': '0.0.0',
+    'foo': '1.0.1',
+    'bar': '1.0.1'
+  });
+
+  testResolve('override incompatible with locked dependency', {
+    'myapp 0.0.0': {
+      'foo': 'any'
+    },
+    'foo 1.0.0': { 'bar': '1.0.0' },
+    'foo 1.0.1': { 'bar': '1.0.1' },
+    'foo 1.0.2': { 'bar': '1.0.2' },
+    'bar 1.0.0': {},
+    'bar 1.0.1': {},
+    'bar 1.0.2': {}
+  }, lockfile: {
+    'foo': '1.0.1'
+  }, overrides: {
+    'foo': '>1.0.1'
+  }, result: {
+    'myapp from root': '0.0.0',
+    'foo': '1.0.2',
+    'bar': '1.0.2'
+  });
+
+  testResolve('no version that matches override', {
+    'myapp 0.0.0': {},
+    'foo 2.0.0': {},
+    'foo 2.1.3': {}
+  }, overrides: {
+    'foo': '>=1.0.0 <2.0.0'
+  }, error: noVersion(['myapp']));
+
+  testResolve('override a bad source without error', {
+    'myapp 0.0.0': {
+      'foo from bad': 'any'
+    },
+    'foo 0.0.0': {}
+  }, overrides: {
+    'foo': 'any'
+  }, result: {
+    'myapp from root': '0.0.0',
+    'foo': '0.0.0'
+  });
+}
+
+testResolve(String description, Map packages, {
+    Map lockfile, Map overrides, Map result, FailMatcherBuilder error,
+    int maxTries, bool useBleedingEdgeSdkVersion}) {
+  _testResolve(test, description, packages, lockfile: lockfile,
+      overrides: overrides, result: result, error: error, maxTries: maxTries,
       useBleedingEdgeSdkVersion: useBleedingEdgeSdkVersion);
 }
 
-solo_testResolve(description, packages, {
-             lockfile, result, FailMatcherBuilder error, int maxTries,
-             bool useBleedingEdgeSdkVersion}) {
+solo_testResolve(String description, Map packages, {
+    Map lockfile, Map overrides, Map result, FailMatcherBuilder error,
+    int maxTries, bool useBleedingEdgeSdkVersion}) {
   log.showSolver();
   _testResolve(solo_test, description, packages, lockfile: lockfile,
-      result: result, error: error, maxTries: maxTries,
+      overrides: overrides, result: result, error: error, maxTries: maxTries,
       useBleedingEdgeSdkVersion: useBleedingEdgeSdkVersion);
 }
 
 _testResolve(void testFn(String description, Function body),
-             description, packages, {
-             lockfile, result, FailMatcherBuilder error, int maxTries,
-             bool useBleedingEdgeSdkVersion}) {
+    String description, Map packages, {
+    Map lockfile, Map overrides, Map result, FailMatcherBuilder error,
+    int maxTries, bool useBleedingEdgeSdkVersion}) {
   if (maxTries == null) maxTries = 1;
   if (useBleedingEdgeSdkVersion == null) useBleedingEdgeSdkVersion = false;
 
   testFn(description, () {
     var cache = new SystemCache('.');
     source1 = new MockSource('mock1');
     source2 = new MockSource('mock2');
     cache.register(source1);
     cache.register(source2);
     cache.sources.setDefault(source1.name);
@@ skipping 27 matching lines @@
     // Parse the lockfile.
     var realLockFile = new LockFile.empty();
     if (lockfile != null) {
       lockfile.forEach((name, version) {
         version = new Version.parse(version);
         realLockFile.packages[name] =
             new PackageId(name, source1.name, version, name);
       });
     }
 
+    // Parse the overrides.
+    var realOverrides = [];
+    if (overrides != null) {
+      overrides.forEach((spec, constraint) {
+        realOverrides.add(parseSpec(spec).withConstraint(
+            new VersionConstraint.parse(constraint)));
+      });
+    }
+
     // Make a version number like the continuous build's version.
     var previousVersion = sdk.version;
     if (useBleedingEdgeSdkVersion) {
       sdk.version = new Version(0, 1, 2, build: '0_r12345_juser');
     }
 
     // Resolve the versions.
     var future = resolveVersions(cache.sources, root,
-        lockFile: realLockFile);
+        lockFile: realLockFile, overrides: realOverrides);
 
     var matcher;
     if (result != null) {
       matcher = new SolveSuccessMatcher(result, maxTries);
     } else if (error != null) {
       matcher = error(maxTries);
     }
 
     future = future.whenComplete(() {
       if (useBleedingEdgeSdkVersion) {
@@ skipping 320 matching lines @@
   }
 
   var source = "mock1";
   if (match[7] != null) {
     source = match[7];
     if (source == "root") source = null;
   }
 
   return new PackageId(name, source, parsedVersion, description);
 }