fix tests that rely on sessions

redash/handlers/users.py

@@ -169,8 +169,11 @@ class UserResource(BaseResource):
         @after_this_request
         def update_identity_cookie(response):
             session['user_id'] = user.get_id()
+
             remember_token = request.cookies.get('remember_token')
-            response.set_cookie('remember_token', re.sub('.*\\|', user.get_id() + '|', remember_token))
+            if remember_token:
+                response.set_cookie('remember_token', re.sub('.*\\|', user.get_id() + '|', remember_token))
+
             return response
 
         require_admin_or_owner(user_id)
@@ -192,7 +195,7 @@ class UserResource(BaseResource):
 
         if 'groups' in params and not self.current_user.has_permission('admin'):
             abort(403, message="Must be admin to change groups membership.")
-        
+
         if 'email' in params:
             _, domain = params['email'].split('@', 1)
 

tests/__init__.py

@@ -26,7 +26,7 @@ logging.getLogger("metrics").setLevel("ERROR")
 
 def authenticate_request(c, user):
     with c.session_transaction() as sess:
-        sess['user_id'] = user.id
+        sess['user_id'] = user.get_id()
 
 
 @contextmanager