Skip to content

Commit 38c2577

Browse files
committed
add built in and udf test
1 parent 939dd98 commit 38c2577

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

extensions/spark/kyuubi-spark-authz/src/test/scala/org/apache/kyuubi/plugin/spark/authz/FunctionPrivilegesBuilderSuite.scala

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -193,4 +193,25 @@ class HiveFunctionPrivilegesBuilderSuite extends FunctionPrivilegesBuilderSuite
193193
}
194194
}
195195

196+
test("Built-in and UDF Function Call Query") {
197+
val plan = sql(s"SELECT kyuubi_fun_0('TESTSTRING'), " +
198+
s"kyuubi_fun_0(value)," +
199+
s"abs(key)," +
200+
s"abs(-100)," +
201+
s"lower(value)," +
202+
s"lower('TESTSTRING') " +
203+
s"FROM $reusedTable").queryExecution.analyzed
204+
val (inputs, _, _) = PrivilegesBuilder.buildFunctions(plan, spark)
205+
assert(inputs.size === 2)
206+
inputs.foreach { po =>
207+
assert(po.actionType === PrivilegeObjectActionType.OTHER)
208+
assert(po.privilegeObjectType === PrivilegeObjectType.FUNCTION)
209+
assert(po.dbname startsWith reusedDb.toLowerCase)
210+
assert(po.objectName startsWith functionNamePrefix.toLowerCase)
211+
val accessType = ranger.AccessType(po, QUERY, isInput = true)
212+
assert(accessType === AccessType.SELECT)
213+
}
214+
}
215+
216+
196217
}

0 commit comments

Comments
 (0)