Add tests for UserController