source.dussan.org Git - sonarqube.git/blob

   1 /*
   2  * SonarQube
   3  * Copyright (C) 2009-2023 SonarSource SA
   4  * mailto:info AT sonarsource DOT com
   5  *
   6  * This program is free software; you can redistribute it and/or
   7  * modify it under the terms of the GNU Lesser General Public
   8  * License as published by the Free Software Foundation; either
   9  * version 3 of the License, or (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  14  * Lesser General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU Lesser General Public License
  17  * along with this program; if not, write to the Free Software Foundation,
  18  * Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
  19  */
  20 package org.sonar.server.platform.db.migration.sql;
  21
  22 import org.junit.Test;
  23 import org.sonar.db.dialect.Dialect;
  24 import org.sonar.db.dialect.H2;
  25 import org.sonar.db.dialect.MsSql;
  26 import org.sonar.db.dialect.Oracle;
  27 import org.sonar.db.dialect.PostgreSql;
  28 import org.sonar.server.platform.db.migration.def.BooleanColumnDef;
  29
  30 import static org.assertj.core.api.Assertions.assertThat;
  31 import static org.assertj.core.api.Assertions.assertThatThrownBy;
  32 import static org.sonar.server.platform.db.migration.def.BooleanColumnDef.newBooleanColumnDefBuilder;
  33 import static org.sonar.server.platform.db.migration.def.DecimalColumnDef.newDecimalColumnDefBuilder;
  34 import static org.sonar.server.platform.db.migration.def.VarcharColumnDef.newVarcharColumnDefBuilder;
  35
  36 public class AlterColumnsBuilderTest {
  37
  38   private static final String TABLE_NAME = "issues";
  39
  40   @Test
  41   public void update_columns_on_h2() {
  42     assertThat(createSampleBuilder(new H2()).build())
  43       .containsOnly(
  44         "ALTER TABLE issues ALTER COLUMN value DOUBLE NULL",
  45         "ALTER TABLE issues ALTER COLUMN name VARCHAR (10) NULL");
  46   }
  47
  48   @Test
  49   public void update_not_nullable_column_on_h2() {
  50     assertThat(createNotNullableBuilder(new H2()).build())
  51       .containsOnly("ALTER TABLE issues ALTER COLUMN name VARCHAR (10) NOT NULL");
  52   }
  53
  54   @Test
  55   public void update_columns_on_mssql() {
  56     assertThat(createSampleBuilder(new MsSql()).build())
  57       .containsOnly(
  58         "ALTER TABLE issues ALTER COLUMN value DECIMAL (30,20) NULL",
  59         "ALTER TABLE issues ALTER COLUMN name NVARCHAR (10) NULL");
  60   }
  61
  62   @Test
  63   public void update_not_nullable_column_on_mssql() {
  64     assertThat(createNotNullableBuilder(new MsSql()).build())
  65       .containsOnly("ALTER TABLE issues ALTER COLUMN name NVARCHAR (10) NOT NULL");
  66   }
  67
  68   @Test
  69   public void update_columns_on_postgres() {
  70     assertThat(createSampleBuilder(new PostgreSql()).build())
  71       .containsOnly("ALTER TABLE issues " +
  72         "ALTER COLUMN value TYPE NUMERIC (30,20), ALTER COLUMN value DROP NOT NULL, " +
  73         "ALTER COLUMN name TYPE VARCHAR (10), ALTER COLUMN name DROP NOT NULL");
  74   }
  75
  76   @Test
  77   public void update_not_nullable_column_on_postgres() {
  78     assertThat(createNotNullableBuilder(new PostgreSql()).build())
  79       .containsOnly("ALTER TABLE issues ALTER COLUMN name TYPE VARCHAR (10), ALTER COLUMN name SET NOT NULL");
  80   }
  81
  82   @Test
  83   public void update_columns_on_oracle() {
  84     assertThat(createSampleBuilder(new Oracle()).build())
  85       .containsOnly(
  86         "ALTER TABLE issues MODIFY (value NUMERIC (30,20) NULL)",
  87         "ALTER TABLE issues MODIFY (name VARCHAR2 (10 CHAR) NULL)");
  88   }
  89
  90   @Test
  91   public void update_not_nullable_column_on_oracle() {
  92     assertThat(createNotNullableBuilder(new Oracle()).build())
  93       .containsOnly("ALTER TABLE issues MODIFY (name VARCHAR2 (10 CHAR) NOT NULL)");
  94   }
  95
  96   @Test
  97   public void fail_with_ISE_if_no_column() {
  98     assertThatThrownBy(() -> new AlterColumnsBuilder(new H2(), TABLE_NAME).build())
  99       .isInstanceOf(IllegalStateException.class)
 100       .hasMessage("No column has been defined");
 101   }
 102
 103   /**
 104    * As we want DEFAULT value to be removed from all tables, it is supported
 105    * only on creation of tables and columns, not on alter.
 106    */
 107   @Test
 108   public void updateColumn_throws_IAE_if_default_value_is_defined() {
 109     BooleanColumnDef column = newBooleanColumnDefBuilder()
 110       .setColumnName("enabled")
 111       .setIsNullable(false)
 112       .setDefaultValue(false)
 113       .build();
 114     AlterColumnsBuilder alterColumnsBuilder = new AlterColumnsBuilder(new H2(), TABLE_NAME);
 115
 116     assertThatThrownBy(() -> alterColumnsBuilder.updateColumn(column))
 117       .isInstanceOf(IllegalArgumentException.class)
 118       .hasMessage("Default value is not supported on alter of column 'enabled'");
 119   }
 120
 121   private AlterColumnsBuilder createSampleBuilder(Dialect dialect) {
 122     return new AlterColumnsBuilder(dialect, TABLE_NAME)
 123       .updateColumn(
 124         newDecimalColumnDefBuilder()
 125           .setColumnName("value")
 126           .setPrecision(30)
 127           .setScale(20)
 128           .setIsNullable(true)
 129           .build())
 130       .updateColumn(
 131         newVarcharColumnDefBuilder()
 132           .setColumnName("name")
 133           .setLimit(10)
 134           .setIsNullable(true)
 135           .build());
 136   }
 137
 138   private AlterColumnsBuilder createNotNullableBuilder(Dialect dialect) {
 139     return new AlterColumnsBuilder(dialect, TABLE_NAME)
 140       .updateColumn(
 141         newVarcharColumnDefBuilder()
 142           .setColumnName("name")
 143           .setLimit(10)
 144           .setIsNullable(false)
 145           .build());
 146   }
 147
 148 }