diff --git a/java/testFramework/src/com/intellij/testFramework/fixtures/JavaCodeInsightTestFixture.java b/java/testFramework/src/com/intellij/testFramework/fixtures/JavaCodeInsightTestFixture.java index 0e94b11d159f..f7a308580bac 100644 --- a/java/testFramework/src/com/intellij/testFramework/fixtures/JavaCodeInsightTestFixture.java +++ b/java/testFramework/src/com/intellij/testFramework/fixtures/JavaCodeInsightTestFixture.java @@ -17,6 +17,7 @@ package com.intellij.testFramework.fixtures; import com.intellij.psi.JavaPsiFacade; import com.intellij.psi.PsiClass; +import com.intellij.psi.PsiPackage; import org.jetbrains.annotations.NonNls; import org.jetbrains.annotations.NotNull; @@ -32,4 +33,7 @@ public interface JavaCodeInsightTestFixture extends CodeInsightTestFixture { @NotNull PsiClass findClass(@NotNull @NonNls String name); + + @NotNull + PsiPackage findPackage(@NotNull @NonNls String name); } diff --git a/java/testFramework/src/com/intellij/testFramework/fixtures/impl/JavaCodeInsightTestFixtureImpl.java b/java/testFramework/src/com/intellij/testFramework/fixtures/impl/JavaCodeInsightTestFixtureImpl.java index 32d51da7a14d..92e748c8c76d 100644 --- a/java/testFramework/src/com/intellij/testFramework/fixtures/impl/JavaCodeInsightTestFixtureImpl.java +++ b/java/testFramework/src/com/intellij/testFramework/fixtures/impl/JavaCodeInsightTestFixtureImpl.java @@ -62,4 +62,11 @@ public class JavaCodeInsightTestFixtureImpl extends CodeInsightTestFixtureImpl i assertNotNull("Class " + name + " not found", aClass); return aClass; } + + @NotNull + public PsiPackage findPackage(@NotNull @NonNls final String name) { + final PsiPackage aPackage = getJavaFacade().findPackage(name); + assertNotNull("Package " + name + " not found", aPackage); + return aPackage; + } }