SNOW-79076: Add test to ensure error code 390111 is ignored upon session closure

Author: sfc-gh-stakeda

test/test_unit_connection.py

@@ -167,3 +167,55 @@ def mock_post_request(url, headers, json_body, **kwargs):
         warehouse='TESTWH',
     )
     assert con.service_name == 'FAKE_SERVICE_NAME'
+
+
+@patch(
+    'snowflake.connector.network.SnowflakeRestful._post_request'
+)
+def test_connection_ignore_exception(mockSnowflakeRestfulPostRequest):
+    def mock_post_request(url, headers, json_body, **kwargs):
+        global mock_cnt
+        ret = None
+        if mock_cnt == 0:
+            # return from /v1/login-request
+            ret = {
+                u'success': True,
+                u'message': None,
+                u'data': {
+                    u'token': u'TOKEN',
+                    u'masterToken': u'MASTER_TOKEN',
+                    u'idToken': u'ID_TOKEN',
+                    u'parameters': [
+                        {'name': 'SERVICE_NAME', 'value': "FAKE_SERVICE_NAME"}
+                    ],
+                }}
+        elif mock_cnt == 1:
+            ret = {
+                u'success': False,
+                u'message': "Session gone",
+                u'data': None,
+                u'code': 390111
+            }
+        mock_cnt += 1
+        return ret
+
+    # POST requests mock
+    mockSnowflakeRestfulPostRequest.side_effect = mock_post_request
+
+    global mock_cnt
+    mock_cnt = 0
+
+    account = 'testaccount'
+    user = 'testuser'
+
+    # connection
+    con = snowflake.connector.connect(
+        account=account,
+        user=user,
+        password='testpassword',
+        database='TESTDB',
+        warehouse='TESTWH',
+    )
+    # Test to see if closing connection works or raises an exception. If an exception is raised, test will fail.
+    con.close()
+