| // Copyright (c) 2017, 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. |
| |
| import 'package:compiler/src/common.dart'; |
| import 'package:compiler/src/compiler.dart'; |
| import 'package:compiler/src/elements/elements.dart'; |
| import 'package:compiler/src/resolution/tree_elements.dart'; |
| import 'package:compiler/src/tree/nodes.dart'; |
| import 'package:compiler/src/types/types.dart'; |
| import 'package:expect/expect.dart'; |
| |
| import '../annotated_code_helper.dart'; |
| import '../memory_compiler.dart'; |
| import 'enumerator.dart'; |
| |
| checkCode(String annotatedCode) async { |
| AnnotatedCode code = new AnnotatedCode(annotatedCode); |
| Map<Id, String> expectedMap = computeExpectedMap(code); |
| Compiler compiler = |
| compilerFor(memorySourceFiles: {'main.dart': code.sourceCode}); |
| compiler.stopAfterTypeInference = true; |
| Uri mainUri = Uri.parse('memory:main.dart'); |
| await compiler.run(mainUri); |
| compiler.mainApp.forEachLocalMember((member) { |
| if (member.isFunction) { |
| checkMember(compiler, expectedMap, member); |
| } else if (member.isClass) { |
| member.forEachLocalMember((member) { |
| checkMember(compiler, expectedMap, member); |
| }); |
| } |
| }); |
| expectedMap.forEach((Id id, String expected) { |
| reportHere( |
| compiler.reporter, |
| new SourceSpan(mainUri, id.value, id.value + 1), |
| 'expected:${expected},actual:null'); |
| }); |
| } |
| |
| void checkMember( |
| Compiler compiler, Map<Id, String> expectedMap, MemberElement member) { |
| ResolvedAst resolvedAst = member.resolvedAst; |
| if (resolvedAst.kind != ResolvedAstKind.PARSED) return; |
| compiler.reporter.withCurrentElement(member.implementation, () { |
| resolvedAst.node.accept(new TypeMaskChecker( |
| compiler.reporter, |
| expectedMap, |
| resolvedAst.elements, |
| compiler.globalInference.results.resultOf(member))); |
| }); |
| } |
| |
| Map<Id, String> computeExpectedMap(AnnotatedCode code) { |
| Map<Id, String> map = <Id, String>{}; |
| for (Annotation annotation in code.annotations) { |
| map[new Id(annotation.offset)] = annotation.text; |
| } |
| return map; |
| } |
| |
| class TypeMaskChecker extends Visitor with AstEnumeratorMixin { |
| final DiagnosticReporter reporter; |
| final Map<Id, String> expectedMap; |
| final TreeElements elements; |
| final GlobalTypeInferenceElementResult result; |
| |
| TypeMaskChecker(this.reporter, this.expectedMap, this.elements, this.result); |
| |
| visitNode(Node node) { |
| node.visitChildren(this); |
| } |
| |
| String annotationForId(Id id) { |
| if (id == null) return null; |
| return expectedMap.remove(id); |
| } |
| |
| void checkValue(Node node, String expected, TypeMask value) { |
| if (value != null || expected != null) { |
| String valueText = '$value'; |
| if (valueText != expected) { |
| reportHere(reporter, node, 'expected:${expected},actual:${value}'); |
| } |
| Expect.equals(expected, valueText); |
| } |
| } |
| |
| void checkSend(Send node) { |
| Id id = computeId(node); |
| TypeMask value = result.typeOfSend(node); |
| String expected = annotationForId(id); |
| checkValue(node, expected, value); |
| } |
| |
| visitSend(Send node) { |
| checkSend(node); |
| visitNode(node); |
| } |
| |
| visitSendSet(SendSet node) { |
| checkSend(node); |
| visitNode(node); |
| } |
| } |