adding dbmigration ddlgeneration test for Ebean @DBArray extension.

ebean-orm · Apr 20, 2017 · 3a8f32b · 3a8f32b
1 parent 485e62b
commit 3a8f32b
Show file tree

Hide file tree

Showing 3 changed files with 33 additions and 0 deletions.
diff --git a/src/test/java/io/ebean/dbmigration/ddlgeneration/BaseDdlHandlerTest.java b/src/test/java/io/ebean/dbmigration/ddlgeneration/BaseDdlHandlerTest.java
@@ -5,6 +5,7 @@
 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 @@ private DdlHandler h2Handler() {
   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 void addColumn_nullable_noConstraint() throws Exception {
 
     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 {

diff --git a/src/test/java/io/ebean/dbmigration/ddlgeneration/Helper.java b/src/test/java/io/ebean/dbmigration/ddlgeneration/Helper.java
@@ -55,6 +55,10 @@ public static AddColumn getAlterTableAddColumn() {
     return (AddColumn) alterTableChangeSetChildren.get(0);
   }
 
+  public static AddColumn getAlterTableAddDbArrayColumn() {
+    return (AddColumn) alterTableChangeSetChildren.get(1);
+  }
+
   public static DropColumn getDropColumn() {
     return (DropColumn) changeSetChildren.get(2);
   }

diff --git a/src/test/resources/container/test-alter-table.xml b/src/test/resources/container/test-alter-table.xml
@@ -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>