Refactor auth tests