25 lines
656 B
Python
25 lines
656 B
Python
|
import json
|
||
|
|
import re
|
||
|
|
from typing import Dict, Optional
|
||
|
|
|
||
|
|
from werkzeug.test import TestResponse
|
||
|
|
|
||
|
|
|
||
|
|
def assert_errored_response(
|
||
|
|
response: TestResponse,
|
||
|
|
status_code: int,
|
||
|
|
error_message: Optional[str] = None,
|
||
|
|
status: Optional[str] = 'error',
|
||
|
|
match: Optional[str] = None,
|
||
|
|
) -> Dict:
|
||
|
|
assert response.content_type == 'application/json'
|
||
|
|
assert response.status_code == status_code
|
||
|
|
|
||
|
|
data = json.loads(response.data.decode())
|
||
|
|
assert status in data['status']
|
||
|
|
if error_message is not None:
|
||
|
|
assert error_message in data['message']
|
||
|
|
if match is not None:
|
||
|
|
assert re.match(match, data['message'])
|
||
|
|
return data
|