Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
liyuanzhong001
DolphinScheduler
提交
3364a8c8
DolphinScheduler
项目概览
liyuanzhong001
/
DolphinScheduler
与 Fork 源项目一致
Fork自
apache / DolphinScheduler
通知
11
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
DolphinScheduler
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
3364a8c8
编写于
2月 08, 2020
作者:
X
xiaochun.liu
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
add script runner ut
上级
fccbcc8d
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
78 addition
and
0 deletion
+78
-0
dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/utils/ScriptRunnerTest.java
...pache/dolphinscheduler/common/utils/ScriptRunnerTest.java
+78
-0
未找到文件。
dolphinscheduler-common/src/test/java/org/apache/dolphinscheduler/common/utils/ScriptRunnerTest.java
0 → 100644
浏览文件 @
3364a8c8
/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package
org.apache.dolphinscheduler.common.utils
;
import
org.junit.Assert
;
import
org.junit.Test
;
import
org.mockito.Mockito
;
import
java.io.StringReader
;
import
java.sql.*
;
public
class
ScriptRunnerTest
{
@Test
public
void
testRunScript
()
{
//connection is null
Exception
exception
=
null
;
ScriptRunner
s
=
new
ScriptRunner
(
null
,
true
,
true
);
try
{
s
.
runScript
(
new
StringReader
(
"select 1"
));
}
catch
(
Exception
e
)
{
exception
=
e
;
}
Assert
.
assertNotNull
(
exception
);
//connect is not null
try
{
Connection
conn
=
Mockito
.
mock
(
Connection
.
class
);
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
s2
=
new
ScriptRunner
(
conn
,
true
,
true
);
s2
.
runScript
(
new
StringReader
(
"select 1;"
));
Mockito
.
verify
(
md
).
getColumnLabel
(
0
);
}
catch
(
Exception
e
)
{
Assert
.
assertNotNull
(
e
);
}
}
@Test
public
void
testRunScriptWithDbName
()
{
//connect is not null
try
{
Connection
conn
=
Mockito
.
mock
(
Connection
.
class
);
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"
);
Mockito
.
verify
(
md
).
getColumnLabel
(
0
);
}
catch
(
Exception
e
)
{
Assert
.
assertNotNull
(
e
);
}
}
}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录