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.DALLocator;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

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.PerformChangesOperation;

import org.gvsig.postgresql.dal.TestUtils;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestPerformChanges extends TestCase {

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

  public TestPerformChanges(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 testPerformChanges() throws Exception {

    JDBCHelper helper = TestUtils.getJDBCHelper();

    JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

    OperationsFactory operations = helper.getOperations();

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

    FeatureStore sourceStore = TestUtils.openSourceStore1();

    TableReference table = operations.createTableReference(

            "dbtest", 

            sqlbuilder.default_schema(), 

            "test", 

            null

    );

    FeatureType featureType = sourceStore.getDefaultFeatureType();

    FeatureTypeChanged featureTypeChanged = new FeatureTypeChanged() {

      @Override

      public FeatureType getSource() {

        EditableFeatureType ft = DALLocator.getDataManager().createFeatureType();

        EditableFeatureAttributeDescriptor id = ft.add("ID", DataTypes.INT);

        id.copyFrom(featureType.getAttributeDescriptor("ID"));

        return ft;

      }

      @Override

      public FeatureType getTarget() {

        return featureType;

      }

    };

    List<FeatureTypeChanged> featureTypesChangeds = Collections.singletonList(featureTypeChanged);

    PerformChangesOperation performChanges = operations.createPerformChanges(

            table, 

            sourceStore.getDefaultFeatureType(),

            null, 

            null, 

            null, 

            featureTypesChangeds.iterator()

    );

    String insertSQL = performChanges.getInsertSQL();

    String deleteSQL = performChanges.getDeleteSQL();

    String updateSQL = performChanges.getUpdateSQL();

    List<String> updateTableSQLs = performChanges.getUpdateTableSQLs();

    assertEquals("Insert SQL", expectedSQLs.get(0), insertSQL);

    assertEquals("Delete SQL", expectedSQLs.get(1), deleteSQL);

    assertEquals("Update SQL", expectedSQLs.get(2), updateSQL);

    assertEquals("UpdateTable num. SQLs", expectedSQLs.size()-3, updateTableSQLs.size());

    for (int i = 0; i < updateTableSQLs.size(); i++) {

      assertEquals("UpdateTable SQL "+i, expectedSQLs.get(i+3), updateTableSQLs.get(i));

    }

    // FIXME: El ALTER TABLE de una columna geometria no funciona

  }

}