def test_emergency_reset_get_authenticated(request, redirect, read_token_file,
i18n_url, super_fn):
request.user.is_authenticated = True
route = mod.EmergencyReset()
route.get()
read_token_file.assert_called_once_with()
redirect.assert_called_once_with(i18n_url.return_value)
assert not super_fn.called
def test_emergency_reset_get_not_authenticated(request, redirect,
read_token_file, super_fn):
request.user.is_authenticated = False
route = mod.EmergencyReset()
route.get()
read_token_file.assert_called_once_with()
assert not redirect.called
assert super_fn.called
def test_emergency_reset_recreate_user(request, get_reset_token, create, exts):
route = mod.EmergencyReset()
route.recreate_user('usr', 'pwd')
create.assert_called_once_with('usr',
'pwd',
is_superuser=True,
db=exts.databases.librarian,
reset_token=get_reset_token.return_value)
def test_emergency_reset_clear_auth_databases(exts, request):
route = mod.EmergencyReset()
route.clear_auth_databases()
db = exts.databases.librarian
db.Delete.assert_any_call('users')
db.execute.assert_any_call(db.Delete.return_value)
db.Delete.assert_any_call('sessions')
db.execute.assert_any_call(db.Delete.return_value)
def test_emergency_reset_get_reset_token(request, generate_reset_token):
route = mod.EmergencyReset()
# test case of GET request
request.method = 'GET'
assert route.get_reset_token() == generate_reset_token.return_value
# test case of POST request
request.method = 'POST'
assert route.get_reset_token() == request.params.get.return_value
def test_emergency_reset_form_valid(request, clear_auth_databases,
recreate_user, template, lazy_gettext):
route = mod.EmergencyReset()
route.form = mock.Mock()
route.form.processed_data = {'username': '******', 'password1': 'pwd'}
resp = route.form_valid()
clear_auth_databases.assert_called_once_with()
recreate_user.assert_called_once_with('usr', 'pwd')
assert isinstance(resp, mod.EmergencyReset.HTTPResponse)
def test_emergency_reset_read_token_file(open_fn, isfile, abort, request):
isfile.return_value = True
# set up mocked empty file object
mocked_file = mock.Mock()
mocked_file.read.return_value = 'token'
ctx_manager = mock.MagicMock()
ctx_manager.__enter__.return_value = mocked_file
open_fn.return_value = ctx_manager
# perform test
route = mod.EmergencyReset()
assert route.read_token_file() == 'token'
assert not abort.called
def test_emergency_reset_read_token_file_empty(open_fn, isfile, abort,
request):
isfile.return_value = True
# set up mocked empty file object
mocked_file = mock.Mock()
mocked_file.read.return_value = ''
ctx_manager = mock.MagicMock()
ctx_manager.__enter__.return_value = mocked_file
open_fn.return_value = ctx_manager
# perform test
route = mod.EmergencyReset()
with mock.patch.object(route, 'config') as config:
route.read_token_file()
abort.assert_called_once_with(404)
open_fn.assert_called_once_with(config.get.return_value, 'r')
def test_emergency_reset_read_token_file_not_found(isfile, abort, request):
isfile.return_value = False
route = mod.EmergencyReset()
route.read_token_file()
abort.assert_called_once_with(404)
def test_emergency_reset_get_context(request, get_reset_token):
route = mod.EmergencyReset()
ctx = route.get_context()
assert ctx['reset_token'] == get_reset_token.return_value
