Skip to content

H
h2database
提交 1f0773ea authored 作者: Evgenij Ryazanov's avatar Evgenij Ryazanov
浏览文件
操作

Improve TestKeywords

上级 12627199
隐藏空白字符变更
内嵌 并排
正在显示 包含 89 行增加 和 27 行删除
h2/src/test/org/h2/test/unit/TestKeywords.java
浏览文件 @ 1f0773ea
...@@ -9,7 +9,8 @@ import java.sql.Connection; ...@@ -9,7 +9,8 @@ import java.sql.Connection;
import java.sql.DriverManager; import java.sql.DriverManager;
import java.sql.ResultSet; import java.sql.ResultSet;
import java.sql.Statement; import java.sql.Statement;
import java.util.HashSet; import java.util.HashMap;
import java.util.Map.Entry;
import org.h2.command.Parser; import org.h2.command.Parser;
import org.h2.test.TestBase; import org.h2.test.TestBase;
...@@ -25,6 +26,14 @@ import org.objectweb.asm.Opcodes; ...@@ -25,6 +26,14 @@ import org.objectweb.asm.Opcodes;
*/ */
public class TestKeywords extends TestBase { public class TestKeywords extends TestBase {
private enum TokenType {
IDENTIFIER,
KEYWORD,
CONTEXT_SENSITIVE_KEYWORD;
}
/** /**
* Run just this test. * Run just this test.
* *
...@@ -37,13 +46,13 @@ public class TestKeywords extends TestBase { ...@@ -37,13 +46,13 @@ public class TestKeywords extends TestBase {
@Override @Override
public void test() throws Exception { public void test() throws Exception {
final HashSet<String> set = new HashSet<>(); final HashMap<String, TokenType> tokens = new HashMap<>();
ClassReader r = new ClassReader(Parser.class.getResourceAsStream("Parser.class")); ClassReader r = new ClassReader(Parser.class.getResourceAsStream("Parser.class"));
r.accept(new ClassVisitor(Opcodes.ASM7) { r.accept(new ClassVisitor(Opcodes.ASM7) {
@Override @Override
public FieldVisitor visitField(int access, String name, String descriptor, String signature, public FieldVisitor visitField(int access, String name, String descriptor, String signature,
Object value) { Object value) {
add(set, value); add(value);
return null; return null;
} }
...@@ -53,18 +62,18 @@ public class TestKeywords extends TestBase { ...@@ -53,18 +62,18 @@ public class TestKeywords extends TestBase {
return new MethodVisitor(Opcodes.ASM7) { return new MethodVisitor(Opcodes.ASM7) {
@Override @Override
public void visitLdcInsn(Object value) { public void visitLdcInsn(Object value) {
add(set, value); add(value);
} }
}; };
} }
void add(HashSet<String> set, Object value) { void add(Object value) {
if (!(value instanceof String)) { if (!(value instanceof String)) {
return; return;
} }
String s = (String) value; String s = (String) value;
int l = s.length(); int l = s.length();
if (l == 0 || ParserUtil.getSaveTokenType(s, false, 0, l, true) != ParserUtil.IDENTIFIER) { if (l == 0) {
return; return;
} }
for (int i = 0; i < l; i++) { for (int i = 0; i < l; i++) {
...@@ -73,38 +82,91 @@ public class TestKeywords extends TestBase { ...@@ -73,38 +82,91 @@ public class TestKeywords extends TestBase {
return; return;
} }
} }
set.add(s); final TokenType type;
switch (ParserUtil.getSaveTokenType(s, false, 0, l, true)) {
case ParserUtil.IDENTIFIER:
type = TokenType.IDENTIFIER;
break;
case ParserUtil.KEYWORD:
type = TokenType.CONTEXT_SENSITIVE_KEYWORD;
break;
default:
type = TokenType.KEYWORD;
}
tokens.put(s, type);
} }
}, ClassReader.SKIP_DEBUG | ClassReader.SKIP_FRAMES); }, ClassReader.SKIP_DEBUG | ClassReader.SKIP_FRAMES);
try (Connection conn = DriverManager.getConnection("jdbc:h2:mem:keywords")) { try (Connection conn = DriverManager.getConnection("jdbc:h2:mem:keywords")) {
Statement stat = conn.createStatement(); Statement stat = conn.createStatement();
for (String s : set) { for (Entry<String, TokenType> entry : tokens.entrySet()) {
String s = entry.getKey();
TokenType type = entry.getValue();
Throwable exception1 = null, exception2 = null;
try { try {
stat.execute("CREATE TABLE " + s + '(' + s + " INT)"); stat.execute("CREATE TABLE " + s + '(' + s + " INT)");
stat.execute("INSERT INTO " + s + '(' + s + ") VALUES (10)"); stat.execute("INSERT INTO " + s + '(' + s + ") VALUES (10)");
try (ResultSet rs = stat.executeQuery("SELECT " + s + " FROM " + s)) { } catch (Throwable t) {
assertTrue(rs.next()); exception1 = t;
assertEquals(10, rs.getInt(1)); }
assertFalse(rs.next()); if (exception1 == null) {
try {
try (ResultSet rs = stat.executeQuery("SELECT " + s + " FROM " + s)) {
assertTrue(rs.next());
assertEquals(10, rs.getInt(1));
assertFalse(rs.next());
}
try (ResultSet rs = stat.executeQuery("SELECT SUM(" + s + ") " + s + " FROM " + s + ' ' + s)) {
assertTrue(rs.next());
assertEquals(10, rs.getInt(1));
assertFalse(rs.next());
assertEquals(s, rs.getMetaData().getColumnLabel(1));
}
stat.execute("DROP TABLE " + s);
stat.execute("CREATE TABLE TEST(" + s + " VARCHAR) AS VALUES '-'");
String str;
try (ResultSet rs = stat.executeQuery("SELECT TRIM(" + s + " FROM '--a--') FROM TEST")) {
assertTrue(rs.next());
str = rs.getString(1);
}
stat.execute("DROP TABLE TEST");
try (ResultSet rs = stat
.executeQuery("SELECT ROW_NUMBER() OVER(" + s + ") WINDOW " + s + " AS ()")) {
}
if (!"a".equals(str)) {
exception2 = new AssertionError();
}
} catch (Throwable t) {
exception2 = t;
stat.execute("DROP TABLE IF EXISTS TEST");
} }
try (ResultSet rs = stat.executeQuery("SELECT SUM(" + s + ") " + s + " FROM " + s + ' ' + s)) { }
assertTrue(rs.next()); switch (type) {
assertEquals(10, rs.getInt(1)); case IDENTIFIER:
assertFalse(rs.next()); if (exception1 != null) {
assertEquals(s, rs.getMetaData().getColumnLabel(1)); throw new AssertionError(s + " must be a keyword.", exception1);
} }
stat.execute("DROP TABLE " + s); if (exception2 != null) {
stat.execute("CREATE TABLE TEST(" + s + " VARCHAR) AS VALUES '-'"); throw new AssertionError(s + " must be a context-sensitive keyword.", exception2);
try (ResultSet rs = stat.executeQuery("SELECT TRIM(" + s + " FROM '--a--') FROM TEST")) {
assertTrue(rs.next());
assertEquals("a", rs.getString(1));
} }
stat.execute("DROP TABLE TEST"); break;
try (ResultSet rs = stat case KEYWORD:
.executeQuery("SELECT ROW_NUMBER() OVER(" + s + ") WINDOW " + s + " AS ()")) { if (exception1 == null && exception2 == null) {
throw new AssertionError(s + " may be removed from a list of keywords.");
} }
} catch (Throwable t) { if (exception1 == null) {
throw new AssertionError(s + " cannot be used as identifier.", t); throw new AssertionError(s + " may be a context-sensitive keyword.");
}
break;
case CONTEXT_SENSITIVE_KEYWORD:
if (exception1 != null) {
throw new AssertionError(s + " must be a keyword.", exception1);
}
if (exception2 == null) {
throw new AssertionError(s + " may be removed from a list of context-sensitive keywords.");
}
break;
default:
fail();
} }
} }
} }
......
Markdown 格式
0% 或
您添加了 0 人 到此讨论。请谨慎行事。
请先完成此评论的编辑！
请 注册 或者 后发表评论