From 509524b0c4b38b23c587b08b98c9da2ff7fafc1c Mon Sep 17 00:00:00 2001 From: Roman Shevchenko Date: Wed, 16 Jun 2010 16:09:08 +0400 Subject: [PATCH] findPackage() added to Java code insight text fixture --- .../testFramework/fixtures/JavaCodeInsightTestFixture.java | 4 ++++ .../fixtures/impl/JavaCodeInsightTestFixtureImpl.java | 7 +++++++ 2 files changed, 11 insertions(+) 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; + } }