我有一份很有活力的工作。我想为这个spark作业编写单元测试，其形式如下：

Class(): Unit = {

    ......    
    spark.sql(query(depends on input arguments))

    }

这是一个配置单元分区查询。
我想为此编写一个单元测试，并且一直在使用 argumentCaptor 。我不确定要模拟什么、如何模拟以及用什么替换模拟功能。我对scala和spark都是新手。有人能帮忙吗？