Replace ScalaTest with Mockito for ZK mock

ScalaTest only works with traits and classes without arguments.