Revision 45c0bcf8 snf-django-lib/snf_django/utils/testing.py
b/snf-django-lib/snf_django/utils/testing.py | ||
---|---|---|
132 | 132 |
""" |
133 | 133 |
with patch("snf_django.lib.api.get_token") as get_token: |
134 | 134 |
get_token.return_value = "DummyToken" |
135 |
with patch('astakosclient.AstakosClient.get_endpoints') as m1: |
|
136 |
m1.return_value = { |
|
137 |
'access': { |
|
138 |
'serviceCatalog': [{ |
|
139 |
'endpoints': [{'SNF:uiURL': 'ui_url', |
|
140 |
'publicURL': 'accounts_url', |
|
141 |
'region': 'default', |
|
142 |
'versionId': 'v1.0'}], |
|
143 |
'name': 'astakos_account', |
|
144 |
'type': 'account'}] |
|
145 |
} |
|
146 |
} |
|
147 |
with patch('astakosclient.AstakosClient.get_user_info') as m2: |
|
148 |
m2.return_value = {"uuid": text.udec(user, 'utf8')} |
|
149 |
with patch('astakosclient.AstakosClient.get_quotas') as m3: |
|
150 |
m3.return_value = { |
|
151 |
"system": { |
|
152 |
"pithos.diskspace": { |
|
153 |
"usage": 0, |
|
154 |
"limit": 1073741824, # 1GB |
|
155 |
"pending": 0 |
|
156 |
} |
|
135 |
with patch('astakosclient.AstakosClient.get_user_info') as m2: |
|
136 |
m2.return_value = {"uuid": text.udec(user, 'utf8')} |
|
137 |
with patch('astakosclient.AstakosClient.get_quotas') as m3: |
|
138 |
m3.return_value = { |
|
139 |
"system": { |
|
140 |
"pithos.diskspace": { |
|
141 |
"usage": 0, |
|
142 |
"limit": 1073741824, # 1GB |
|
143 |
"pending": 0 |
|
157 | 144 |
} |
158 | 145 |
} |
159 |
issue_fun = \ |
|
160 |
"astakosclient.AstakosClient.issue_one_commission" |
|
161 |
with patch(issue_fun) as m3: |
|
162 |
serials = [] |
|
163 |
append = serials.append |
|
164 |
|
|
165 |
def get_serial(*args, **kwargs): |
|
166 |
global serial |
|
167 |
serial += 1 |
|
168 |
append(serial) |
|
169 |
return serial |
|
170 |
|
|
171 |
m3.side_effect = get_serial |
|
172 |
resolv_fun = \ |
|
173 |
'astakosclient.AstakosClient.resolve_commissions' |
|
174 |
with patch(resolv_fun) as m4: |
|
175 |
m4.return_value = {'accepted': serials, |
|
176 |
'rejected': [], |
|
177 |
'failed': []} |
|
178 |
users_fun = \ |
|
179 |
'astakosclient.AstakosClient.get_usernames' |
|
180 |
with patch(users_fun) as m5: |
|
181 |
|
|
182 |
def get_usernames(*args, **kwargs): |
|
183 |
uuids = args[-1] |
|
184 |
return dict((uuid, uuid) for uuid in uuids) |
|
185 |
|
|
186 |
m5.side_effect = get_usernames |
|
187 |
yield |
|
146 |
} |
|
147 |
issue_fun = \ |
|
148 |
"astakosclient.AstakosClient.issue_one_commission" |
|
149 |
with patch(issue_fun) as m3: |
|
150 |
serials = [] |
|
151 |
append = serials.append |
|
152 |
|
|
153 |
def get_serial(*args, **kwargs): |
|
154 |
global serial |
|
155 |
serial += 1 |
|
156 |
append(serial) |
|
157 |
return serial |
|
158 |
|
|
159 |
m3.side_effect = get_serial |
|
160 |
resolv_fun = \ |
|
161 |
'astakosclient.AstakosClient.resolve_commissions' |
|
162 |
with patch(resolv_fun) as m4: |
|
163 |
m4.return_value = {'accepted': serials, |
|
164 |
'rejected': [], |
|
165 |
'failed': []} |
|
166 |
users_fun = \ |
|
167 |
'astakosclient.AstakosClient.get_usernames' |
|
168 |
with patch(users_fun) as m5: |
|
169 |
|
|
170 |
def get_usernames(*args, **kwargs): |
|
171 |
uuids = args[-1] |
|
172 |
return dict((uuid, uuid) for uuid in uuids) |
|
173 |
|
|
174 |
m5.side_effect = get_usernames |
|
175 |
yield |
|
188 | 176 |
|
189 | 177 |
|
190 | 178 |
serial = 0 |
Also available in: Unified diff