Index: pkg/analyzer/test/src/task/strong/strong_test_helper.dart |
diff --git a/pkg/analyzer/test/src/task/strong/strong_test_helper.dart b/pkg/analyzer/test/src/task/strong/strong_test_helper.dart |
index 19ed5b3ef3cade4fb7611e78fd655b935e0e6523..e04fb308c1606490e669ce9c6fca61e3d405e024 100644 |
--- a/pkg/analyzer/test/src/task/strong/strong_test_helper.dart |
+++ b/pkg/analyzer/test/src/task/strong/strong_test_helper.dart |
@@ -52,7 +52,9 @@ void addFile(String content, {String name: '/main.dart'}) { |
/// |
/// See [addFile] for more information about how to encode expectations in |
/// the file text. |
-void check() { |
+/// |
+/// Returns the main resolved library. This can be used for further checks. |
+CompilationUnit check() { |
_checkCalled = true; |
expect(files.getFile('/main.dart').exists, true, |
@@ -93,12 +95,16 @@ void check() { |
_expectErrors(resolved, errors); |
} |
} |
+ |
+ return initialLibrary; |
} |
/// Adds a file using [addFile] and calls [check]. |
-void checkFile(String content) { |
+/// |
+/// Also returns the resolved compilation unit. |
+CompilationUnit checkFile(String content) { |
addFile(content); |
- check(); |
+ return check(); |
} |
SourceSpanWithContext _createSpanHelper( |