check not null

AKSW · Feb 22, 2016 · df3b7a7 · df3b7a7
1 parent 45d34ae
commit df3b7a7
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/rdfunit-junit/src/main/java/org/aksw/rdfunit/junit/RdfUnitJunitRunner.java b/rdfunit-junit/src/main/java/org/aksw/rdfunit/junit/RdfUnitJunitRunner.java
@@ -184,7 +184,7 @@ private void setAdditionalData() throws InitializationError {
         }
     }
 
-    private SchemaSource createSchemaSourceFromSchema() {
+    private SchemaSource createSchemaSourceFromSchema() throws InitializationError {
         return SchemaSourceFactory.createSchemaSourceSimple(getSchema().uri(), getSchemaReader());
     }
 
@@ -199,8 +199,12 @@ private RdfReader getSchemaReader() {
         return schemaReader;
     }
 
-    private Schema getSchema() {
-        return getTestClass().getAnnotation(Schema.class);
+    private Schema getSchema() throws InitializationError {
+        Schema schema = getTestClass().getAnnotation(Schema.class);
+        if (schema == null) {
+            throw new InitializationError("Cannot read Schema");
+        }
+        return schema;
     }
 
     RdfReader getAdditionalDataModel() {