|
9 | 9 | */ |
10 | 10 | package net.sf.jsqlparser.statement.create; |
11 | 11 |
|
| 12 | +import static net.sf.jsqlparser.test.TestUtils.assertSqlCanBeParsedAndDeparsed; |
12 | 13 | import static net.sf.jsqlparser.util.validation.ValidationTestAsserts.validateNotAllowed; |
13 | 14 | import static org.assertj.core.api.Assertions.assertThat; |
14 | 15 | import static org.junit.jupiter.api.Assertions.assertEquals; |
| 16 | +import static org.junit.jupiter.api.Assertions.assertInstanceOf; |
15 | 17 | import static org.junit.jupiter.api.Assertions.assertNotNull; |
16 | 18 | import static org.junit.jupiter.api.Assertions.assertNull; |
17 | 19 | import static org.junit.jupiter.api.Assertions.assertThrows; |
18 | | - |
19 | 20 | import java.util.ArrayList; |
20 | 21 | import java.util.Arrays; |
21 | 22 | import java.util.List; |
| 23 | +import java.util.function.Consumer; |
| 24 | +import java.util.stream.Stream; |
22 | 25 | import net.sf.jsqlparser.JSQLParserException; |
| 26 | +import net.sf.jsqlparser.expression.CollateExpression; |
23 | 27 | import net.sf.jsqlparser.expression.Expression; |
24 | 28 | import net.sf.jsqlparser.expression.ExpressionVisitorAdapter; |
25 | 29 | import net.sf.jsqlparser.expression.LongValue; |
| 30 | +import net.sf.jsqlparser.expression.StringValue; |
| 31 | +import net.sf.jsqlparser.expression.operators.relational.ParenthesedExpressionList; |
| 32 | +import net.sf.jsqlparser.parser.AbstractJSqlParser.Dialect; |
| 33 | +import net.sf.jsqlparser.parser.CCJSqlParser; |
26 | 34 | import net.sf.jsqlparser.parser.CCJSqlParserUtil; |
27 | 35 | import net.sf.jsqlparser.parser.feature.Feature; |
28 | | -import net.sf.jsqlparser.util.validation.feature.FeaturesAllowed; |
29 | | -import net.sf.jsqlparser.parser.AbstractJSqlParser.Dialect; |
30 | 36 | import net.sf.jsqlparser.statement.Statement; |
31 | 37 | import net.sf.jsqlparser.statement.StatementVisitorAdapter; |
32 | 38 | import net.sf.jsqlparser.statement.alter.Alter; |
| 39 | +import net.sf.jsqlparser.statement.alter.AlterDomain; |
| 40 | +import net.sf.jsqlparser.statement.alter.AlterRelation; |
| 41 | +import net.sf.jsqlparser.statement.create.domain.CreateDomain; |
33 | 42 | import net.sf.jsqlparser.statement.create.table.ColumnDefinition; |
34 | 43 | import net.sf.jsqlparser.statement.create.table.ColumnOption; |
35 | 44 | import net.sf.jsqlparser.statement.create.table.CreateTable; |
36 | 45 | import net.sf.jsqlparser.statement.select.SelectVisitorAdapter; |
37 | 46 | import net.sf.jsqlparser.util.deparser.ExpressionDeParser; |
38 | 47 | import net.sf.jsqlparser.util.deparser.SelectDeParser; |
39 | 48 | import net.sf.jsqlparser.util.deparser.StatementDeParser; |
| 49 | +import net.sf.jsqlparser.util.validation.feature.FeaturesAllowed; |
40 | 50 | import org.junit.jupiter.api.Test; |
41 | 51 | import org.junit.jupiter.params.ParameterizedTest; |
| 52 | +import org.junit.jupiter.params.provider.Arguments; |
| 53 | +import org.junit.jupiter.params.provider.MethodSource; |
42 | 54 | import org.junit.jupiter.params.provider.ValueSource; |
43 | 55 |
|
44 | 56 | class ColumnDefaultExpressionTest { |
@@ -161,6 +173,94 @@ void neighboringAndLegacyColumnOptionsStillRoundTrip(String sql) throws JSQLPars |
161 | 173 | assertRoundTrip(CCJSqlParserUtil.parse(sql)); |
162 | 174 | } |
163 | 175 |
|
| 176 | + @ParameterizedTest |
| 177 | + @ValueSource( |
| 178 | + strings = {"CREATE TABLE t (c VARCHAR(30) ", "ALTER TABLE t ADD COLUMN c VARCHAR(30) ", |
| 179 | + "ALTER TABLE t MODIFY COLUMN c VARCHAR(30) ", |
| 180 | + "ALTER TABLE t CHANGE COLUMN old_c c VARCHAR(30) "}) |
| 181 | + void collateAfterDefaultIsAColumnAttribute(String prefix) throws JSQLParserException { |
| 182 | + String sql = prefix + "NOT NULL DEFAULT 'N' COLLATE utf8mb4_unicode_ci COMMENT 'flag'" |
| 183 | + + (prefix.startsWith("CREATE") ? ")" : ""); |
| 184 | + List<Consumer<CCJSqlParser>> configurations = Arrays.asList(parser -> { |
| 185 | + }, parser -> parser.withDialect(Dialect.MYSQL)); |
| 186 | + for (Consumer<CCJSqlParser> configuration : configurations) { |
| 187 | + Statement statement = assertSqlCanBeParsedAndDeparsed(sql, true, configuration); |
| 188 | + ColumnDefinition column = firstColumn(statement); |
| 189 | + assertThat(column.getColumnOptions()).extracting(ColumnOption::getKind) |
| 190 | + .containsExactly(ColumnOption.Kind.NULLABILITY, ColumnOption.Kind.DEFAULT, |
| 191 | + ColumnOption.Kind.COLLATE, ColumnOption.Kind.COMMENT); |
| 192 | + assertInstanceOf(StringValue.class, defaultOption(column).getDefaultExpression()); |
| 193 | + assertEquals("utf8mb4_unicode_ci", column.getColumnOptions().get(2).getCollation()); |
| 194 | + assertThat(column.getColumnSpecs()).containsExactly("NOT", "NULL", "DEFAULT", "'N'", |
| 195 | + "COLLATE", "utf8mb4_unicode_ci", "COMMENT", "'flag'"); |
| 196 | + assertRoundTrip(statement); |
| 197 | + } |
| 198 | + } |
| 199 | + |
| 200 | + static Stream<Arguments> defaultValuesBeforeCollate() { |
| 201 | + return Stream.of(Arguments.of(Dialect.MYSQL, "_utf8mb4'N'", "utf8mb4_bin"), |
| 202 | + Arguments.of(Dialect.MYSQL, "'N'", "'utf8mb4_bin'"), |
| 203 | + Arguments.of(Dialect.MYSQL, "1", "utf8mb4_bin"), |
| 204 | + Arguments.of(Dialect.MYSQL, "-1", "utf8mb4_bin"), |
| 205 | + Arguments.of(Dialect.POSTGRESQL, "'a' || 'b'", "\"C\""), |
| 206 | + Arguments.of(Dialect.POSTGRESQL, "lower('A' COLLATE \"C\")", "\"C\""), |
| 207 | + Arguments.of(Dialect.POSTGRESQL, "CASE WHEN true THEN 'a' END", "\"C\""), |
| 208 | + Arguments.of(Dialect.POSTGRESQL, "'x'::text", "\"C\""), |
| 209 | + Arguments.of(Dialect.POSTGRESQL, "'x'", "pg_catalog.\"default\"")); |
| 210 | + } |
| 211 | + |
| 212 | + @ParameterizedTest |
| 213 | + @MethodSource("defaultValuesBeforeCollate") |
| 214 | + void defaultValueEndsBeforeTopLevelCollate(Dialect dialect, String value, String collation) |
| 215 | + throws JSQLParserException { |
| 216 | + String definition = " VARCHAR(30) DEFAULT " + value + " COLLATE " + collation + " NOT NULL"; |
| 217 | + String alter = dialect == Dialect.MYSQL ? "ALTER TABLE t MODIFY COLUMN c" |
| 218 | + : "ALTER TABLE t ADD COLUMN c"; |
| 219 | + for (String sql : Arrays.asList("CREATE TABLE t (c" + definition + ")", |
| 220 | + alter + definition)) { |
| 221 | + ColumnDefinition column = firstColumn(assertSqlCanBeParsedAndDeparsed(sql, true, |
| 222 | + parser -> parser.withDialect(dialect))); |
| 223 | + assertThat(column.getColumnOptions()).extracting(ColumnOption::getKind) |
| 224 | + .containsExactly(ColumnOption.Kind.DEFAULT, ColumnOption.Kind.COLLATE, |
| 225 | + ColumnOption.Kind.NULLABILITY); |
| 226 | + assertEquals(value, defaultOption(column).getDefaultExpression().toString()); |
| 227 | + assertEquals(collation, column.getColumnOptions().get(1).getCollation()); |
| 228 | + } |
| 229 | + } |
| 230 | + |
| 231 | + @Test |
| 232 | + void parenthesizedCollateStaysInTheDefaultValue() throws JSQLParserException { |
| 233 | + ColumnDefinition column = firstColumn(assertSqlCanBeParsedAndDeparsed( |
| 234 | + "ALTER TABLE t MODIFY COLUMN c VARCHAR(3) DEFAULT ('N' COLLATE utf8mb4_bin) NOT NULL", |
| 235 | + true, parser -> parser.withDialect(Dialect.MYSQL))); |
| 236 | + assertThat(column.getColumnOptions()).extracting(ColumnOption::getKind) |
| 237 | + .containsExactly(ColumnOption.Kind.DEFAULT, ColumnOption.Kind.NULLABILITY); |
| 238 | + Expression value = defaultOption(column).getDefaultExpression(); |
| 239 | + assertInstanceOf(ParenthesedExpressionList.class, value); |
| 240 | + assertInstanceOf(CollateExpression.class, ((ParenthesedExpressionList<?>) value).get(0)); |
| 241 | + } |
| 242 | + |
| 243 | + @Test |
| 244 | + void setDefaultAndDomainDefaultsKeepTheExpressionCollate() throws JSQLParserException { |
| 245 | + // PostgreSQL's SET DEFAULT takes an a_expr, which includes COLLATE. CREATE DOMAIN is |
| 246 | + // unchanged. |
| 247 | + Consumer<CCJSqlParser> postgres = parser -> parser.withDialect(Dialect.POSTGRESQL); |
| 248 | + Alter table = (Alter) assertSqlCanBeParsedAndDeparsed( |
| 249 | + "ALTER TABLE t ALTER COLUMN c SET DEFAULT 'x' COLLATE \"C\"", true, postgres); |
| 250 | + assertInstanceOf(CollateExpression.class, table.getAlterExpressions().get(0) |
| 251 | + .getColumnDefaults().get(0).getDefaultExpression()); |
| 252 | + AlterRelation view = (AlterRelation) assertSqlCanBeParsedAndDeparsed( |
| 253 | + "ALTER VIEW v ALTER COLUMN c SET DEFAULT 'x' COLLATE \"C\"", true, postgres); |
| 254 | + assertInstanceOf(CollateExpression.class, |
| 255 | + view.getActions().get(0).getDefaultExpression()); |
| 256 | + AlterDomain alterDomain = (AlterDomain) assertSqlCanBeParsedAndDeparsed( |
| 257 | + "ALTER DOMAIN d SET DEFAULT 'x' COLLATE \"C\"", true, postgres); |
| 258 | + assertInstanceOf(CollateExpression.class, alterDomain.getDefaultExpression()); |
| 259 | + CreateDomain createDomain = (CreateDomain) assertSqlCanBeParsedAndDeparsed( |
| 260 | + "CREATE DOMAIN d AS text DEFAULT 'x' COLLATE \"C\"", true, postgres); |
| 261 | + assertInstanceOf(CollateExpression.class, createDomain.getDefaultExpression()); |
| 262 | + } |
| 263 | + |
164 | 264 | @ParameterizedTest |
165 | 265 | @ValueSource(strings = {"CREATE TABLE t (a INT DEFAULT (1 + 2", |
166 | 266 | "CREATE TABLE t (a INT DEFAULT (1 +))", |
|
0 commit comments