| Index: compiler/javatests/com/google/dart/compiler/backend/common/TypeHeuristicImplementationTest.java
|
| diff --git a/compiler/javatests/com/google/dart/compiler/backend/common/TypeHeuristicImplementationTest.java b/compiler/javatests/com/google/dart/compiler/backend/common/TypeHeuristicImplementationTest.java
|
| index 89f28cc5c6d7b07f16e66b62e08ee99466bc5307..c2da30d46042924f530f97b126d7720b25fe8e4f 100644
|
| --- a/compiler/javatests/com/google/dart/compiler/backend/common/TypeHeuristicImplementationTest.java
|
| +++ b/compiler/javatests/com/google/dart/compiler/backend/common/TypeHeuristicImplementationTest.java
|
| @@ -851,9 +851,13 @@ public class TypeHeuristicImplementationTest extends CompilerTestCase {
|
| }
|
|
|
| private DartClass getClass(DartUnit unit, String name) {
|
| - DartNode node = unit.getLibrary().getTopLevelNode(name);
|
| - if (node instanceof DartClass) {
|
| - return (DartClass) node;
|
| + for (DartNode topLevelNode : unit.getTopLevelNodes()) {
|
| + if (topLevelNode instanceof DartClass) {
|
| + DartClass dartClass = (DartClass) topLevelNode;
|
| + if (dartClass.getName().getTargetName().equals(name)) {
|
| + return dartClass;
|
| + }
|
| + }
|
| }
|
| return null;
|
| }
|
|
|