diff --git a/tests/__init__.py b/tests/__init__.py index ed15d963d..501fe2a89 100644 --- a/tests/__init__.py +++ b/tests/__init__.py @@ -14,6 +14,7 @@ class TestClient(FlaskClient): def login(self, user, mocker=None, service=None): # Skipping authentication here and just log them in with self.session_transaction() as session: + session['current_session_id'] = user.current_session_id session['user_id'] = user.id if mocker: mocker.patch('app.user_api_client.get_user', return_value=user) @@ -63,6 +64,7 @@ def user_json( platform_admin=False, current_session_id='1234', organisations=[], + services=[] ): return { @@ -78,7 +80,8 @@ def user_json( 'max_failed_login_count': max_failed_login_count, 'platform_admin': platform_admin, 'current_session_id': current_session_id, - 'organisations': organisations + 'organisations': organisations, + 'services': services or permissions.keys() }