Could you add unit tests for auth and capture as well?