def test_log_not_logged_in(self, mock_capture):
session_record = Session()
set_session(session_record)
sentry.log_exception()
mock_capture.assert_called_with(extra={
'session': {},
}, )
def test_log_not_logged_in(self, mock_capture):
session_record = Session()
set_session(session_record)
sentry.log_exception()
mock_capture.assert_called_with(
extra={
'session': {},
},
)
def test_log_logged_in(self, mock_capture):
user = UserFactory()
session_record = Session()
session_record.data['auth_user_id'] = user._id
set_session(session_record)
sentry.log_exception()
mock_capture.assert_called_with(extra={
'session': {
'auth_user_id': user._id,
},
}, )
def test_authenticate_two_factor_with_next_url(self):
fake_session = sessions.Session(data={'next_url': '/someendpoint/'})
sessions.set_session(fake_session)
response = authenticate_two_factor(self.user)
assert_true(isinstance(response, BaseResponse))
assert_equal(
response.location,
u'{0}?next=%2Fsomeendpoint%2F'.format(web_url_for('two_factor')))
assert_equal(response.status_code, 302)
def test_authenticate_two_factor_with_next_url(self):
fake_session = sessions.Session(data={'next_url': '/someendpoint/'})
sessions.set_session(fake_session)
response = authenticate_two_factor(self.user)
assert_true(isinstance(response, BaseResponse))
assert_equal(response.location,
u'{0}?next=%2Fsomeendpoint%2F'.format(web_url_for('two_factor'))
)
assert_equal(response.status_code, 302)
def test_log_logged_in(self, mock_capture):
user = UserFactory()
session_record = Session()
session_record.data['auth_user_id'] = user._id
set_session(session_record)
sentry.log_exception()
mock_capture.assert_called_with(
extra={
'session': {
'auth_user_id': user._id,
},
},
)
def test_verify_two_factor_with_valid_code(self):
fake_session = sessions.Session(
data={
'two_factor_auth': {
'auth_user_username': self.user.username,
'auth_user_id': self.user._primary_key,
'auth_user_fullname': self.user.fullname,
}
})
sessions.set_session(fake_session)
response = verify_two_factor(
self.user._id, _valid_code(self.user_settings.totp_secret))
assert_true(isinstance(response, BaseResponse))
assert_equal(response.location, u'/dashboard/')
assert_equal(response.status_code, 302)
def test_verify_two_factor_with_valid_code(self):
fake_session = sessions.Session(data={
'two_factor_auth':{
'auth_user_username': self.user.username,
'auth_user_id': self.user._primary_key,
'auth_user_fullname': self.user.fullname,
}
})
sessions.set_session(fake_session)
response = verify_two_factor(self.user._id,
_valid_code(self.user_settings.totp_secret)
)
assert_true(isinstance(response, BaseResponse))
assert_equal(response.location, u'/dashboard/')
assert_equal(response.status_code, 302)
def setUp(self):
decoratorapp = Flask('decorators')
self.ctx = decoratorapp.test_request_context()
self.ctx.push()
# TODO: Think of something better @sloria @jmcarp
sessions.set_session(Session())
def setUp(self):
decoratorapp = Flask('decorators')
self.ctx = decoratorapp.test_request_context()
self.ctx.push()
# TODO: Think of something better @sloria @jmcarp
sessions.set_session(Session())
