tests: update mock objects to match recent changes