private <T extends Annotation> Collection<T> getAnnotations( DbUnitTestContext testContext, Class<T> annotationType) { List<T> annotations = new ArrayList<T>(); addAnnotationToList( annotations, AnnotationUtils.findAnnotation(testContext.getTestClass(), annotationType)); addAnnotationToList( annotations, AnnotationUtils.findAnnotation(testContext.getTestMethod(), annotationType)); return annotations; }
private IDataSet loadDataset(DbUnitTestContext testContext, String dataSetLocation) throws Exception { DataSetLoader dataSetLoader = testContext.getDataSetLoader(); if (StringUtils.hasLength(dataSetLocation)) { IDataSet dataSet = dataSetLoader.loadDataSet(testContext.getTestClass(), dataSetLocation); Assert.notNull( dataSet, "Unable to load dataset from \"" + dataSetLocation + "\" using " + dataSetLoader.getClass()); return dataSet; } return null; }