Application: gvSIG desktop

package org.gvsig.fmap.dal.store.h2.operations.sql;

import java.util.ArrayList;

import java.util.List;

import junit.framework.TestCase;

import org.gvsig.fmap.dal.DataTypes;

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

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

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

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

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

import org.gvsig.fmap.dal.store.h2.TestUtils;

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

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestAppend extends TestCase {

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

  public TestAppend(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 testAppend() throws Exception {

    JDBCHelper helper = TestUtils.createJDBCHelper();

    JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

    OperationsFactory operations = helper.getOperations();

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

    FeatureStore sourceStore = TestUtils.openSourceStore1();

    TableReference table = operations.createTableReference(

            "dbtest", 

            sqlbuilder.default_schema(), 

            "test", 

            null

    );

    FeatureType featureType = sourceStore.getDefaultFeatureType();

    AppendOperation append = operations.createAppend(

            table,

            featureType

    );

    append.begin(FeatureStore.SUBMODE_NONE);

    List<String> SQLs = new ArrayList<>();

    SQLs.addAll(append.getPreviousSQLs());

    SQLs.add( append.getSQL());

    SQLs.addAll(append.getPostSQLs());

    assertEquals("Append num SQLs", expectedSQLs.size(), SQLs.size());

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

      assertEquals("Append SQL "+i, expectedSQLs.get(i), SQLs.get(i));

    }

    for (Feature sourceFeature : sourceStore.getFeatureSet()) {

      List<Object> parametersSQL = append.getSQLParameters(TestUtils.getFeatureProvider(sourceFeature));

      List<Object> expectedParameters = this.getExpectedparameters(sourceFeature);

      assertEquals("Append num. parameters", parametersSQL.size(), expectedParameters.size());

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

        assertEquals("Append parameter "+i, expectedParameters.get(i), parametersSQL.get(i));

      }

    }

    append.end();

  }

  private List<Object> getExpectedparameters(Feature sourceFeature) {

    FeatureType ft = sourceFeature.getType();

    List<Object> values = new ArrayList<>();

    for (FeatureAttributeDescriptor attr : ft) {

      Object v = sourceFeature.get(attr.getIndex());

      values.add(v);

      if( attr.getType()==DataTypes.GEOMETRY ) {

        String srs = attr.getSRS().getAbrev();

        values.add(Integer.parseInt(srs.split(":")[1]));

      }

    }

    return values;

  }

}