#919 - io.ebean package initial

src/test/java/io/ebean/dbmigration/ddlgeneration/BaseDdlHandlerTest.java

@@ -0,0 +1,145 @@
+package io.ebean.dbmigration.ddlgeneration;
+
+import io.ebean.BaseTestCase;
+import io.ebean.Ebean;
+import io.ebean.config.ServerConfig;
+import io.ebean.config.dbplatform.h2.H2Platform;
+import io.ebean.config.dbplatform.postgres.PostgresPlatform;
+import io.ebean.dbmigration.migration.ChangeSet;
+import io.ebean.dbmigration.model.CurrentModel;
+import io.ebeaninternal.api.SpiEbeanServer;
+import org.junit.Ignore;
+import org.junit.Test;
+
+import static org.assertj.core.api.Assertions.assertThat;
+
+
+public class BaseDdlHandlerTest extends BaseTestCase {
+
+  ServerConfig serverConfig = new ServerConfig();
+
+  private DdlHandler h2Handler() {
+    return new H2Platform().createDdlHandler(serverConfig);
+  }
+
+  private DdlHandler postgresHandler() {
+    return new PostgresPlatform().createDdlHandler(serverConfig);
+  }
+
+  @Test
+  public void addColumn_nullable_noConstraint() throws Exception {
+
+    DdlWrite write = new DdlWrite();
+
+    DdlHandler handler = h2Handler();
+    handler.generate(write, Helper.getAddColumn());
+
+    assertThat(write.apply().getBuffer()).isEqualTo("alter table foo add column added_to_foo varchar(20);\n\n");
+  }
+
+
+  @Test
+  public void addColumn_withForeignKey() throws Exception {
+
+    DdlWrite write = new DdlWrite();
+
+    DdlHandler handler = h2Handler();
+    handler.generate(write, Helper.getAlterTableAddColumn());
+
+    String buffer = write.apply().getBuffer();
+    assertThat(buffer).contains("alter table foo add column some_id integer;");
+
+    String fkBuffer = write.applyForeignKeys().getBuffer();
+    assertThat(fkBuffer).contains("alter table foo add constraint fk_foo_some_id foreign key (some_id) references bar (id) on delete restrict on update restrict;");
+    assertThat(fkBuffer).contains("create index idx_foo_some_id on foo (some_id);");
+    assertThat(write.dropAll().getBuffer()).isEqualTo("");
+  }
+
+  @Test
+  public void dropColumn() throws Exception {
+
+    DdlWrite write = new DdlWrite();
+    DdlHandler handler = h2Handler();
+
+    handler.generate(write, Helper.getDropColumn());
+
+    assertThat(write.apply().getBuffer()).isEqualTo("alter table foo drop column col2;\n\n");
+    assertThat(write.dropAll().getBuffer()).isEqualTo("");
+  }
+
+
+  @Test
+  public void createTable() throws Exception {
+
+    DdlWrite write = new DdlWrite();
+    DdlHandler handler = h2Handler();
+
+    handler.generate(write, Helper.getCreateTable());
+
+    String createTableDDL = Helper.asText(this, "/assert/create-table.txt");
+
+    assertThat(write.apply().getBuffer()).isEqualTo(createTableDDL);
+    assertThat(write.dropAll().getBuffer().trim()).isEqualTo("drop table if exists foo;");
+  }
+
+  @Test
+  public void generateChangeSet() throws Exception {
+
+    DdlWrite write = new DdlWrite();
+    DdlHandler handler = h2Handler();
+
+    handler.generate(write, Helper.getChangeSet());
+
+    String apply = Helper.asText(this, "/assert/BaseDdlHandlerTest/baseApply.sql");
+    String rollbackLast = Helper.asText(this, "/assert/BaseDdlHandlerTest/baseDropAll.sql");
+
+    assertThat(write.apply().getBuffer()).isEqualTo(apply);
+    assertThat(write.dropAll().getBuffer()).isEqualTo(rollbackLast);
+  }
+
+
+  @Ignore
+  @Test
+  public void generateChangeSetFromModel() throws Exception {
+
+    SpiEbeanServer defaultServer = (SpiEbeanServer) Ebean.getDefaultServer();
+
+    ChangeSet createChangeSet = new CurrentModel(defaultServer).getChangeSet();
+
+    DdlWrite write = new DdlWrite();
+
+    DdlHandler handler = h2Handler();
+    handler.generate(write, createChangeSet);
+
+    String apply = Helper.asText(this, "/assert/changeset-apply.txt");
+    String rollbackLast = Helper.asText(this, "/assert/changeset-dropAll.txt");
+
+    assertThat(write.apply().getBuffer()).isEqualTo(apply);
+    assertThat(write.dropAll().getBuffer()).isEqualTo(rollbackLast);
+  }
+
+  @Ignore
+  @Test
+  public void generateChangeSetFromModel_given_postgresTypes() throws Exception {
+
+    SpiEbeanServer defaultServer = (SpiEbeanServer) Ebean.getDefaultServer();
+
+    ChangeSet createChangeSet = new CurrentModel(defaultServer).getChangeSet();
+
+    DdlWrite write = new DdlWrite();
+
+    DdlHandler handler = postgresHandler();
+    handler.generate(write, createChangeSet);
+
+    String apply = Helper.asText(this, "/assert/changeset-pg-apply.sql");
+    String applyLast = Helper.asText(this, "/assert/changeset-pg-applyLast.sql");
+    String rollbackFirst = Helper.asText(this, "/assert/changeset-pg-rollbackFirst.sql");
+    String rollbackLast = Helper.asText(this, "/assert/changeset-pg-rollbackLast.sql");
+
+    assertThat(write.apply().getBuffer()).isEqualTo(apply);
+    assertThat(write.applyForeignKeys().getBuffer()).isEqualTo(applyLast);
+    assertThat(write.dropAllForeignKeys().getBuffer()).isEqualTo(rollbackFirst);
+    assertThat(write.dropAll().getBuffer()).isEqualTo(rollbackLast);
+  }
+
+}