|
@@ -54,10 +54,10 @@ public class UdfFuncMapperTest extends BaseDaoTest {
|
|
*
|
|
*
|
|
* @return UdfFunc
|
|
* @return UdfFunc
|
|
*/
|
|
*/
|
|
- private UdfFunc insertOne() {
|
|
+ private UdfFunc insertOne(String funcName) {
|
|
UdfFunc udfFunc = new UdfFunc();
|
|
UdfFunc udfFunc = new UdfFunc();
|
|
udfFunc.setUserId(1);
|
|
udfFunc.setUserId(1);
|
|
- udfFunc.setFuncName("dolphin_udf_func");
|
|
+ udfFunc.setFuncName(funcName);
|
|
udfFunc.setClassName("org.apache.dolphinscheduler.test.mr");
|
|
udfFunc.setClassName("org.apache.dolphinscheduler.test.mr");
|
|
udfFunc.setType(UdfType.HIVE);
|
|
udfFunc.setType(UdfType.HIVE);
|
|
udfFunc.setResourceId(1);
|
|
udfFunc.setResourceId(1);
|
|
@@ -76,7 +76,7 @@ public class UdfFuncMapperTest extends BaseDaoTest {
|
|
private UdfFunc insertOne(User user) {
|
|
private UdfFunc insertOne(User user) {
|
|
UdfFunc udfFunc = new UdfFunc();
|
|
UdfFunc udfFunc = new UdfFunc();
|
|
udfFunc.setUserId(user.getId());
|
|
udfFunc.setUserId(user.getId());
|
|
- udfFunc.setFuncName("dolphin_udf_func");
|
|
+ udfFunc.setFuncName("dolphin_udf_func" + user.getUserName());
|
|
udfFunc.setClassName("org.apache.dolphinscheduler.test.mr");
|
|
udfFunc.setClassName("org.apache.dolphinscheduler.test.mr");
|
|
udfFunc.setType(UdfType.HIVE);
|
|
udfFunc.setType(UdfType.HIVE);
|
|
udfFunc.setResourceId(1);
|
|
udfFunc.setResourceId(1);
|
|
@@ -164,7 +164,7 @@ public class UdfFuncMapperTest extends BaseDaoTest {
|
|
@Test
|
|
@Test
|
|
public void testUpdate() {
|
|
public void testUpdate() {
|
|
|
|
|
|
- UdfFunc udfFunc = insertOne();
|
|
+ UdfFunc udfFunc = insertOne("func1");
|
|
udfFunc.setResourceName("dolphin_resource_update");
|
|
udfFunc.setResourceName("dolphin_resource_update");
|
|
udfFunc.setResourceId(2);
|
|
udfFunc.setResourceId(2);
|
|
udfFunc.setClassName("org.apache.dolphinscheduler.test.mrUpdate");
|
|
udfFunc.setClassName("org.apache.dolphinscheduler.test.mrUpdate");
|
|
@@ -181,7 +181,7 @@ public class UdfFuncMapperTest extends BaseDaoTest {
|
|
@Test
|
|
@Test
|
|
public void testDelete() {
|
|
public void testDelete() {
|
|
|
|
|
|
- UdfFunc udfFunc = insertOne();
|
|
+ UdfFunc udfFunc = insertOne("func2");
|
|
|
|
|
|
int delete = udfFuncMapper.deleteById(udfFunc.getId());
|
|
int delete = udfFuncMapper.deleteById(udfFunc.getId());
|
|
Assert.assertEquals(delete, 1);
|
|
Assert.assertEquals(delete, 1);
|
|
@@ -193,9 +193,9 @@ public class UdfFuncMapperTest extends BaseDaoTest {
|
|
@Test
|
|
@Test
|
|
public void testQueryUdfByIdStr() {
|
|
public void testQueryUdfByIdStr() {
|
|
|
|
|
|
- UdfFunc udfFunc = insertOne();
|
|
+ UdfFunc udfFunc = insertOne("func3");
|
|
|
|
|
|
- UdfFunc udfFunc1 = insertOne();
|
|
+ UdfFunc udfFunc1 = insertOne("func4");
|
|
Integer[] idArray = new Integer[]{udfFunc.getId(), udfFunc1.getId()};
|
|
Integer[] idArray = new Integer[]{udfFunc.getId(), udfFunc1.getId()};
|
|
|
|
|
|
List<UdfFunc> udfFuncList = udfFuncMapper.queryUdfByIdStr(idArray, "");
|
|
List<UdfFunc> udfFuncList = udfFuncMapper.queryUdfByIdStr(idArray, "");
|