adding dbmigration ddlgeneration test for Ebean @DbArray extension.

This commit is contained in:
Michael Benz
2017-04-20 14:16:50 +02:00
parent 485e62b5d6
commit 3a8f32b173
3 changed files with 33 additions and 0 deletions
@@ -5,6 +5,7 @@ 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.config.dbplatform.sqlserver.SqlServerPlatform;
import io.ebean.dbmigration.migration.ChangeSet;
import io.ebean.dbmigration.model.CurrentModel;
import io.ebeaninternal.api.SpiEbeanServer;
@@ -25,6 +26,10 @@ public class BaseDdlHandlerTest extends BaseTestCase {
private DdlHandler postgresHandler() {
return new PostgresPlatform().createDdlHandler(serverConfig);
}
private DdlHandler sqlserverHandler() {
return new SqlServerPlatform().createDdlHandler(serverConfig);
}
@Test
public void addColumn_nullable_noConstraint() throws Exception {
@@ -36,7 +41,27 @@ public class BaseDdlHandlerTest extends BaseTestCase {
assertThat(write.apply().getBuffer()).isEqualTo("alter table foo add column added_to_foo varchar(20);\n\n");
}
/**
* Test the functionality of the Ebean {@literal @}DbArray extension during DDL generation.
*/
@Test
public void addColumn_dbarray() throws Exception {
DdlWrite write = new DdlWrite();
DdlHandler postgresHandler = postgresHandler();
postgresHandler.generate(write, Helper.getAlterTableAddDbArrayColumn());
assertThat(write.apply().getBuffer()).isEqualTo("alter table foo add column dbarray_added_to_foo varchar[];\n\n");
write = new DdlWrite();
DdlHandler sqlserverHandler = sqlserverHandler();
sqlserverHandler.generate(write, Helper.getAlterTableAddDbArrayColumn());
//MS SqlServer does not support @DbArray like PostgreSQL and should just use JSON in a VARCHAR type.
assertThat(write.apply().getBuffer()).isEqualTo("alter table foo add dbarray_added_to_foo varchar;\n\n");
}
@Test
public void addColumn_withForeignKey() throws Exception {
@@ -55,6 +55,10 @@ public class Helper {
return (AddColumn) alterTableChangeSetChildren.get(0);
}
public static AddColumn getAlterTableAddDbArrayColumn() {
return (AddColumn) alterTableChangeSetChildren.get(1);
}
public static DropColumn getDropColumn() {
return (DropColumn) changeSetChildren.get(2);
}
@@ -8,6 +8,10 @@
<column name="some_id" type="integer" references="bar.id" foreignKeyName="fk_foo_some_id" foreignKeyIndex="idx_foo_some_id"/>
</addColumn>
<addColumn tableName="foo">
<column name="dbarray_added_to_foo" type="varchar[]"/>
</addColumn>
</changeSet>
</migration>