diff options
| -rw-r--r-- | src/home/api/web_api_client.py | 2 | ||||
| -rwxr-xr-x | test/test_api.py | 14 |
2 files changed, 12 insertions, 4 deletions
diff --git a/src/home/api/web_api_client.py b/src/home/api/web_api_client.py index 299bb6e..6f8ff27 100644 --- a/src/home/api/web_api_client.py +++ b/src/home/api/web_api_client.py @@ -164,7 +164,7 @@ class WebAPIClient: timeout=self.timeout, **kwargs) - if r.headers['content-type'] != 'application/json': + if not r.headers['content-type'].startswith('application/json'): raise ApiResponseError(r.status_code, 'TypeError', 'content-type is not application/json') data = json.loads(r.text) diff --git a/test/test_api.py b/test/test_api.py index 959b2b3..1f6361c 100755 --- a/test/test_api.py +++ b/test/test_api.py @@ -1,7 +1,15 @@ #!/usr/bin/env python3 -from home.api import WebAPIClient -from home.api.types import BotType -from home.config import config +import sys +import os.path +sys.path.extend([ + os.path.realpath( + os.path.join(os.path.dirname(os.path.join(__file__)), '..') + ) +]) + +from src.home.api import WebAPIClient +from src.home.api.types import BotType +from src.home.config import config if __name__ == '__main__': |