def test_auth_success(self): logger.debug('in test_auth_success') adminid = 'admin_{}'.format(str(uuid.uuid4())) LoginModel.create( username=adminid, password='******', role='admin' ) adm_login = { 'username':adminid, 'password':'******', } success_token = mock.Mock(return_value='admintoken') auth.get_token = success_token self.assertEqual(auth.authentication('admin', adm_login)[0], 1) self.assertEqual(auth.authentication('admin', adm_login)[1], 'admintoken')
def test_auth_fail(self): logger.debug('in test_auth_success fail') adminid = 'admin_{}'.format(str(uuid.uuid4())) LoginModel.create( username=adminid, password='******', role='admin' ) adm_login = { 'username':adminid, 'password':'******', } fail_token = mock.Mock(return_value='admintoken') auth.get_token = fail_token self.assertEqual(auth.authentication('admin', adm_login)[0], 0) self.assertEqual(auth.authentication('admin', adm_login)[1], 'auth failed, password not match') self.assertEqual(auth.authentication('doctor', adm_login)[0], 0) self.assertEqual(auth.authentication('doctor', adm_login)[1], 'auth failed')