Add test for reading model from URL

atextor · Mar 16, 2021 · e5f05e9 · e5f05e9
1 parent 9e21805
commit e5f05e9
Showing 1 changed file with 24 additions and 3 deletions.
diff --git a/write/src/test/java/de/atextor/owlcli/write/RdfWriterTest.java b/write/src/test/java/de/atextor/owlcli/write/RdfWriterTest.java
@@ -22,8 +22,10 @@
 
 import java.io.ByteArrayInputStream;
 import java.io.ByteArrayOutputStream;
+import java.io.IOException;
 import java.io.InputStream;
 import java.io.StringReader;
+import java.net.URL;
 
 import static org.assertj.core.api.Assertions.assertThat;
 
@@ -48,14 +50,19 @@ private InputStream turtleInputStream() {
         return new ByteArrayInputStream( turtleDocument.getBytes() );
     }
 
-    private boolean canBeParsedAs( final String document, final String format, final int expectedNumberOfStatements ) {
+    private Model parseModel( final String document, final String format ) {
         final Model model = ModelFactory.createDefaultModel();
         try {
             model.read( new StringReader( document ), "", format );
+            return model;
         } catch ( final Throwable t ) {
-            return false;
+            return null;
         }
-        return model.listStatements().toList().size() == expectedNumberOfStatements;
+    }
+
+    private boolean canBeParsedAs( final String document, final String format, final int expectedNumberOfStatements ) {
+        final Model model = parseModel( document, format );
+        return model != null && model.listStatements().toList().size() == expectedNumberOfStatements;
     }
 
     @Test
@@ -103,4 +110,18 @@ public void testWriteRdfXml() {
         assertThat( canBeParsedAs( out.toString(), "N-TRIPLE", 8 ) ).isFalse();
         assertThat( canBeParsedAs( out.toString(), "RDF/XML", 8 ) ).isTrue();
     }
+
+    @Test
+    public void testReadFromUrl() throws IOException {
+        final URL url = new URL( "http://purl.org/atextor/ontology/turtle-formatting" );
+        final Configuration configuration = Configuration.builder()
+            .inputFormat( Configuration.Format.TURTLE )
+            .outputFormat( Configuration.Format.TURTLE ).build();
+
+        final ByteArrayOutputStream out = new ByteArrayOutputStream();
+        final Try<Void> result = writer.write( url, out, configuration );
+
+        assertThat( result ).hasSize( 1 );
+        assertThat( parseModel( out.toString(), "TURTLE" ) ).isNotNull();
+    }
 }