SPARK-2277: add UT