a ñ»jAã@sBddlZddlZddlmZGdd„dejƒZedkr>e ¡dS)éN)Úpatchc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)Ú UtilsTestcCs\ddlm}|ƒ}|Wdƒn1s,0Y| | ¡¡| | ddd¡¡dS)zTest NoLock context managerr)ÚNoLockN)Úwebsocket._utilsrÚ assertIsNoneÚ __enter__Ú__exit__)ÚselfrÚlock©r ú@E:\soccer\ml_env\Lib\site-packages\websocket/tests/test_utils.pyÚ test_nolocks  zUtilsTest.test_nolockcCsvddlm}| |d d¡ƒ¡| |d d¡ƒ¡| |dƒ¡| |d d¡ƒ¡| |dƒ¡| |d ƒ¡d S) 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)rrÚ assertTrueÚencodeÚ assertFalse)r rr r r Ú!test_utf8_validation_with_wsaccel(s z+UtilsTest.test_utf8_validation_with_wsaccelcsdtjvrtjd=ddl}|j‰‡fdd„}td|dÊddlm}| | d  d ¡¡¡| | d ¡¡| | d   d ¡¡¡| | d   d ¡¡¡| | d   d ¡¡¡| | d  d ¡¡¡| | d¡¡|  | d¡¡|  | d¡¡Wdƒn1s 0YdS)z.mock_importzbuiltins.__import__)Z side_effectrrrz ASCII textrrÚórr) ÚsysÚmodulesÚbuiltinsÚ __import__rrZ_utilsrrrr)r r&r!Úutilsr r r Útest_utf8_validation_fallback7s    z'UtilsTest.test_utf8_validation_fallbackcCsVddlm}tdƒ}| ||ƒd¡tddƒ}| ||ƒd¡tƒ}| ||ƒ¡dS)z!Test extract_err_message functionr)Úextract_err_messagezTest error messagez First argú Second argN)rr*Ú ExceptionÚ assertEqualr)r r*Z exc_with_argsZexc_multi_argsÚ exc_no_argsr r r Útest_extract_err_message^s  z"UtilsTest.test_extract_err_messagecCslddlm}tddƒ}| ||ƒd¡tddƒ}| ||ƒ¡tdƒ}| ||ƒ¡tƒ}| ||ƒ¡dS) z Test extract_error_code functionr)Úextract_error_codei”z Not foundz Error messager+z Single argN)rr0r,r-r)r r0Z exc_with_codeZexc_with_stringZexc_single_argr.r r r Útest_extract_error_codens   z!UtilsTest.test_extract_error_codecCsdtjvrtjd=dS)zClean up after testsrN)r$r%)r r r r ÚtearDown‚s zUtilsTest.tearDownN) Ú__name__Ú __module__Ú __qualname__r rr)r/r1r2r r r r rs 'rÚ__main__)r$ZunittestZ unittest.mockrZTestCaserr3Úmainr r r r Ús  p