|
1
5 package org.h2.test.db;
6
7 import java.io.InputStream  ;
8 import java.io.InputStreamReader ;
9 import java.io.LineNumberReader ;
10 import java.sql.Connection ;
11 import java.sql.ResultSet ;
12
13 import org.h2.test.TestBase;
14 import org.h2.util.ScriptReader;
15
16 public class TestScriptSimple extends TestBase {
17
18 private Connection conn;
19 public void test() throws Exception {
20 if(config.memory || config.big || config.networked) {
21 return;
22 }
23 deleteDb("scriptSimple");
24 reconnect();
25 String infile = "org/h2/test/testSimple.in.txt";
26 InputStream is = getClass().getClassLoader().getResourceAsStream(infile);
27 LineNumberReader lineReader = new LineNumberReader (new InputStreamReader (is, "Cp1252"));
28 ScriptReader reader = new ScriptReader(lineReader);
29 while(true) {
30 String sql = reader.readStatement();
31 if(sql == null) {
32 break;
33 }
34 sql = sql.trim();
35 if("@reconnect".equals(sql)) {
37 reconnect();
38 } else if(sql.length() == 0) {
39 } else if(sql.startsWith("select")) {
41 ResultSet rs = conn.createStatement().executeQuery(sql);
42 while(rs.next()) {
43 String expected = reader.readStatement().trim();
44 String got = "> " + rs.getString(1);
45 check(expected, got);
46 }
47 } else {
48 conn.createStatement().execute(sql);
49 }
50 }
51 conn.close();
52 }
53
54 private void reconnect() throws Exception {
55 if(conn != null) {
56 conn.close();
57 }
58 conn = getConnection("scriptSimple");
59 }
60
61
62 }
63 |
Popular Tags |
Java API By Example, From Geeks To Geeks.