Include correct black-formatted files

Author: bcantoni

tableauserverclient/server/server.py

@@ -335,21 +335,23 @@ def is_signed_in(self):
 
     def configure_ssl(self, *, allow_weak_dh=False):
         """Configure SSL/TLS settings for the server connection.
-        
+
         Parameters
         ----------
         allow_weak_dh : bool, optional
             If True, allows connections to servers with DH keys that are considered too small by modern Python versions.
             WARNING: This reduces security and should only be used as a temporary workaround.
         """
         if allow_weak_dh:
-            logger.warning("WARNING: Allowing weak Diffie-Hellman keys. This reduces security and should only be used temporarily.")
+            logger.warning(
+                "WARNING: Allowing weak Diffie-Hellman keys. This reduces security and should only be used temporarily."
+            )
             self._ssl_context = ssl.create_default_context()
             # Allow weak DH keys by setting minimum key size to 512 bits (default is 1024 in Python 3.12+)
             self._ssl_context.set_dh_parameters(min_key_bits=512)
-            self.add_http_options({'verify': self._ssl_context})
+            self.add_http_options({"verify": self._ssl_context})
         else:
             self._ssl_context = None
             # Remove any custom SSL context if we're reverting to default settings
-            if 'verify' in self._http_options:
-                del self._http_options['verify']
+            if "verify" in self._http_options:
+                del self._http_options["verify"]

test/test_ssl_config.py

@@ -7,27 +7,27 @@
 
 
 class TestSSLConfig(unittest.TestCase):
-    @patch('requests.session')
-    @patch('tableauserverclient.server.endpoint.Endpoint.set_parameters')
+    @patch("requests.session")
+    @patch("tableauserverclient.server.endpoint.Endpoint.set_parameters")
     def setUp(self, mock_set_parameters, mock_session):
         """Set up test fixtures with mocked session and request validation"""
         # Mock the session
         self.mock_session = MagicMock()
         mock_session.return_value = self.mock_session
-        
+
         # Mock request preparation
         self.mock_request = MagicMock()
         self.mock_session.prepare_request.return_value = self.mock_request
-        
+
         # Create server instance with mocked components
-        self.server = Server('http://test')
+        self.server = Server("http://test")
 
     def test_default_ssl_config(self):
         """Test that by default, no custom SSL context is used"""
         self.assertIsNone(self.server._ssl_context)
-        self.assertNotIn('verify', self.server.http_options)
+        self.assertNotIn("verify", self.server.http_options)
 
-    @patch('ssl.create_default_context')
+    @patch("ssl.create_default_context")
     def test_weak_dh_config(self, mock_create_context):
         """Test that weak DH keys can be allowed when configured"""
         # Setup mock SSL context
@@ -36,15 +36,15 @@ def test_weak_dh_config(self, mock_create_context):
 
         # Configure SSL with weak DH
         self.server.configure_ssl(allow_weak_dh=True)
-        
+
         # Verify SSL context was created and configured correctly
         mock_create_context.assert_called_once()
         mock_context.set_dh_parameters.assert_called_once_with(min_key_bits=512)
-        
+
         # Verify context was added to http options
-        self.assertEqual(self.server.http_options['verify'], mock_context)
+        self.assertEqual(self.server.http_options["verify"], mock_context)
 
-    @patch('ssl.create_default_context')
+    @patch("ssl.create_default_context")
     def test_disable_weak_dh_config(self, mock_create_context):
         """Test that SSL config can be reset to defaults"""
         # Setup mock SSL context
@@ -54,23 +54,24 @@ def test_disable_weak_dh_config(self, mock_create_context):
         # First enable weak DH
         self.server.configure_ssl(allow_weak_dh=True)
         self.assertIsNotNone(self.server._ssl_context)
-        self.assertIn('verify', self.server.http_options)
+        self.assertIn("verify", self.server.http_options)
 
         # Then disable it
         self.server.configure_ssl(allow_weak_dh=False)
         self.assertIsNone(self.server._ssl_context)
-        self.assertNotIn('verify', self.server.http_options)
+        self.assertNotIn("verify", self.server.http_options)
 
-    @patch('ssl.create_default_context')
+    @patch("ssl.create_default_context")
     def test_warning_on_weak_dh(self, mock_create_context):
         """Test that a warning is logged when enabling weak DH keys"""
         logging.getLogger().setLevel(logging.WARNING)
-        with self.assertLogs(level='WARNING') as log:
+        with self.assertLogs(level="WARNING") as log:
             self.server.configure_ssl(allow_weak_dh=True)
             self.assertTrue(
-                any('WARNING: Allowing weak Diffie-Hellman keys' in record for record in log.output),
-                "Expected warning about weak DH keys was not logged"
+                any("WARNING: Allowing weak Diffie-Hellman keys" in record for record in log.output),
+                "Expected warning about weak DH keys was not logged",
             )
 
-if __name__ == '__main__':
-    unittest.main() 
+
+if __name__ == "__main__":
+    unittest.main()