From 71efab1493f28480328f44ae43bce810f8fb870f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=AB=98=E9=93=81?= Date: Sun, 3 Oct 2021 11:29:33 +0800 Subject: [PATCH] add testcase for #4442 --- .../bvt/sql/postgresql/issues/Issue4442.java | 31 +++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 src/test/java/com/alibaba/druid/bvt/sql/postgresql/issues/Issue4442.java diff --git a/src/test/java/com/alibaba/druid/bvt/sql/postgresql/issues/Issue4442.java b/src/test/java/com/alibaba/druid/bvt/sql/postgresql/issues/Issue4442.java new file mode 100644 index 0000000000..05859d6fbf --- /dev/null +++ b/src/test/java/com/alibaba/druid/bvt/sql/postgresql/issues/Issue4442.java @@ -0,0 +1,31 @@ +package com.alibaba.druid.bvt.sql.postgresql.issues; + +import com.alibaba.druid.DbType; +import com.alibaba.druid.sql.SQLUtils; +import com.alibaba.druid.sql.ast.SQLStatement; +import com.alibaba.druid.sql.dialect.mysql.visitor.MySqlSchemaStatVisitor; +import com.alibaba.druid.sql.parser.SQLParserUtils; +import com.alibaba.druid.sql.parser.SQLStatementParser; +import com.alibaba.druid.sql.visitor.SchemaStatVisitor; +import com.alibaba.druid.stat.TableStat; +import org.junit.Test; + +import java.util.Map; + +import static org.junit.Assert.assertFalse; + +public class Issue4442 { + @Test + public void test_idle2() throws Exception { + String sql = "INSERT INTO users (id, level)\n" + + "VALUES (1, 0)\n" + + "ON CONFLICT (id) DO UPDATE\n" + + "SET level = users.level + 1;"; + SQLStatementParser parser = SQLParserUtils.createSQLStatementParser(sql, DbType.postgresql); + SQLStatement statement = parser.parseStatement(); + SchemaStatVisitor visitor = SQLUtils.createSchemaStatVisitor(DbType.postgresql); + statement.accept(visitor); + Map tableMap = visitor.getTables(); + assertFalse(tableMap.isEmpty()); + } +}