2020
2121from cloudant ._2to3 import Cookie
2222from cloudant .client import Cloudant
23- from cloudant .client_session import IAMSession
23+ from cloudant ._client_session import IAMSession
2424
2525MOCK_API_KEY = 'CqbrIYzdO3btWV-5t4teJLY_etfT_dkccq-vO-5vCXSo'
2626
@@ -97,7 +97,7 @@ def test_iam_set_credentials(self):
9797
9898 self .assertEquals (iam ._api_key , new_api_key )
9999
100- @mock .patch ('cloudant.client_session .ClientSession.request' )
100+ @mock .patch ('cloudant._client_session .ClientSession.request' )
101101 def test_iam_get_access_token (self , m_req ):
102102 m_response = mock .MagicMock ()
103103 m_response .json .return_value = MOCK_OIDC_TOKEN_RESPONSE
@@ -122,8 +122,8 @@ def test_iam_get_access_token(self, m_req):
122122 self .assertTrue (m_response .raise_for_status .called )
123123 self .assertTrue (m_response .json .called )
124124
125- @mock .patch ('cloudant.client_session .ClientSession.request' )
126- @mock .patch ('cloudant.client_session .IAMSession._get_access_token' )
125+ @mock .patch ('cloudant._client_session .ClientSession.request' )
126+ @mock .patch ('cloudant._client_session .IAMSession._get_access_token' )
127127 def test_iam_login (self , m_token , m_req ):
128128 m_token .return_value = MOCK_ACCESS_TOKEN
129129 m_response = mock .MagicMock ()
@@ -150,7 +150,7 @@ def test_iam_logout(self):
150150 iam .logout ()
151151 self .assertEqual (len (iam .cookies .keys ()), 0 )
152152
153- @mock .patch ('cloudant.client_session .ClientSession.get' )
153+ @mock .patch ('cloudant._client_session .ClientSession.get' )
154154 def test_iam_get_session_info (self , m_get ):
155155 m_info = {'ok' : True , 'info' : {'authentication_db' : '_users' }}
156156
@@ -166,8 +166,8 @@ def test_iam_get_session_info(self, m_get):
166166 self .assertEqual (info , m_info )
167167 self .assertTrue (m_response .raise_for_status .called )
168168
169- @mock .patch ('cloudant.client_session .IAMSession.login' )
170- @mock .patch ('cloudant.client_session .ClientSession.request' )
169+ @mock .patch ('cloudant._client_session .IAMSession.login' )
170+ @mock .patch ('cloudant._client_session .ClientSession.request' )
171171 def test_iam_first_request (self , m_req , m_login ):
172172 # mock 200
173173 m_response_ok = mock .MagicMock ()
@@ -191,8 +191,8 @@ def test_iam_first_request(self, m_req, m_login):
191191 self .assertEqual (m_req .call_count , 1 )
192192 self .assertEqual (resp .status_code , 200 )
193193
194- @mock .patch ('cloudant.client_session .IAMSession.login' )
195- @mock .patch ('cloudant.client_session .ClientSession.request' )
194+ @mock .patch ('cloudant._client_session .IAMSession.login' )
195+ @mock .patch ('cloudant._client_session .ClientSession.request' )
196196 def test_iam_renew_cookie_on_expiry (self , m_req , m_login ):
197197 # mock 200
198198 m_response_ok = mock .MagicMock ()
@@ -213,8 +213,8 @@ def test_iam_renew_cookie_on_expiry(self, m_req, m_login):
213213 self .assertEqual (m_req .call_count , 1 )
214214 self .assertEqual (resp .status_code , 200 )
215215
216- @mock .patch ('cloudant.client_session .IAMSession.login' )
217- @mock .patch ('cloudant.client_session .ClientSession.request' )
216+ @mock .patch ('cloudant._client_session .IAMSession.login' )
217+ @mock .patch ('cloudant._client_session .ClientSession.request' )
218218 def test_iam_renew_cookie_on_401_success (self , m_req , m_login ):
219219 # mock 200
220220 m_response_ok = mock .MagicMock ()
@@ -243,8 +243,8 @@ def test_iam_renew_cookie_on_401_success(self, m_req, m_login):
243243 self .assertEqual (m_login .call_count , 2 )
244244 self .assertEqual (m_req .call_count , 3 )
245245
246- @mock .patch ('cloudant.client_session .IAMSession.login' )
247- @mock .patch ('cloudant.client_session .ClientSession.request' )
246+ @mock .patch ('cloudant._client_session .IAMSession.login' )
247+ @mock .patch ('cloudant._client_session .ClientSession.request' )
248248 def test_iam_renew_cookie_on_401_failure (self , m_req , m_login ):
249249 # mock 401
250250 m_response_bad = mock .MagicMock ()
@@ -269,8 +269,8 @@ def test_iam_renew_cookie_on_401_failure(self, m_req, m_login):
269269 self .assertEqual (m_login .call_count , 3 )
270270 self .assertEqual (m_req .call_count , 4 )
271271
272- @mock .patch ('cloudant.client_session .IAMSession.login' )
273- @mock .patch ('cloudant.client_session .ClientSession.request' )
272+ @mock .patch ('cloudant._client_session .IAMSession.login' )
273+ @mock .patch ('cloudant._client_session .ClientSession.request' )
274274 def test_iam_renew_cookie_disabled (self , m_req , m_login ):
275275 # mock 401
276276 m_response_bad = mock .MagicMock ()
@@ -292,8 +292,8 @@ def test_iam_renew_cookie_disabled(self, m_req, m_login):
292292 self .assertEqual (m_login .call_count , 1 ) # no attempt to renew
293293 self .assertEqual (m_req .call_count , 2 )
294294
295- @mock .patch ('cloudant.client_session .IAMSession.login' )
296- @mock .patch ('cloudant.client_session .ClientSession.request' )
295+ @mock .patch ('cloudant._client_session .IAMSession.login' )
296+ @mock .patch ('cloudant._client_session .ClientSession.request' )
297297 def test_iam_client_create (self , m_req , m_login ):
298298 # mock 200
299299 m_response_ok = mock .MagicMock ()
@@ -315,8 +315,8 @@ def test_iam_client_create(self, m_req, m_login):
315315 self .assertEqual (m_req .call_count , 1 )
316316 self .assertEqual (dbs , ['animaldb' ])
317317
318- @mock .patch ('cloudant.client_session .IAMSession.login' )
319- @mock .patch ('cloudant.client_session .IAMSession.set_credentials' )
318+ @mock .patch ('cloudant._client_session .IAMSession.login' )
319+ @mock .patch ('cloudant._client_session .IAMSession.set_credentials' )
320320 def test_iam_client_session_login (self , m_set , m_login ):
321321 # create IAM client
322322 client = Cloudant .iam ('foo' , MOCK_API_KEY )
@@ -331,8 +331,8 @@ def test_iam_client_session_login(self, m_set, m_login):
331331 self .assertEqual (m_login .call_count , 2 )
332332 self .assertEqual (m_set .call_count , 2 )
333333
334- @mock .patch ('cloudant.client_session .IAMSession.login' )
335- @mock .patch ('cloudant.client_session .IAMSession.set_credentials' )
334+ @mock .patch ('cloudant._client_session .IAMSession.login' )
335+ @mock .patch ('cloudant._client_session .IAMSession.set_credentials' )
336336 def test_iam_client_session_login_with_new_credentials (self , m_set , m_login ):
337337 # create IAM client
338338 client = Cloudant .iam ('foo' , MOCK_API_KEY )
0 commit comments