fix api client

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__':