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')
