Add: OK response with custom headers

1 files changed, 5 insertions, 4 deletions

diff --git a/ick2/responses.py b/ick2/responses.py
index 1ceae52..f0c5c8f 100644
--- a/ick2/responses.py
+++ b/ick2/responses.py
@@ -25,10 +25,11 @@ def response(status_code, body, headers):  # pragma: no cover
     return apifw.Response(obj)
 
 
-def OK(body):  # pragma: no cover
-    headers = {
-        'Content-Type': 'application/json',
-    }
+def OK(body, headers=None):  # pragma: no cover
+    if headers is None:
+        headers = {}
+    elif 'Content-Type' not in headers:
+        headers['Content-Type'] = 'application/json'
     return response(apifw.HTTP_OK, body, headers)