add unit test for spark driver