Open Source Projects Europe

/**

 * Copyright (c) 2013/2014 Verein zur Foerderung der IT-Sicherheit in Oesterreich (SBA).

 * The work has been developed in the TIMBUS Project and the above-mentioned are Members of the TIMBUS Consortium.

 * TIMBUS is supported by the European Union under the 7th Framework Programme for research and technological

 * development and demonstration activities (FP7/2007-2013) under grant agreement no. 269940.

 *

 * Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with

 * the License. You may obtain a copy of the License at: http://www.apache.org/licenses/LICENSE-2.0

 * Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on

 * an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied, including without

 * limitation, any warranties or conditions of TITLE, NON-INFRINGEMENT, MERCHANTIBITLY, or FITNESS FOR A PARTICULAR

 * PURPOSE. In no event and under no legal theory, whether in tort (including negligence), contract, or otherwise,

 * unless required by applicable law or agreed to in writing, shall any Contributor be liable for damages, including

 * any direct, indirect, special, incidental, or consequential damages of any character arising as a result of this

 * License or out of the use or inability to use the Work.

 * See the License for the specific language governing permissions and limitation under the License.

 */

package org.sbaresearch.owl;

import com.hp.hpl.jena.ontology.OntModel;

import com.hp.hpl.jena.ontology.OntModelSpec;

import com.hp.hpl.jena.query.QuerySolution;

import com.hp.hpl.jena.rdf.model.Model;

import com.hp.hpl.jena.rdf.model.ModelFactory;

import org.apache.commons.io.IOUtils;

import org.apache.jena.riot.RiotException;

import org.junit.Test;

import org.semanticweb.owlapi.model.OWLOntologyStorageException;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.List;

import static org.hamcrest.CoreMatchers.is;

import static org.hamcrest.Matchers.containsInAnyOrder;

import static org.junit.Assert.assertThat;

import static org.junit.Assert.assertTrue;

public class JenaQueryFacadeTest {

    private String sparqlGetAllClasses = "" +

            "PREFIX rdfs: <http://www.w3.org/2000/01/rdf-schema#>\n" +

            "SELECT ?subject WHERE { ?subject rdfs:subClassOf ?object }";

    private List<String> allClasses = new ArrayList<String>() {{

        add(TestResources.ontologyBaseUrl + "#A");

        add(TestResources.ontologyBaseUrl + "#B");

        add(TestResources.ontologyBaseUrl + "#C");

    }};

    @Test(expected = IllegalArgumentException.class)

    public void testQuery_emptyQuery_shouldThrowException() throws OWLOntologyStorageException {

        JenaQueryFacade queryFacade = createQueryFacade();

        queryFacade.query("");

    }

    @Test(expected = IllegalArgumentException.class)

    public void testQuery_queryIsNull_shouldThrowException() throws OWLOntologyStorageException {

        JenaQueryFacade queryFacade = createQueryFacade();

        queryFacade.query(null);

    }

    @Test(expected = InvalidQueryException.class)

    public void testQuery_invalidQuery_shouldThrowException() throws OWLOntologyStorageException {

        JenaQueryFacade queryFacade = createQueryFacade();

        queryFacade.query("blah");

    }

    @Test

    public void testExtractColumn_shouldReturnSpecifiedColumn() throws OWLOntologyStorageException {

        JenaQueryFacade queryFacade = createQueryFacade();

        List<String> actual = JenaQueryFacade.extractColumn(queryFacade.query(sparqlGetAllClasses), "subject");

        assertThat(actual, containsInAnyOrder(allClasses.toArray()));

    }

    @Test

    public void testRemoveInColumn_resultsContainNeedle_shouldRemoveNeedle() {

        List<String> actual = JenaQueryFacade.removeInColumn(Arrays.asList("blah#01", "blah#02"), "blah#");

        assertThat(actual, containsInAnyOrder("01", "02"));

    }

    @Test

    public void testRemoveInColumn_resultsDoesNotContainNeedle_shouldNotChangeAnything() {

        List<String> actual = JenaQueryFacade.removeInColumn(Arrays.asList("blah#01", "blah#02"), "blub#");

        assertThat(actual, containsInAnyOrder("blah#01", "blah#02"));

    }

    @Test

    public void testRemoveNamespace_containingNamespaces_shouldReturnFragments() {

        List<String> actual = JenaQueryFacade.removeNamespace(Arrays.asList("blah#01", "blah#02"));

        assertThat(actual, containsInAnyOrder("01", "02"));

    }

    @Test

    public void testRemoveNamespace_containingFragmentsOnly_shouldNotAlterList() {

        List<String> actual = JenaQueryFacade.removeNamespace(Arrays.asList("01", "02"));

        assertThat(actual, containsInAnyOrder("01", "02"));

    }

    @Test

    public void testQuery_validQuery_shouldReturnResults() throws OWLOntologyStorageException {

        JenaQueryFacade queryFacade = createQueryFacade();

        List<QuerySolution> actual = queryFacade.query(sparqlGetAllClasses);

        for (QuerySolution solution : actual) {

            assertTrue(allClasses.contains(solution.get("subject").toString()));

        }

        assertThat(actual.size(), is(allClasses.size()));

    }

    @Test(expected = RiotException.class)

    public void testQuery_queryWithUndefinedPrefix_shouldThrowException() throws OWLOntologyStorageException {

        JenaQueryFacade queryFacade = createQueryFacadeWithoutPrefix();

        queryFacade.query(sparqlGetAllClasses);

    }

    private JenaQueryFacade createQueryFacade() {

        return new JenaQueryFacade(createModel(TestResources.testOntology));

    }

    private JenaQueryFacade createQueryFacadeWithoutPrefix() {

        return new JenaQueryFacade(createModel(TestResources.testOntologyWithoutPrefix));

    }

    private Model createModel(String ontology) {

        OntModel ontologyModel = ModelFactory.createOntologyModel(OntModelSpec.OWL_MEM);

        return ontologyModel.read(IOUtils.toInputStream(ontology), TestResources.ontologyBaseUrl);

    }

}