@Test public void testFindDataProviderLocationShouldReturnTestClassForNotSetLocationInUseDataProviderAnnotation() { // Given: doReturn(new Class<?>[0]).when(useDataProvider).location(); // When: TestClass result = underTest.findDataProviderLocation(useDataProvider); // Then: assertThat(result).isEqualTo(testClass); }
@Test public void testFindDataProviderLocationShouldReturnTestClassContainingSetLocationInUseDataProviderAnnotation() { // Given: final Class<?> dataProviderLocation = DataProviderRunnerTest.class; doReturn(new Class<?>[] {dataProviderLocation}).when(useDataProvider).location(); // When: TestClass result = underTest.findDataProviderLocation(useDataProvider); // Then: assertThat(result).isNotNull(); // assertThat(result.getJavaClass()).isEqualTo(dataProviderLocation); assertThat(result.getName()).isEqualTo(dataProviderLocation.getName()); }