test: explicitly set session cookie expiry

Author: veronika-alraheem

test/unit/test_couchdb_session_auth.py

@@ -25,6 +25,7 @@
 
 from ibmcloudant import CouchDbSessionAuthenticator
 from ibmcloudant.cloudant_v1 import CloudantV1
+from ibmcloudant.couchdb_session_token_manager import CouchDbSessionTokenManager
 
 request = requests.request
 request_args = None
@@ -73,6 +74,7 @@ def post_session(request):
 
     @responses.activate
     def test_header_passing(self):
+        self.cookie_expire_time = None
         self.client.set_default_headers({"yes": "works"})
         response = self.client.get_session_information(headers={"foo": "bar"})
         self.assertIsNotNone(response)
@@ -89,6 +91,7 @@ def test_header_passing(self):
 
     @responses.activate
     def test_disable_ssl_verification_on(self):
+        self.cookie_expire_time = None
         original_http_client = self.client.get_http_client()
         mock_session = MockSession()
         try:
@@ -101,6 +104,7 @@ def test_disable_ssl_verification_on(self):
 
     @responses.activate
     def test_disable_ssl_verification_off(self):
+        self.cookie_expire_time = None
         original_http_client = self.client.get_http_client()
         mock_session = MockSession()
         try:
@@ -139,6 +143,7 @@ def test_invalid_disable_ssl_verification_type(self):
 
     @responses.activate
     def test_cookie_refresh(self):
+        self.cookie_expire_time = None
         self.client.get_session_information()
         self.assertEqual(responses.calls[-1].request.headers["Cookie"], "AuthSession=foobar")
         self.cookie_value = "bar"
@@ -154,6 +159,7 @@ def test_cookie_refresh(self):
 
     @responses.activate
     def test_cookie_expired(self):
+        self.cookie_expire_time = None
         self.client.get_session_information()
         self.assertEqual(responses.calls[-1].request.headers["Cookie"], "AuthSession=foobar")
         self.cookie_value = "bar"
@@ -164,6 +170,7 @@ def test_cookie_expired(self):
 
     @responses.activate
     def test_cookie_not_yet_expired(self):
+        self.cookie_expire_time = None
         self.client.get_session_information()
         self.assertEqual(responses.calls[-1].request.headers["Cookie"], "AuthSession=foobar")
         self.cookie_value = "bar"
@@ -180,12 +187,23 @@ def test_refresh_time_calculation(self):
         self.assertAlmostEqual(self.authenticator.token_manager.refresh_time,
                                self.authenticator.token_manager._get_current_time() + 8 * 60, delta=3)
 
+    @responses.activate
+    def test_refresh_time_calculation_no_expiry(self):
+        self.cookie_expire_time = None
+        self.client.get_session_information()
+        token_manager = self.authenticator.token_manager
+        default_ttl = CouchDbSessionTokenManager._DEFAULT_SESSION_COOKIE_TTL_SECONDS
+        self.assertAlmostEqual(token_manager.expire_time,
+                               token_manager._get_current_time() + default_ttl, delta=3)
+        self.assertAlmostEqual(token_manager.refresh_time,
+                               token_manager._get_current_time() + default_ttl * 0.8, delta=3)
+
     @responses.activate
     def test_cookie_without_expiry(self):
+        self.cookie_expire_time = None
         self.client.get_session_information()
         self.assertEqual(responses.calls[-1].request.headers["Cookie"], "AuthSession=foobar")
         self.cookie_value = "bar"
-        self.cookie_expire_time = None
         self.authenticator.token_manager.refresh_time = 0
         self.client.get_session_information()
         self.assertEqual(responses.calls[-1].request.headers["Cookie"], "AuthSession=bar")
@@ -197,6 +215,7 @@ def test_cookie_without_expiry(self):
 
     @responses.activate
     def test_set_service_url(self):
+        self.cookie_expire_time = None
         self.client.get_session_information()
         self.assertEqual(responses.calls[-2].request.url, "http://cloudant.example/_session")
         self.assertEqual(responses.calls[-2].request.method, "POST")