extras/rya.geoindexing/geo.mongo/src/test/java/org/apache/rya/indexing/geotemporal/mongo/MongoGeoTemporalIndexerIT.java

/**
 * Licensed to the Apache Software Foundation (ASF) under one
 * or more contributor license agreements.  See the NOTICE file
 * distributed with this work for additional information
 * regarding copyright ownership.  The ASF licenses this file
 * to you 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.  See the License for the
 * specific language governing permissions and limitations
 * under the License.
 */
package org.apache.rya.indexing.geotemporal.mongo;

import static org.apache.rya.indexing.geotemporal.GeoTemporalTestUtils.makeInstant;
import static org.apache.rya.indexing.geotemporal.GeoTemporalTestUtils.point;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;

import java.util.Optional;

import org.apache.rya.api.domain.RyaStatement;
import org.apache.rya.api.resolver.RdfToRyaConversions;
import org.apache.rya.indexing.GeoConstants;
import org.apache.rya.indexing.TemporalInstant;
import org.apache.rya.indexing.geotemporal.model.Event;
import org.apache.rya.indexing.geotemporal.storage.EventStorage;
import org.apache.rya.mongodb.MongoITBase;
import org.junit.Before;
import org.junit.Test;
import org.eclipse.rdf4j.model.Resource;
import org.eclipse.rdf4j.model.IRI;
import org.eclipse.rdf4j.model.Value;
import org.eclipse.rdf4j.model.ValueFactory;
import org.eclipse.rdf4j.model.impl.SimpleValueFactory;

import com.vividsolutions.jts.geom.Geometry;

/**
 * Integration tests the methods of {@link MongoGeoTemporalIndexer}.
 */
public class MongoGeoTemporalIndexerIT extends MongoITBase {
    private MongoGeoTemporalIndexer indexer;

    @Before
    public void makeTestIndexer() throws Exception {
        indexer = new MongoGeoTemporalIndexer();
        indexer.setConf(conf);
        indexer.init();
    }

    @Test
    public void ensureEvent() throws Exception {
        final RyaStatement geoStmnt = statement(point(0, 0));
        final RyaStatement timeStmnt = statement(makeInstant(0));

        final EventStorage store = indexer.getEventStorage();

        indexer.storeStatement(geoStmnt);
        Optional<Event> evnt = store.get(geoStmnt.getSubject());
        assertTrue(evnt.isPresent());
        Event expected = Event.builder()
            .setSubject(geoStmnt.getSubject())
            .setGeometry(point(0, 0))
            .build();
        assertEquals(expected, evnt.get());

        indexer.storeStatement(timeStmnt);
        evnt = store.get(timeStmnt.getSubject());
        assertTrue(evnt.isPresent());
        expected = Event.builder()
            .setSubject(geoStmnt.getSubject())
            .setGeometry(point(0, 0))
            .setTemporalInstant(makeInstant(0))
            .build();
        assertEquals(expected, evnt.get());

        indexer.deleteStatement(geoStmnt);
        evnt = store.get(timeStmnt.getSubject());
        assertTrue(evnt.isPresent());
        expected = Event.builder()
            .setSubject(timeStmnt.getSubject())
            .setTemporalInstant(makeInstant(0))
            .build();
        assertEquals(expected, evnt.get());

        indexer.deleteStatement(timeStmnt);
        evnt = store.get(timeStmnt.getSubject());
        assertTrue(evnt.isPresent());
        expected = Event.builder()
            .setSubject(timeStmnt.getSubject())
            .build();
        assertEquals(expected, evnt.get());
    }

    private static RyaStatement statement(final Geometry geo) {
        final ValueFactory vf = SimpleValueFactory.getInstance();
        final Resource subject = vf.createIRI("uri:test");
        final IRI predicate = GeoConstants.GEO_AS_WKT;
        final Value object = vf.createLiteral(geo.toString(), GeoConstants.XMLSCHEMA_OGC_WKT);
        return RdfToRyaConversions.convertStatement(vf.createStatement(subject, predicate, object));
    }

    private static RyaStatement statement(final TemporalInstant instant) {
        final ValueFactory vf = SimpleValueFactory.getInstance();
        final Resource subject = vf.createIRI("uri:test");
        final IRI predicate = vf.createIRI("Property:atTime");
        final Value object = vf.createLiteral(instant.toString());
        return RdfToRyaConversions.convertStatement(vf.createStatement(subject, predicate, object));
    }
}