avoid using repartition in fetch as it is slow
integration test now can handle session and driver retry