Update tests to use new helpers in vumi

This needs praekelt/vumi#647 to land before it'll work.