Revision f3787696 snf-django-lib/snf_django/utils/testing.py
| b/snf-django-lib/snf_django/utils/testing.py | ||
|---|---|---|
| 116 | 116 |
return inner |
| 117 | 117 |
return wrapper |
| 118 | 118 |
|
| 119 |
serial = 0 |
|
| 120 |
|
|
| 119 | 121 |
|
| 120 | 122 |
@contextmanager |
| 121 | 123 |
def astakos_user(user): |
| ... | ... | |
| 131 | 133 |
get_token.return_value = "DummyToken" |
| 132 | 134 |
with patch('astakosclient.AstakosClient.get_user_info') as m:
|
| 133 | 135 |
m.return_value = {"uuid": user}
|
| 134 |
yield |
|
| 135 |
|
|
| 136 |
serial = 0 |
|
| 136 |
with patch('astakosclient.AstakosClient.get_quotas') as m2:
|
|
| 137 |
m2.return_value = {
|
|
| 138 |
"system": {
|
|
| 139 |
"pithos.diskspace": {
|
|
| 140 |
"usage": 0, |
|
| 141 |
"limit": 1073741824, |
|
| 142 |
"pending": 0 |
|
| 143 |
} |
|
| 144 |
} |
|
| 145 |
} |
|
| 146 |
with patch('astakosclient.AstakosClient.issue_one_commission') as m3:
|
|
| 147 |
serials = [] |
|
| 148 |
append = serials.append |
|
| 149 |
|
|
| 150 |
def get_serial(*args, **kwargs): |
|
| 151 |
global serial |
|
| 152 |
serial += 1 |
|
| 153 |
append(serial) |
|
| 154 |
return serial |
|
| 155 |
|
|
| 156 |
m3.side_effect = get_serial |
|
| 157 |
with patch('astakosclient.AstakosClient.resolve_commissions') as m4:
|
|
| 158 |
m4.return_value = {'accepted': serials,
|
|
| 159 |
'rejected': [], |
|
| 160 |
'failed': []} |
|
| 161 |
with patch('astakosclient.AstakosClient.get_usernames') as m5:
|
|
| 162 |
|
|
| 163 |
def get_usernames(*args, **kwargs): |
|
| 164 |
uuids = args[-1] |
|
| 165 |
return dict((uuid, uuid) for uuid in uuids) |
|
| 166 |
|
|
| 167 |
m5.side_effect = get_usernames |
|
| 168 |
yield |
|
| 137 | 169 |
|
| 138 | 170 |
|
| 139 | 171 |
@contextmanager |
Also available in: Unified diff