Added integration tests covering async and segmented querying