1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
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()
|
