Add test around login