@@ -132,10 +132,17 @@ def test_default_state(self):
132132 assert not credentials .valid
133133 assert credentials .expired
134134
135- def make_request (self , data , status = http_client .OK , headers = None , side_effect = None ):
135+ def make_request (
136+ self ,
137+ data ,
138+ status = http_client .OK ,
139+ headers = None ,
140+ side_effect = None ,
141+ use_data_bytes = True ,
142+ ):
136143 response = mock .create_autospec (transport .Response , instance = False )
137144 response .status = status
138- response .data = _helpers .to_bytes (data )
145+ response .data = _helpers .to_bytes (data ) if use_data_bytes else data
139146 response .headers = headers or {}
140147
141148 request = mock .create_autospec (transport .Request , instance = False )
@@ -144,7 +151,8 @@ def make_request(self, data, status=http_client.OK, headers=None, side_effect=No
144151
145152 return request
146153
147- def test_refresh_success (self , mock_donor_credentials ):
154+ @pytest .mark .parametrize ("use_data_bytes" , [True , False ])
155+ def test_refresh_success (self , use_data_bytes , mock_donor_credentials ):
148156 credentials = self .make_credentials (lifetime = None )
149157 token = "token"
150158
@@ -154,7 +162,9 @@ def test_refresh_success(self, mock_donor_credentials):
154162 response_body = {"accessToken" : token , "expireTime" : expire_time }
155163
156164 request = self .make_request (
157- data = json .dumps (response_body ), status = http_client .OK
165+ data = json .dumps (response_body ),
166+ status = http_client .OK ,
167+ use_data_bytes = use_data_bytes ,
158168 )
159169
160170 credentials .refresh (request )
0 commit comments