Revision f12bcb3d snf-astakos-app/astakos/test/stress.py

b/snf-astakos-app/astakos/test/stress.py
47 47
os.environ['DJANGO_SETTINGS_MODULE'] = 'synnefo.settings'
48 48

  
49 49
from astakos.im.models import AstakosUser
50
from astakos.im.functions import get_related_project_id
50
from astakos.im.functions import get_related_project_id, ProjectError
51 51
from astakos.im import quotas
52 52
from views import submit, approve, join, leave
53 53
from snf_django.lib.db.transaction import commit_on_success_strict
54
from django.core.exceptions import PermissionDenied
55 54

  
56 55
USERS = {}
57 56
PROJECTS = {}
......
125 124
                        % (prefix, now, prec))
126 125
            app_id = submit(name, user_id, prec)
127 126
            prec = app_id
128
        except PermissionDenied as e:
127
        except ProjectError as e:
129 128
            logger.info('Limit reached')
130 129
        except Exception as e:
131 130
            logger.exception(e)
......
167 166
            logger.info('%s%s: user %s joining project %s'
168 167
                        % (prefix, now, user_id, proj_id))
169 168
            join(proj_id, user_id)
170
        except PermissionDenied as e:
169
        except ProjectError as e:
171 170
            logger.info('Membership already exists')
172 171
        except Exception as e:
173 172
            logger.exception(e)
......
177 176
            logger.info('%s%s: user %s leaving project %s'
178 177
                        % (prefix, now, user_id, proj_id))
179 178
            leave(proj_id, user_id)
180
        except IOError as e:
179
        except ProjectError as e:
181 180
            logger.info('No such membership')
182 181
        except Exception as e:
183 182
            logger.exception(e)

Also available in: Unified diff