Add tests for login/register