Add tests for auth service