initial commit

2026-06-25 21:29:21 +00:00
commit 0d0a7456de
2738 changed files with 542622 additions and 0 deletions
@@ -0,0 +1,91 @@
+# -*- coding: utf-8 -*-
+import sys
+import unittest
+from unittest.mock import patch
+
+"""
+test_ssl_compat.py
+websocket - WebSocket client library for Python
+
+Copyright 2025 engn33r
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+"""
+
+class SSLCompatTest(unittest.TestCase):
+    def test_ssl_available(self):
+        """Test that SSL is available in normal conditions"""
+        import websocket._ssl_compat as ssl_compat
+
+        # In normal conditions, SSL should be available
+        self.assertTrue(ssl_compat.HAVE_SSL)
+        self.assertIsNotNone(ssl_compat.ssl)
+
+        # SSL exception classes should be available
+        self.assertTrue(hasattr(ssl_compat, "SSLError"))
+        self.assertTrue(hasattr(ssl_compat, "SSLEOFError"))
+        self.assertTrue(hasattr(ssl_compat, "SSLWantReadError"))
+        self.assertTrue(hasattr(ssl_compat, "SSLWantWriteError"))
+
+    def test_ssl_not_available(self):
+        """Test fallback behavior when SSL is not available"""
+        # Remove ssl_compat from modules to force reimport
+        if "websocket._ssl_compat" in sys.modules:
+            del sys.modules["websocket._ssl_compat"]
+
+        # Mock the ssl module to not be available
+        import builtins
+
+        original_import = builtins.__import__
+
+        def mock_import(name, *args, **kwargs):
+            if name == "ssl":
+                raise ImportError("No module named 'ssl'")
+            return original_import(name, *args, **kwargs)
+
+        with patch("builtins.__import__", side_effect=mock_import):
+            import websocket._ssl_compat as ssl_compat
+
+            # SSL should not be available
+            self.assertFalse(ssl_compat.HAVE_SSL)
+            self.assertIsNone(ssl_compat.ssl)
+
+            # Fallback exception classes should be available and functional
+            self.assertTrue(issubclass(ssl_compat.SSLError, Exception))
+            self.assertTrue(issubclass(ssl_compat.SSLEOFError, Exception))
+            self.assertTrue(issubclass(ssl_compat.SSLWantReadError, Exception))
+            self.assertTrue(issubclass(ssl_compat.SSLWantWriteError, Exception))
+
+            # Test that exceptions can be instantiated
+            ssl_error = ssl_compat.SSLError("test error")
+            self.assertIsInstance(ssl_error, Exception)
+            self.assertEqual(str(ssl_error), "test error")
+
+            ssl_eof_error = ssl_compat.SSLEOFError("test eof")
+            self.assertIsInstance(ssl_eof_error, Exception)
+
+            ssl_want_read = ssl_compat.SSLWantReadError("test read")
+            self.assertIsInstance(ssl_want_read, Exception)
+
+            ssl_want_write = ssl_compat.SSLWantWriteError("test write")
+            self.assertIsInstance(ssl_want_write, Exception)
+
+    def tearDown(self):
+        """Clean up after tests"""
+        # Ensure ssl_compat is reimported fresh for next test
+        if "websocket._ssl_compat" in sys.modules:
+            del sys.modules["websocket._ssl_compat"]
+
+
+if __name__ == "__main__":
+    unittest.main()