Add capability for querying partner service during test step