status code in test for py3

pynamodb · May 5, 2016 · 4499b22 · 4499b22
1 parent f5a8014
commit 4499b22
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pynamodb/tests/test_base_connection.py b/pynamodb/tests/test_base_connection.py
@@ -1668,8 +1668,10 @@ def test_make_api_call_retries_properly(self, requests_session_mock, session_moc
         # mock response
         deserializable_response = requests.Response()
         deserializable_response._content = json.dumps({'hello': 'world'}).encode('utf-8')
+        deserializable_response.status_code = 200
         bad_response = requests.Response()
         bad_response._content = 'not_json'.encode('utf-8')
+        bad_response.status_code = 503
 
         prepared_request = requests.Request('GET', 'http://lyft.com').prepare()
         session_mock.create_client.return_value._endpoint.create_request.return_value = prepared_request