o ʷ6iA@sHddlZddlZddlmZ GdddejZedkr"edSdS)N)patchc@s<eZdZddZddZddZddZd d Zd d Zd S) UtilsTestcCs\ddlm}|}|Wdn1swY||||ddddS)zTest NoLock context managerr)NoLockN)websocket._utilsr assertIsNone __enter____exit__)selfrlockr _/var/www/hoanhtaovolam_webdjango/env/lib/python3.10/site-packages/websocket/tests/test_utils.py test_nolocks zUtilsTest.test_nolockcCsvddlm}||dd||dd||d||dd||d||d d S) z/Test UTF-8 validation when wsaccel is availabler) validate_utf8 Hello, World!utf-8🌟 Unicode test Hello, bytesHéllo with accéntsN)rr assertTrueencode assertFalse)r rr r r !test_utf8_validation_with_wsaccel(s z+UtilsTest.test_utf8_validation_with_wsaccelcsdtjvr tjd=ddl}|jfdd}td|dfddlm}||d d ||d ||d d ||d d ||d d ||d d ||d| |d| |dWddS1swYdS)z.mock_importzbuiltins.__import__) side_effectrrrz ASCII textrrrr) sysmodulesbuiltins __import__rr_utilsrrrr)r r)r#utilsr r!r test_utf8_validation_fallback7s"   "z'UtilsTest.test_utf8_validation_fallbackcCsVddlm}td}|||dtdd}|||dt}|||dS)z!Test extract_err_message functionr)extract_err_messagezTest error messagez First arg Second argN)rr. Exception assertEqualr)r r. exc_with_argsexc_multi_args exc_no_argsr r r test_extract_err_message^s  z"UtilsTest.test_extract_err_messagecCslddlm}tdd}|||dtdd}|||td}|||t}|||dS) z Test extract_error_code functionr)extract_error_codeiz Not foundz Error messager/z Single argN)rr6r0r1r)r r6 exc_with_codeexc_with_stringexc_single_argr4r r r test_extract_error_codens   z!UtilsTest.test_extract_error_codecCsdtjvr tjd=dSdS)zClean up after testsrN)r'r()r r r r tearDowns  zUtilsTest.tearDownN) __name__ __module__ __qualname__r rr-r5r:r;r r r r rs' r__main__)r'unittest unittest.mockrTestCaserr<mainr r r r s p