|
13 | 13 | BASE_URL = 'http://0.0.0.0:8081' |
14 | 14 |
|
15 | 15 |
|
| 16 | +def post_form(url, values, headers=None): |
| 17 | + out = '{}' |
| 18 | + try: |
| 19 | + data = urllib.urlencode(values) |
| 20 | + req = urllib2.Request(url, data) |
| 21 | + if headers: |
| 22 | + for header in headers: |
| 23 | + req.add_header(header, headers[header]) |
| 24 | + response = urllib2.urlopen(req) |
| 25 | + out = response.read() |
| 26 | + rc = response.getcode() |
| 27 | + if VERBOSE: |
| 28 | + print("code: ", rc, " recv: '", out, "'") |
| 29 | + if rc != 500: |
| 30 | + return (rc, json.loads(out)) |
| 31 | + return (rc, False) |
| 32 | + except urllib2.HTTPError as e: |
| 33 | + if e.code == 400: |
| 34 | + out = e.read(); |
| 35 | + if VERBOSE: |
| 36 | + print("code: ", e.code, " recv: '", out, "'") |
| 37 | + return (e.code, json.loads(out)) |
| 38 | + except Exception as e: |
| 39 | + print(traceback.format_exc()) |
| 40 | + return (False, e) |
| 41 | + |
| 42 | + |
16 | 43 | def post(url, data, headers): |
17 | 44 | out = '{}' |
18 | 45 | try: |
@@ -176,15 +203,29 @@ def get_success_pure(url, data, headers): |
176 | 203 | assert(code == 200), 'expected 200' |
177 | 204 | return msg |
178 | 205 |
|
179 | | -def post_success(url, data, headers, print_f=None): |
| 206 | +def post_success(url, data, headers=None, print_f=None): |
180 | 207 | (code, msg) = post(url, data, headers) |
181 | 208 | if print_f: |
182 | 209 | print_f(code, msg) |
183 | 210 | assert(code == 200), 'expected 200' |
184 | 211 | result = get_result(msg) |
185 | 212 | return result |
186 | 213 |
|
187 | | -def post_success_pure(url, data, headers): |
| 214 | +def post_form_success(url, data, headers=None, print_f=None): |
| 215 | + (code, msg) = post_form(url, data, headers) |
| 216 | + if print_f: |
| 217 | + print_f(code, msg) |
| 218 | + assert(code == 200), 'expected 200' |
| 219 | + return msg |
| 220 | + |
| 221 | +def post_form_ec500(url, data, headers=None, print_f=None): |
| 222 | + (code, msg) = post_form(url, data, headers) |
| 223 | + if print_f: |
| 224 | + print_f(code, msg) |
| 225 | + assert(code == 500), 'expected 500' |
| 226 | + return msg |
| 227 | + |
| 228 | +def post_success_pure(url, data, headers=None): |
188 | 229 | (code, msg) = post(url, data, headers) |
189 | 230 | assert(code == 200), 'expected 200' |
190 | 231 | return msg |
|
0 commit comments