@Test(expected = IllegalArgumentException.class)
public void testGetDataProviderMethodShouldThrowIllegalArgumentExceptionIfTestMethodIsNull() {
// Given:
// When:
underTest.getDataProviderMethod(null);
// Then: expect exception
}
@Test
public void testGetDataProviderMethodShouldReturnNullForNonDataProviderMethod() {
// Given:
doReturn(null).when(testMethod).getAnnotation(UseDataProvider.class);
// When:
FrameworkMethod result = underTest.getDataProviderMethod(testMethod);
// Then:
assertThat(result).isNull();
}
@Test
public void testGetDataProviderMethodShouldReturnNullForNotFoundDataProviderMethod() {
// Given:
doReturn(useDataProvider).when(testMethod).getAnnotation(UseDataProvider.class);
doReturn("notAvailableDataProviderMethod").when(useDataProvider).value();
doReturn(testClass).when(underTest).findDataProviderLocation(useDataProvider);
doReturn(asList(dataProviderMethod)).when(testClass).getAnnotatedMethods(DataProvider.class);
doReturn("availableDataProviderMethod").when(dataProviderMethod).getName();
// When:
FrameworkMethod result = underTest.getDataProviderMethod(testMethod);
// Then:
assertThat(result).isNull();
}
@Test
public void testGetDataProviderMethodShouldReturnDataProviderMethodIfItExists() {
// Given:
final String dataProviderMethodName = "availableDataProviderMethod";
doReturn(useDataProvider).when(testMethod).getAnnotation(UseDataProvider.class);
doReturn(dataProviderMethodName).when(useDataProvider).value();
doReturn(testClass).when(underTest).findDataProviderLocation(useDataProvider);
doReturn(asList(dataProviderMethod)).when(testClass).getAnnotatedMethods(DataProvider.class);
doReturn(dataProviderMethodName).when(dataProviderMethod).getName();
// When:
FrameworkMethod result = underTest.getDataProviderMethod(testMethod);
// Then:
assertThat(result).isEqualTo(dataProviderMethod);
}