0001-toulouse_smart-manage-status-code-in-mock_response-5.patch
tests/test_toulouse_smart.py | ||
---|---|---|
46 | 46 |
@httmock.urlmatch() |
47 | 47 |
def error(url, request): |
48 | 48 |
assert False, 'request to %s' % (url,) |
49 | 49 | |
50 | 50 |
@functools.wraps(func) |
51 | 51 |
def wrapper(*args, **kwargs): |
52 | 52 |
handlers = [] |
53 | 53 |
for row in path_contents: |
54 |
path, content = row |
|
54 |
path, content = row[:2] |
|
55 |
if len(row) == 3: |
|
56 |
status_code = row[2] |
|
57 |
else: |
|
58 |
status_code = 200 |
|
55 | 59 | |
56 | 60 |
@httmock.urlmatch(path=path) |
57 | 61 |
def handler(url, request): |
58 |
return content
|
|
62 |
return httmock.response(status_code, content)
|
|
59 | 63 | |
60 | 64 |
handlers.append(handler) |
61 | 65 |
handlers.append(error) |
62 | 66 | |
63 | 67 |
with httmock.HTTMock(*handlers): |
64 | 68 |
return func(*args, **kwargs) |
65 | 69 | |
66 | 70 |
return wrapper |
67 |
- |