/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 *
 *    (C) 2015, Open Source Geospatial Foundation (OSGeo)
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation;
 *    version 2.1 of the License.
 *
 *    This library is distributed in the hope that it will be useful,
 *    but WITHOUT ANY WARRANTY; without even the implied warranty of
 *    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 *    Lesser General Public License for more details.
 */
package org.geotools.data.solr;

import com.vividsolutions.jts.geom.Geometry;
import org.geotools.data.Query;
import org.geotools.data.simple.SimpleFeatureCollection;
import org.geotools.data.simple.SimpleFeatureIterator;
import org.geotools.data.simple.SimpleFeatureSource;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.feature.type.GeometryDescriptor;

import java.util.HashMap;
import java.util.Map;
import java.util.Properties;

public class SolrSingleLayerMappingTest extends SolrTestSupport {

    @Override
    protected void setUpInternal() throws Exception {
        super.setUpInternal();
    }

    @Override
    protected Map createConnectionParams(String url, Properties fixture) {
        Map params = new HashMap();
        params.put(SolrDataStoreFactory.URL.key, url);
        params.put(SolrDataStoreFactory.LAYER_MAPPER.key, "SINGLE");
        params.put(SolrDataStoreFactory.NAMESPACE.key, SolrDataStoreFactory.NAMESPACE.sample);

        return params;
    }

    public void testTypeNames() throws Exception {
        String[] names = dataStore.getTypeNames();
        assertEquals(1, names.length);
        assertEquals(names[0], coreName(dataStore));
    }

    public void testSchema() throws Exception {
        init(coreName(dataStore));
        SimpleFeatureType schema = dataStore.getSchema(coreName(dataStore));
        assertNotNull(schema);

        assertNotNull(schema.getGeometryDescriptor());
        assertTrue(schema.getDescriptor("geo") instanceof GeometryDescriptor);
        assertTrue(schema.getDescriptor("geo2") instanceof GeometryDescriptor);
        assertTrue(schema.getDescriptor("geo3") instanceof GeometryDescriptor);

    }
    public void testFeatureSource() throws Exception {
        init(coreName(dataStore));
        SimpleFeatureSource featureSource = dataStore.getFeatureSource(coreName(dataStore));
        assertEquals(13, featureSource.getCount(Query.ALL));

        SimpleFeatureCollection features = featureSource.getFeatures(Query.ALL);
        SimpleFeatureIterator it = features.features();
        while (it.hasNext()) {
            SimpleFeature f = it.next();
            assertTrue(f.getAttribute("geo") instanceof Geometry);
            assertTrue(f.getAttribute("geo2") instanceof Geometry);
            assertTrue(f.getAttribute("geo3") instanceof Geometry);
        }
    }

    String coreName(SolrDataStore dataStore) {
        String[] split = dataStore.getSolrServer().getBaseURL().split("/");
        return split[split.length-1];
    }
}