From 2d5d5265d4f293ccfcc207a4905c7536b4d14f3d Mon Sep 17 00:00:00 2001
From: Michal Holub <holub.michal@gmail.com>
Date: Thu, 8 Mar 2018 11:15:37 +0100
Subject: [PATCH] response.content no longer breaks if bytes is returned
---
.gitignore | 1 +
tests/test_middleware.py | 6 ++++--
2 files changed, 5 insertions(+), 2 deletions(-)
diff --git a/.gitignore b/.gitignore
index f82852a..1e0408d 100644
--- a/.gitignore
+++ b/.gitignore
@@ -5,3 +5,4 @@ __pycache__/
*.egg-info/
.coverage
.pytest_cache/
+.idea
\ No newline at end of file
diff --git a/tests/test_middleware.py b/tests/test_middleware.py
index 37dbd17..d599f74 100644
--- a/tests/test_middleware.py
+++ b/tests/test_middleware.py
@@ -48,7 +48,8 @@ def test_wrong_header(snapshot):
request.META.get.assert_called_once_with('HTTP_AUTHORIZATION')
assert response.status_code == 400
- snapshot.assert_match(json.loads(response.content))
+ as_str = response.content.decode() if hasattr(response.content, 'decode') else response.content
+ snapshot.assert_match(json.loads(as_str))
def test_invalid_token(snapshot):
@@ -61,7 +62,8 @@ def test_invalid_token(snapshot):
request.META.get.assert_called_once_with('HTTP_AUTHORIZATION')
assert response.status_code == 401
- snapshot.assert_match(json.loads(response.content))
+ as_str = response.content.decode() if hasattr(response.content,'decode') else response.content
+ snapshot.assert_match(json.loads(as_str))
def test_unknown_user(snapshot):
--
GitLab