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()); + } +}