uk.co.flax.biosolr.ontology.core.ols.graph.GraphTest.java Source code

Introduction

Here is the source code for uk.co.flax.biosolr.ontology.core.ols.graph.GraphTest.java

Source

/**
 * Copyright (c) 2015 Lemur Consulting Ltd.
 * <p>
 * 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
 * <p>
 * http://www.apache.org/licenses/LICENSE-2.0
 * <p>
 * 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.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package uk.co.flax.biosolr.ontology.core.ols.graph;

import com.fasterxml.jackson.databind.DeserializationFeature;
import com.fasterxml.jackson.databind.ObjectMapper;
import org.junit.Test;
import uk.co.flax.biosolr.ontology.core.ols.OLSHttpClientTest;

import java.io.IOException;
import java.net.URISyntaxException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Collections;
import java.util.List;

import static org.junit.Assert.*;

/**
 * Unit tests for the graph class.
 *
 * <p>Created by Matt Pearce on 27/10/15.</p>
 * @author Matt Pearce
 */
public class GraphTest {

    static final String GRAPH_FILE = "ols/graph.json";

    @Test
    public void deserialize_fromFile() throws Exception {
        Graph graph = readGraphFromFile(GRAPH_FILE);
        assertNotNull(graph);
        assertNotNull(graph.getNodes());
        assertFalse(graph.getNodes().isEmpty());
        assertNotNull(graph.getEdges());
        assertFalse(graph.getEdges().isEmpty());
    }

    @Test
    public void getEdgesBySource_nullEdges() throws Exception {
        Graph graph = new Graph(null, null);
        Collection<Edge> edges = graph.getEdgesBySource("http://www.ebi.ac.uk/efo/EFO_0005580", true);
        assertNotNull(edges);
        assertEquals(0, edges.size());
    }

    @Test
    public void getEdgesBySource_badSource() throws Exception {
        Graph graph = new Graph(null, Collections.emptyList());
        Collection<Edge> edges = graph.getEdgesBySource("http://www.ebi.ac.uk/efo/EFO_0005580", true);
        assertNotNull(edges);
        assertEquals(0, edges.size());
    }

    @Test
    public void getEdgesBySource_includeChildren() throws Exception {
        Graph graph = readGraphFromFile(GRAPH_FILE);
        Collection<Edge> edges = graph.getEdgesBySource("http://www.ebi.ac.uk/efo/EFO_0005580", true);
        assertNotNull(edges);
        assertEquals(4, edges.size());
    }

    @Test
    public void getEdgesBySource_ignoreChildren() throws Exception {
        Graph graph = readGraphFromFile(GRAPH_FILE);
        List<Edge> edges = new ArrayList<>(graph.getEdgesBySource("http://www.ebi.ac.uk/efo/EFO_0005580", false));
        assertNotNull(edges);
        assertEquals(1, edges.size());
        assertEquals("has_disease_location", edges.get(0).getLabel());
    }

    private Graph readGraphFromFile(String filePath) throws URISyntaxException, IOException {
        ObjectMapper mapper = new ObjectMapper();
        mapper.disable(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES);
        return mapper.readValue(OLSHttpClientTest.getFile(filePath), Graph.class);
    }

}