@Test
public void shouldNotFindClassForAnnotationType() {
// given
given(psiClass.isAnnotationType()).willReturn(true);
// when
PsiClass result = builderFinder.findClassForBuilder(psiClass);
// then
assertThat(result, is(nullValue()));
}
@Test
public void shouldFindClassForBuilderWhenClassWithTheExactBuildersNameIsPresent() {
// given
given(classFinder.findClass(CLASS_NAME, project)).willReturn(psiClass);
// when
PsiClass result = builderFinder.findClassForBuilder(psiClass);
// then
assertThat(result, is(notNullValue()));
assertThat(result.getName(), is(CLASS_NAME));
}
@Test
public void shouldNotFindClassForBuilderWhenClassFounderReturnsNull() {
// given
given(classFinder.findClass(BUILDER_NAME, project)).willReturn(null);
// when
PsiClass result = builderFinder.findClassForBuilder(builderClass);
// then
assertThat(result, is(nullValue()));
verify(classFinder).findClass(CLASS_NAME, project);
}
