提交 bd10b78e 编写于 作者: X xiaochun.liu

delete dup code

上级 3364a8c8
...@@ -36,6 +36,14 @@ public class ScriptRunnerTest { ...@@ -36,6 +36,14 @@ public class ScriptRunnerTest {
//connect is not null //connect is not null
try { try {
ScriptRunner s2 = getScriptRunner();
s2.runScript(new StringReader("select 1;"));
} catch(Exception e) {
Assert.assertNotNull(e);
}
}
private ScriptRunner getScriptRunner() throws SQLException {
Connection conn = Mockito.mock(Connection.class); Connection conn = Mockito.mock(Connection.class);
Mockito.when(conn.getAutoCommit()).thenReturn(true); Mockito.when(conn.getAutoCommit()).thenReturn(true);
PreparedStatement st = Mockito.mock(PreparedStatement.class); PreparedStatement st = Mockito.mock(PreparedStatement.class);
...@@ -47,30 +55,15 @@ public class ScriptRunnerTest { ...@@ -47,30 +55,15 @@ public class ScriptRunnerTest {
Mockito.when(md.getColumnCount()).thenReturn(1); Mockito.when(md.getColumnCount()).thenReturn(1);
Mockito.when(rs.next()).thenReturn(true, false); Mockito.when(rs.next()).thenReturn(true, false);
ScriptRunner s2 = new ScriptRunner(conn, true, true); ScriptRunner s2 = new ScriptRunner(conn, true, true);
s2.runScript(new StringReader("select 1;"));
Mockito.verify(md).getColumnLabel(0); Mockito.verify(md).getColumnLabel(0);
} catch(Exception e) {
Assert.assertNotNull(e);
}
} }
@Test @Test
public void testRunScriptWithDbName() { public void testRunScriptWithDbName() {
//connect is not null //connect is not null
try { try {
Connection conn = Mockito.mock(Connection.class); ScriptRunner s = getScriptRunner();
Mockito.when(conn.getAutoCommit()).thenReturn(true);
PreparedStatement st = Mockito.mock(PreparedStatement.class);
Mockito.when(conn.createStatement()).thenReturn(st);
ResultSet rs = Mockito.mock(ResultSet.class);
Mockito.when(st.getResultSet()).thenReturn(rs);
ResultSetMetaData md = Mockito.mock(ResultSetMetaData.class);
Mockito.when(rs.getMetaData()).thenReturn(md);
Mockito.when(md.getColumnCount()).thenReturn(1);
Mockito.when(rs.next()).thenReturn(true, false);
ScriptRunner s = new ScriptRunner(conn, true, true);
s.runScript(new StringReader("select 1;"), "test_db"); s.runScript(new StringReader("select 1;"), "test_db");
Mockito.verify(md).getColumnLabel(0);
} catch(Exception e) { } catch(Exception e) {
Assert.assertNotNull(e); Assert.assertNotNull(e);
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册