|
| 1 | +/*- |
| 2 | + * #%L |
| 3 | + * JSQLParser library |
| 4 | + * %% |
| 5 | + * Copyright (C) 2004 - 2026 JSQLParser |
| 6 | + * %% |
| 7 | + * Dual licensed under GNU LGPL 2.1 or Apache License 2.0 |
| 8 | + * #L% |
| 9 | + */ |
| 10 | +package net.sf.jsqlparser.statement.alter; |
| 11 | + |
| 12 | +import static net.sf.jsqlparser.test.TestUtils.assertSqlCanBeParsedAndDeparsed; |
| 13 | +import static org.assertj.core.api.Assertions.assertThat; |
| 14 | +import static org.junit.jupiter.api.Assertions.assertDoesNotThrow; |
| 15 | +import static org.junit.jupiter.api.Assertions.assertEquals; |
| 16 | +import static org.junit.jupiter.api.Assertions.assertInstanceOf; |
| 17 | +import static org.junit.jupiter.api.Assertions.assertNull; |
| 18 | +import static org.junit.jupiter.api.Assertions.assertThrows; |
| 19 | +import static org.junit.jupiter.api.Assertions.assertTrue; |
| 20 | + |
| 21 | +import java.util.Arrays; |
| 22 | +import java.util.List; |
| 23 | +import java.util.function.Consumer; |
| 24 | +import net.sf.jsqlparser.JSQLParserException; |
| 25 | +import net.sf.jsqlparser.parser.AbstractJSqlParser.Dialect; |
| 26 | +import net.sf.jsqlparser.parser.CCJSqlParser; |
| 27 | +import net.sf.jsqlparser.parser.CCJSqlParserUtil; |
| 28 | +import net.sf.jsqlparser.statement.create.table.CheckConstraint; |
| 29 | +import net.sf.jsqlparser.statement.create.table.ConstraintAttributes; |
| 30 | +import net.sf.jsqlparser.statement.create.table.CreateTable; |
| 31 | +import net.sf.jsqlparser.statement.create.table.ForeignKeyConstraint; |
| 32 | +import net.sf.jsqlparser.statement.create.table.NamedConstraint; |
| 33 | +import net.sf.jsqlparser.util.TablesNamesFinder; |
| 34 | +import org.junit.jupiter.api.Test; |
| 35 | +import org.junit.jupiter.params.ParameterizedTest; |
| 36 | +import org.junit.jupiter.params.provider.ValueSource; |
| 37 | + |
| 38 | +/** |
| 39 | + * ALTER TABLE ADD accepts the table constraint forms of CREATE TABLE in every dialect, not only |
| 40 | + * with the MySQL and PostgreSQL dialects. |
| 41 | + */ |
| 42 | +class AlterAddTableConstraintTest { |
| 43 | + private static final List<Dialect> DIALECTS = |
| 44 | + Arrays.asList(null, Dialect.ANSI_SQL, Dialect.ORACLE, Dialect.SQLSERVER, Dialect.H2); |
| 45 | + |
| 46 | + @ParameterizedTest |
| 47 | + @ValueSource(strings = { |
| 48 | + "CHECK (id > 0)", |
| 49 | + "CHECK (id > 0) NOT VALID", |
| 50 | + "CONSTRAINT ck CHECK (id > 0) NOT VALID", |
| 51 | + "FOREIGN KEY (id) REFERENCES parent (id) DEFERRABLE INITIALLY DEFERRED", |
| 52 | + "CONSTRAINT fk FOREIGN KEY (id) REFERENCES parent (id) NOT VALID", |
| 53 | + "CONSTRAINT pk PRIMARY KEY (id) DEFERRABLE INITIALLY DEFERRED", |
| 54 | + "CONSTRAINT uq UNIQUE (id) DEFERRABLE INITIALLY IMMEDIATE", |
| 55 | + "CONSTRAINT ex EXCLUDE USING gist (id WITH =)"}) |
| 56 | + void createAndAlterExposeTheSameConstraintInEveryDialect(String definition) |
| 57 | + throws JSQLParserException { |
| 58 | + for (Dialect dialect : DIALECTS) { |
| 59 | + Consumer<CCJSqlParser> configuration = parser -> { |
| 60 | + if (dialect != null) { |
| 61 | + parser.withDialect(dialect); |
| 62 | + } |
| 63 | + }; |
| 64 | + CreateTable create = (CreateTable) CCJSqlParserUtil |
| 65 | + .parse("CREATE TABLE t (id INT, " + definition + ")", configuration); |
| 66 | + Alter alter = (Alter) assertSqlCanBeParsedAndDeparsed( |
| 67 | + "ALTER TABLE t ADD " + definition + ", ADD COLUMN other INT", true, |
| 68 | + configuration); |
| 69 | + NamedConstraint expected = create.getTableConstraints().get(0); |
| 70 | + NamedConstraint actual = alter.getAlterExpressions().get(0).getConstraint(); |
| 71 | + assertEquals(expected.getClass(), actual.getClass(), String.valueOf(dialect)); |
| 72 | + assertEquals(expected.toString(), actual.toString(), String.valueOf(dialect)); |
| 73 | + assertEquals(2, alter.getAlterExpressions().size()); |
| 74 | + } |
| 75 | + } |
| 76 | + |
| 77 | + @Test |
| 78 | + void attributesAndReferencesAreStructuredWithoutADialect() throws JSQLParserException { |
| 79 | + String sql = "ALTER TABLE t ADD CONSTRAINT fk FOREIGN KEY (id) REFERENCES parent (id) " |
| 80 | + + "DEFERRABLE INITIALLY DEFERRED"; |
| 81 | + Alter alter = (Alter) CCJSqlParserUtil.parse(sql); |
| 82 | + ForeignKeyConstraint foreignKey = assertInstanceOf(ForeignKeyConstraint.class, |
| 83 | + alter.getAlterExpressions().get(0).getConstraint()); |
| 84 | + ConstraintAttributes attributes = foreignKey.getConstraintAttributes(); |
| 85 | + assertEquals(Boolean.TRUE, attributes.getDeferrable()); |
| 86 | + assertEquals(ConstraintAttributes.Initially.DEFERRED, attributes.getInitially()); |
| 87 | + assertThat(TablesNamesFinder.findTables(sql)).containsExactlyInAnyOrder("t", "parent"); |
| 88 | + |
| 89 | + CheckConstraint check = assertInstanceOf(CheckConstraint.class, |
| 90 | + ((Alter) CCJSqlParserUtil.parse("ALTER TABLE t ADD CHECK (id > 0) NOT VALID")) |
| 91 | + .getAlterExpressions().get(0).getConstraint()); |
| 92 | + assertNull(check.getName()); |
| 93 | + assertTrue(check.getConstraintAttributes().isNotValid()); |
| 94 | + } |
| 95 | + |
| 96 | + @Test |
| 97 | + void formsThatAlreadyParsedKeepTheGenericRepresentation() throws JSQLParserException { |
| 98 | + AlterExpression deferrable = ((Alter) CCJSqlParserUtil |
| 99 | + .parse("ALTER TABLE t ADD CONSTRAINT pk PRIMARY KEY (id) DEFERRABLE")) |
| 100 | + .getAlterExpressions().get(0); |
| 101 | + assertThat(deferrable.getConstraints()).hasSize(1); |
| 102 | + assertInstanceOf(DeferrableConstraint.class, deferrable.getConstraints().get(0)); |
| 103 | + |
| 104 | + AlterExpression unique = ((Alter) CCJSqlParserUtil |
| 105 | + .parse("ALTER TABLE t ADD CONSTRAINT UNIQUE KEY uk (id)")) |
| 106 | + .getAlterExpressions().get(0); |
| 107 | + assertEquals("UNIQUE KEY", unique.getConstraintType()); |
| 108 | + } |
| 109 | + |
| 110 | + @ParameterizedTest |
| 111 | + @ValueSource(strings = {"ALTER TABLE t ADD CONSTRAINT nn NOT NULL id", |
| 112 | + "ALTER TABLE t ADD CONSTRAINT uk UNIQUE USING INDEX t_idx"}) |
| 113 | + void postgreSqlOnlyFormsStillRequireThePostgreSqlDialect(String sql) { |
| 114 | + assertThrows(JSQLParserException.class, () -> CCJSqlParserUtil.parse(sql)); |
| 115 | + assertDoesNotThrow( |
| 116 | + () -> CCJSqlParserUtil.parse(sql, |
| 117 | + parser -> parser.withDialect(Dialect.POSTGRESQL))); |
| 118 | + } |
| 119 | + |
| 120 | + @Test |
| 121 | + void enforcedPrimaryKeyIsRejected() { |
| 122 | + JSQLParserException exception = assertThrows(JSQLParserException.class, |
| 123 | + () -> CCJSqlParserUtil |
| 124 | + .parse("ALTER TABLE t ADD CONSTRAINT pk PRIMARY KEY (id) ENFORCED")); |
| 125 | + assertThat(exception).hasStackTraceContaining( |
| 126 | + "ENFORCED is supported only for CHECK and foreign key constraints"); |
| 127 | + } |
| 128 | +} |
0 commit comments