Add initial unit test for login