DexDemo/frontend/venv/lib/python3.12/site-packages/websocket/tests/test_ssl_compat.py

# -*- 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()