/*
 *    GeoTools - The Open Source Java GIS Toolkit
 *    http://geotools.org
 * 
 *    (C) 2002-2008, 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.jdbc;

import static org.easymock.EasyMock.expect;
import static org.easymock.classextension.EasyMock.createMock;
import static org.easymock.classextension.EasyMock.createNiceMock;
import static org.easymock.classextension.EasyMock.replay;
import junit.framework.TestCase;

import org.geotools.data.DataStore;
import org.geotools.data.DataUtilities;
import org.geotools.data.DefaultFeatureResults;
import org.geotools.data.DefaultQuery;
import org.geotools.data.FeatureReader;
import org.geotools.data.Transaction;
import org.opengis.feature.simple.SimpleFeature;
import org.opengis.feature.simple.SimpleFeatureType;

public class JDBCFeatureCollectionTest extends TestCase {

    public void testMaxFeatureOptimized() throws Exception {
        DefaultQuery q = new DefaultQuery("roads");
        q.setMaxFeatures(10);
        
        // mock up the feature source so that it'll return a count of 20
        SimpleFeatureType type = DataUtilities.createType("roads", "_=the_geom:Point,FID:String,NAME:String");
        JDBCFeatureSource fs = createMock(JDBCFeatureSource.class);
        expect(fs.getSchema()).andReturn(type).anyTimes();
        expect(fs.count(q, Transaction.AUTO_COMMIT)).andReturn(20);
        replay(fs);
        
        DefaultFeatureResults results = new JDBCFeatureCollection(fs, q);
        assertEquals(10, results.size());
    }
    
    @SuppressWarnings("unchecked")
    public void testMaxfeaturesHandCount() throws Exception {
        DefaultQuery q = new DefaultQuery("roads");
        q.setMaxFeatures(1);

        // mock up the feature source so that it'll return a count of -1 (too
        // expensive) and then will return a reader
         FeatureReader<SimpleFeatureType, SimpleFeature> fr = createNiceMock(FeatureReader.class);
        expect(fr.hasNext()).andReturn(true).times(2).andReturn(false);
        replay(fr);

        DataStore ds = createMock(JDBC1DataStore.class);
        expect(ds.getFeatureReader(q, Transaction.AUTO_COMMIT)).andReturn(fr);
        replay(ds);

        SimpleFeatureType type = DataUtilities.createType("roads",
                "_=the_geom:Point,FID:String,NAME:String");
        JDBCFeatureSource fs = createMock(JDBCFeatureSource.class);
        expect(fs.getSchema()).andReturn(type).anyTimes();
        expect(fs.count(q, Transaction.AUTO_COMMIT)).andReturn(-1);
        expect(fs.getCount(q)).andReturn(-1);
        expect(fs.getDataStore()).andReturn(ds);
        replay(fs);

        JDBCFeatureCollection results = new JDBCFeatureCollection(fs, q);
        assertEquals(1, results.size());
    }
}