codekingpro/portable-devtools
115k
1�
2 q�jA � �| � S SK r S SKrS SKJr " S S\R 5 r\S:X a \R " 5 gg)� N)�patchc �8 � \ rS rSrS rS rS rS rS rS r Sr3g )4� UtilsTest� c �� � SSK Jn U" 5 nU SSS5 U R UR 5 5 U R UR SSS5 5 g! , ( d f NP= f)zTest NoLock context managerr )�NoLockN)�websocket._utilsr � assertIsNone� __enter__�__exit__)�selfr �locks �`D:\code\apps\devtools\python\user_packages\Python313\site-packages\websocket/tests/test_utils.py� test_nolock�UtilsTest.test_nolock sS � �+��x�� � � �
���$�.�.�*�+� ���$�-�-��d�D�9�:� �T�s �A � 5A*c �~ � SSK Jn U R U" SR S5 5 5 U R U" SR S5 5 5 U R U" S5 5 U R U" SR S5 5 5 U R U" S5 5 U R U" S 5 5 g6) z/Test UTF-8 validation when wsaccel is availabler )�
validate_utf8�
Hello, World!�utf-8� 🌟 Unicode test� Hello, bytes� Héllo with accénts� ��� ��N)r r �7assertTrue�encode� assertFalse)r
r s r �!test_utf8_validation_with_wsaccel�+UtilsTest.test_utf8_validation_with_wsaccel( s� � � 3�
���
�o�&<�&<�W�&E�F�G� ���
�&9�&@�&@��&I�J�K� ���
�o�6�7� ���
�&<�&C�&C�G�&L�M�N�
��� �{�3�4� ��� �{�3�4� c � ^� S[ R ; a [ R S SSKnUR mU4S jn[ SUS9 SSKJn U R UR SR S5 5 5 U R UR S 5 5 U R UR S8R S5 5 5 U R UR S R S5 5 5 U R UR S R S5 5 5 U R UR S
R S5 5 5 U R UR S5 5 U R UR S5 5 U R UR S5 5 SSS5 g! , ( d f g= f)z<Test UTF-8 validation fallback when wsaccel is not available�websocket._utilsr Nc �F >� SU ; a [ SU S35 eT" U /UQ70 UD6$ )N�wsaccelzNo module named '�')� ImportError)�name�args�kwargs�original_imports �r � mock_import�<UtilsTest.test_utf8_validation_fallback.<locals>.mock_importB s5 �� ��D� �!�$5�d�V�1�"=�>�>�"�4�9�$�9�&�9� 9r zbuiltins.__import__)� side_effectr r r z9ASCII textr r � r r r ) �sys�modules�builtins�10__import__r r �_utilsr r r r )r
r1 r+ �utilsr* s @r � test_utf8_validation_fallback�'UtilsTest.test_utf8_validation_fallback7 sb �� � ���� ,�� � � .�/� �"�-�-�� :�11 �(�k�
B� ,�
�O�O�E� /� /��0F�0F�w�0O� P� Q� �O�O�E� /� /�� @� A� �O�O�E� /� /� �0C�0C�G�0L� M� N�
�O�O�E� /� /�0C�0J�0J�7�0S� T� U� �O�O�E� /� /�0F�0M�0M�g�0V� W� X�
�O�O�E� /� /�� � �'�0B� C� D� �O�O�E� /� /�� 4� 5�
� � �U� 0� 0�� =� >� � � �U� 0� 0�� =� >�% C�
B�
B�s
�E2G�12Gc �� � SSK Jn [ S5 nU R U" U5 S5 [ SS5 nU R U" U5 S5 [ 5 nU R U" U5 5 g)z!Test extract_err_message functionr )�extract_err_messagezTest error messagez First arg�13Second argN)r r8 � Exception� assertEqualr14 )r
r8 �
exc_with_args�exc_multi_args� exc_no_argss r �test_extract_err_message�"UtilsTest.test_extract_err_message^ sj � �8� "�"6�7�
� ���,�]�;�=Q�R� #�;� �=�� ���,�^�<�k�J� �k� � ���-�k�:�;r c �$ � SSK Jn [ SS5 nU R U" U5 S5 [ SS5 nU R U" U5 5 [ S5 nU R U" U5 5 [ 5 nU R U" U5 5 g) z Test extract_error_code functionr )�extract_error_codei� z Not foundz
Error messager9 z15Single argN)r rB r: r; r16 )r
rB �
exc_with_code�exc_with_string�exc_single_argr> s r �test_extract_error_code�!UtilsTest.test_extract_error_coden s� � �7� "�#�{�3�
� ���+�M�:�C�@� $�O�\�B�� ���,�_�=�>� #�<�0�� ���,�^�<�=� �k� � ���,�[�9�:r c �P � S[ R ; a [ R S gg)zClean up after testsr" N)r/ r0 )r
s r �tearDown�UtilsTest.tearDown� s" � � ���� ,�� � � .�/� -r � N) �__name__�17__module__� __qualname__�__firstlineno__r r r5 r? rF rI �__static_attributes__rK r r r r s! � � ;�
5� "?�N<� ;�(0r r �__main__)r/ �unittest�
unittest.mockr �TestCaser rL �mainrK r r �<module>rV s@ �� 18� � ��&m0��!�!� m0�` �z�� �M�M�O� r