Fix tests

ZhengguanLi · Jun 27, 2024 · 03cf36f · 03cf36f
1 parent 848822e
commit 03cf36f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/core/src/main/java/com/alibaba/druid/sql/dialect/postgresql/parser/PGSQLStatementParser.java b/core/src/main/java/com/alibaba/druid/sql/dialect/postgresql/parser/PGSQLStatementParser.java
@@ -506,7 +506,6 @@ private void parseEndDollarQuote(SQLBlockStatement block) {
                 block.setLanguage(lexer.stringVal());
                 acceptIdentifier(block.getLanguage());
             }
-            accept(Token.SEMI);
         }
     }
 

diff --git a/.../src/test/java/com/alibaba/druid/bvt/sql/postgresql/pl/PGGetDiagnosticsStatementTest.java b/.../src/test/java/com/alibaba/druid/bvt/sql/postgresql/pl/PGGetDiagnosticsStatementTest.java
@@ -16,12 +16,13 @@ public void test_0() throws Exception {
 
 		List<SQLStatement> statementList = SQLUtils.parseStatements(sql, JdbcConstants.POSTGRESQL);
 		assertEquals(1, statementList.size());
-		
+
 		String output = SQLUtils.toSQLString(statementList, JdbcConstants.POSTGRESQL);
 		assertEquals("DO $$\n" +
 					 "BEGIN\n" +
 					 "\tGET DIAGNOSTICS n = ROW_COUNT;\n" +
-					 "END $$;",
+					 "END;\n" +
+					 "$$;",
 					 output);
 	}
 }