Я пишу модульный тест для «функции обновления» Spark.streaming.mapWithState, и мне нужен Mock для параметра «org.apache.spark.streaming.State», но это закрытый класс, компилятор жалуется на исключение: недопустимый наследство от запечатанного класса гос. Как я должен проверить свою функцию?