Application: gvSIG desktop

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

import java.util.List;

import junit.framework.TestCase;

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

import org.gvsig.sqlite.dal.TestUtils;

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

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

public class TestCreateTable extends TestCase {

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

  public TestCreateTable(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 testCreateTable() throws Exception {

      try {

        JDBCHelper helper = TestUtils.createJDBCHelper();

        JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

        OperationsFactory operations = helper.getOperations();

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

        FeatureStore sourceStore = TestUtils.openSourceStore1();

        TableReference table = operations.createTableReference(

                "dbtest", 

                sqlbuilder.default_schema(), 

                "test", 

                null

        );

        FeatureType featureType = sourceStore.getDefaultFeatureType();

        CreateTableOperation createTable = operations.createTable(

                table,

                featureType,

                null,

                null

        );

        List<String> SQLs = createTable.getSQLs();

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

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

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

        }

      } catch(Exception ex) {

          LOGGER.warn("Error",ex);

          throw ex;

      }

  }

  public void testCreateTableWithDefaultDynamicText() throws Exception {

      try {

        JDBCHelper helper = TestUtils.createJDBCHelper();

        JDBCSQLBuilderBase sqlbuilder = helper.createSQLBuilder();

        OperationsFactory operations = helper.getOperations();

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

        FeatureStore sourceStore = TestUtils.openSourceStore1();

        TableReference table = operations.createTableReference(

                "dbtest", 

                sqlbuilder.default_schema(), 

                "test", 

                null

        );

        FeatureType featureType = sourceStore.getDefaultFeatureType();

        EditableFeatureType eft = featureType.getEditable();

        EditableFeatureAttributeDescriptor date_attr = (EditableFeatureAttributeDescriptor) eft.getAttributeDescriptor("Date");

        date_attr.setDefaultValue("'<%=CURRENT_TIMESTAMP()%>'");

        CreateTableOperation createTable = operations.createTable(

                table,

                eft,

                null,

                null

        );

        List<String> SQLs = createTable.getSQLs();

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

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

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

        }

      } catch(Exception ex) {

          LOGGER.warn("Error",ex);

          throw ex;

      }

  }

}