Add tests for registration and login