99
1010from pybutton .request import HTTPError
1111from pybutton .resources .resource import Resource
12- from pybutton .error import ButtonClientError
12+ from pybutton .error import HTTPResponseError
1313
1414config = {
1515 'hostname' : 'api.usebutton.com' ,
@@ -117,8 +117,10 @@ def side_effect(*args):
117117 try :
118118 resource ._api_request ('/v2/api' , 'GET' , data )
119119 self .assertTrue (False )
120- except ButtonClientError as e :
120+ except HTTPResponseError as e :
121121 self .assertEqual (str (e ), '404 bloop' )
122+ self .assertEqual (e .status_code , 404 )
123+ self .assertTrue (e .cause is not None )
122124
123125 @patch ('pybutton.resources.resource.request' )
124126 def test_api_request_with_byte_response (self , request ):
@@ -138,8 +140,10 @@ def side_effect(*args):
138140 try :
139141 resource ._api_request ('/v2/api' , 'GET' , data )
140142 self .assertTrue (False )
141- except ButtonClientError as e :
143+ except HTTPResponseError as e :
142144 self .assertEqual (str (e ), 'bloop failed' )
145+ self .assertEqual (e .status_code , 404 )
146+ self .assertTrue (e .cause is not None )
143147
144148 @patch ('pybutton.resources.resource.request' )
145149 def test_api_get (self , request ):
0 commit comments