def wrap(*args, **kwargs):
if is_valid_admin(request.user):
return func(*args, **kwargs)
else:
return respond({'msg': 'Invalid administrator!'},
Status.UNAUTHORIZED)
def test_updated_admin(self, app_context, set_repo_get, admin_data):
admin_data['updated_at'] = datetime.now() - timedelta(hours=12)
user = Admin(**admin_data)
assert not is_valid_admin(user)
def test_outdated_admin(self, app_context, set_repo_get, admin_data):
admin_data['auth_at'] = datetime.now() - timedelta(
days=current_app.config['AUTH_VALID_PERIOD_IN_DAY'] + 1)
user = Admin(**admin_data)
assert not is_valid_admin(user)
def test_anonymous(self, app_context, set_repo_get, anonymous_data):
user = Anonymous(**anonymous_data)
assert not is_valid_admin(user)
def test_empty_user(self, app_context, set_repo_get):
assert not is_valid_admin(None)
def test_valid_admin(self, app_context, set_repo_get, admin_data):
user = Admin(**admin_data)
assert is_valid_admin(user)