Improvements to task model

pkg/analyzer/test/src/task/model_test.dart

 // Copyright (c) 2015, 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 test.src.task.model_test;
 
 import 'package:analyzer/src/generated/engine.dart' hide AnalysisTask;
+import 'package:analyzer/src/generated/java_engine.dart';
+import 'package:analyzer/src/generated/source.dart';
 import 'package:analyzer/src/task/model.dart';
 import 'package:analyzer/src/task/targets.dart';
 import 'package:analyzer/task/model.dart';
 import 'package:unittest/unittest.dart';
 
 import '../../generated/test_support.dart';
 import '../../reflective_tests.dart';
 import 'test_support.dart';
 
 main() {
   groupSep = ' | ';
+  runReflectiveTests(AnalysisTaskTest);
   runReflectiveTests(ContributionPointImplTest);
   runReflectiveTests(ResultDescriptorImplTest);
   runReflectiveTests(TaskDescriptorImplTest);
 }
 
 @reflectiveTest
+class AnalysisTaskTest extends EngineTestCase {
+  test_getRequiredInput_missingKey() {
+    SourceTarget target = new SourceTarget(new TestSource());
+    AnalysisTask task = new TestAnalysisTask(null, target);
+    task.inputs = {
+      'a': 'b'
+    };
+    expect(
+        () => task.getRequiredInput('c'),
+        throwsA(new isInstanceOf<AnalysisException>()));
+  }
+
+  test_getRequiredInput_noInputs() {
+    SourceTarget target = new SourceTarget(new TestSource());
+    AnalysisTask task = new TestAnalysisTask(null, target);
+    expect(
+        () => task.getRequiredInput('x'),
+        throwsA(new isInstanceOf<AnalysisException>()));
+  }
+
+  test_getRequiredInput_valid() {
+    SourceTarget target = new SourceTarget(new TestSource());
+    AnalysisTask task = new TestAnalysisTask(null, target);
+    String key = 'a';
+    String value = 'b';
+    task.inputs = {
+      key: value
+    };
+    expect(task.getRequiredInput(key), value);
+  }
+
+  test_getRequiredSource() {
+    Source source = new TestSource();
+    SourceTarget target = new SourceTarget(source);
+    AnalysisTask task = new TestAnalysisTask(null, target);
+    expect(task.getRequiredSource(), source);
+  }
+}
+
+@reflectiveTest
 class ContributionPointImplTest extends EngineTestCase {
   test_contributors_empty() {
-    ContributionPointImpl point = new ContributionPointImpl('point');
+    ContributionPointImpl point = new ContributionPointImpl('point', null);
     List<ResultDescriptor> contributors = point.contributors;
     expect(contributors, isEmpty);
   }
 
   test_contributors_nonEmpty() {
-    ResultDescriptorImpl result1 = new ResultDescriptorImpl('result1');
-    ResultDescriptorImpl result2 = new ResultDescriptorImpl('result2');
-    ContributionPointImpl point = new ContributionPointImpl('point');
+    ResultDescriptorImpl result1 = new ResultDescriptorImpl('result1', null);
+    ResultDescriptorImpl result2 = new ResultDescriptorImpl('result2', null);
+    ContributionPointImpl point = new ContributionPointImpl('point', null);
     point.recordContributor(result1);
     point.recordContributor(result2);
     List<ResultDescriptor> contributors = point.contributors;
     expect(contributors, isNotNull);
     expect(contributors, hasLength(2));
     if (!(contributors[0] == result1 && contributors[1] == result2) ||
         (contributors[0] == result2 && contributors[1] == result1)) {
       fail("Invalid contributors: $contributors");
     }
   }
 
   test_create() {
-    expect(new ContributionPointImpl('name'), isNotNull);
+    expect(new ContributionPointImpl('name', null), isNotNull);
   }
 
   test_name() {
     String name = 'point';
-    ContributionPointImpl point = new ContributionPointImpl(name);
+    ContributionPointImpl point = new ContributionPointImpl(name, null);
     expect(point.name, name);
   }
 }
 
 @reflectiveTest
 class ResultDescriptorImplTest extends EngineTestCase {
   test_create_withContribution() {
-    ContributionPointImpl point = new ContributionPointImpl('point');
+    ContributionPointImpl point = new ContributionPointImpl('point', null);
     ResultDescriptorImpl result =
-        new ResultDescriptorImpl('result', contributesTo: point);
+        new ResultDescriptorImpl('result', null, contributesTo: point);
     expect(result, isNotNull);
     List<ResultDescriptor> contributors = point.contributors;
     expect(contributors, unorderedEquals([result]));
   }
 
   test_create_withoutContribution() {
-    expect(new ResultDescriptorImpl('name'), isNotNull);
+    expect(new ResultDescriptorImpl('name', null), isNotNull);
   }
 
   test_inputFor() {
     SourceTarget target = new SourceTarget(null);
-    ResultDescriptorImpl result = new ResultDescriptorImpl('result');
+    ResultDescriptorImpl result = new ResultDescriptorImpl('result', null);
     TaskInput input = result.inputFor(target);
     expect(input, isNotNull);
   }
 
   test_name() {
     String name = 'result';
-    ResultDescriptorImpl result = new ResultDescriptorImpl(name);
+    ResultDescriptorImpl result = new ResultDescriptorImpl(name, null);
     expect(result.name, name);
   }
 }
 
 @reflectiveTest
 class TaskDescriptorImplTest extends EngineTestCase {
   test_create() {
     String name = 'name';
     BuildTask buildTask = (context, target) {};
     CreateTaskInputs createTaskInputs = (target) {};
     List<ResultDescriptor> results = <ResultDescriptor>[];
-    TaskDescriptorImpl descriptor = new TaskDescriptorImpl(name, buildTask, createTaskInputs, results);
+    TaskDescriptorImpl descriptor =
+        new TaskDescriptorImpl(name, buildTask, createTaskInputs, results);
     expect(descriptor, isNotNull);
     expect(descriptor.name, name);
     expect(descriptor.buildTask, equals(buildTask));
     expect(descriptor.createTaskInputs, equals(createTaskInputs));
     expect(descriptor.results, results);
   }
 
   test_createTask() {
-    BuildTask buildTask = (context, target) => new TestAnalysisTask(context, target);
+    BuildTask buildTask =
+        (context, target) => new TestAnalysisTask(context, target);
     CreateTaskInputs createTaskInputs = (target) {};
     List<ResultDescriptor> results = <ResultDescriptor>[];
-    TaskDescriptorImpl descriptor = new TaskDescriptorImpl('name', buildTask, createTaskInputs, results);
+    TaskDescriptorImpl descriptor =
+        new TaskDescriptorImpl('name', buildTask, createTaskInputs, results);
     AnalysisContext context = null;
     SourceTarget target = new SourceTarget(null);
     Map<String, dynamic> inputs = {};
     AnalysisTask createTask = descriptor.createTask(context, target, inputs);
     expect(createTask, isNotNull);
     expect(createTask.context, context);
     expect(createTask.inputs, inputs);
     expect(createTask.target, target);
   }
 }