Application: gvSIG desktop » gvSIG data provider for PostgreSQL

package org.gvsig.postgresql.dal.operations.sql;

import java.util.Collections;

import java.util.List;

import junit.framework.TestCase;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;

import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory;

import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory.TableReference;

import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase;

import org.gvsig.fmap.dal.store.jdbc2.spi.operations.FetchFeatureTypeOperation;

import org.gvsig.postgresql.dal.TestUtils;

import org.gvsig.tools.library.impl.DefaultLibrariesInitializer;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestFetchFeatureType extends TestCase {

  private static final Logger LOGGER = LoggerFactory.getLogger(TestFetchFeatureType.class);

  public TestFetchFeatureType(String testName) {

    super(testName);

  }

  @Override

  protected void setUp() throws Exception {

    super.setUp();

    new DefaultLibrariesInitializer().fullInitialize();

  }

  @Override

  protected void tearDown() throws Exception {

    super.tearDown();

  }

  // TODO add test methods here. The name must begin with 'test'. For example:

  // public void testHello() {}

  public void testFetchFeatureType() throws Exception {

    JDBCHelper helper = TestUtils.getJDBCHelper();

    JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

    OperationsFactory operations = helper.getOperations();

    List<String> expectedSQLs = TestUtils.getSQLs("fetchFeatureType.sql");

    FeatureStore sourceStore = TestUtils.openSourceStore1();

    TableReference table = operations.createTableReference(

            "dbtest", 

            sqlbuilder.default_schema(), 

            "test", 

            null

    );

    FeatureType featureType = sourceStore.getDefaultFeatureType();

    FetchFeatureTypeOperation fetchFeatureType = operations.createFetchFeatureType(

            featureType.getEditable(),

            table,

            Collections.EMPTY_LIST,

            null, // "Geometry" 

            featureType.getDefaultSRS()

    );

    String retrieveFirstRowOfTableSQL = fetchFeatureType.getSQLToRetrieveFirstRowOfTable();

    String retrievePrimaryKeysFromInformationSchemaSQL = fetchFeatureType.getSQLToRetrievePrimaryKeysFromInformationSchema();

    assertEquals("FetchFeatureType first row SQL", expectedSQLs.get(0), retrieveFirstRowOfTableSQL);

    assertEquals("FetchFeatureType get primery keys SQL", expectedSQLs.get(1), retrievePrimaryKeysFromInformationSchemaSQL);

  }

}