@@ -58,13 +58,12 @@ def read(self):
58
58
59
59
class MockClient (Client ):
60
60
61
- def __init__ (self , host , response_code ):
61
+ def __init__ (self , host , response_code , timeout = None ):
62
62
self .response_code = 200
63
63
Client .__init__ (self , host )
64
64
65
- def _make_request (self , opener , request ):
66
-
67
- if 200 <= self .response_code < 299 : # if successful code
65
+ def _make_request (self , opener , request , timeout = None ):
66
+ if 200 <= self .response_code < 299 : # if successsful code
68
67
return MockResponse (self .response_code )
69
68
else :
70
69
raise handle_error (MockException (self .response_code ))
@@ -74,12 +73,11 @@ class TestClient(unittest.TestCase):
74
73
75
74
def setUp (self ):
76
75
self .host = 'http://api.test.com'
77
- self .client = Client (host = self .host )
78
- self .api_key = 'SENDGRID_API_KEY'
76
+ self .api_key = "SENDGRID_API_KEY"
79
77
self .request_headers = {
80
- 'Content-Type' : 'application/json' ,
81
- 'Authorization' : 'Bearer ' + self .api_key
82
- }
78
+ 'Content-Type' : 'application/json' ,
79
+ 'Authorization' : 'Bearer ' + self .api_key ,
80
+ }
83
81
self .client = Client (host = self .host ,
84
82
request_headers = self .request_headers ,
85
83
version = 3 )
@@ -88,6 +86,7 @@ def test__init__(self):
88
86
default_client = Client (host = self .host )
89
87
self .assertEqual (default_client .host , self .host )
90
88
self .assertEqual (default_client .request_headers , {})
89
+ self .assertIs (default_client .timeout , None )
91
90
methods = ['delete' , 'get' , 'patch' , 'post' , 'put' ]
92
91
self .assertEqual (default_client .methods , methods )
93
92
self .assertEqual (default_client ._version , None )
@@ -97,13 +96,15 @@ def test__init__(self):
97
96
version = 3
98
97
client = Client (host = self .host ,
99
98
request_headers = request_headers ,
100
- version = version )
99
+ version = version ,
100
+ timeout = 10 )
101
101
self .assertEqual (client .host , self .host )
102
102
self .assertEqual (client .request_headers , request_headers )
103
103
methods = ['delete' , 'get' , 'patch' , 'post' , 'put' ]
104
104
self .assertEqual (client .methods , methods )
105
105
self .assertEqual (client ._version , 3 )
106
106
self .assertEqual (client ._url_path , [])
107
+ self .assertEqual (client .timeout , 10 )
107
108
108
109
def test__build_versioned_url (self ):
109
110
url = '/api_keys?hello=1&world=2'
0 commit comments