o M,öhAã@sHddlZddlZddlmZ Gdd„dejƒZedkr"e ¡dSdS)éN)Úpatchc@s<eZdZdd„Zdd„Zdd„Zdd„Zd d „Zd d „Zd S)Ú UtilsTestcCs\ddlm}|ƒ}|Wdƒn1swY| | ¡¡| | ddd¡¡dS)zTest NoLock context managerr)ÚNoLockN)Úwebsocket._utilsrÚ assertIsNoneÚ __enter__Ú__exit__)ÚselfrÚlock©r úf/var/www/www-root/data/www/bot.pdev.uz/venv/lib/python3.10/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_wsaccelcsdtjvr tjd=ddl}|j‰‡fdd„}td|dfddlm}| | d  d ¡¡¡| | d ¡¡| | d   d ¡¡¡| | d   d ¡¡¡| | d   d ¡¡¡| | d  d ¡¡¡| | d¡¡|  | d¡¡|  | d¡¡WdƒdS1s…wYdS)z.mock_importzbuiltins.__import__)Ú side_effectrrrz ASCII textrrÚórr) ÚsysÚmodulesÚbuiltinsÚ __import__rrÚ_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.Ú exc_with_argsÚexc_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)rr6r0r1r)r r6Ú exc_with_codeÚexc_with_stringÚexc_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 ÚtearDown‚s  ÿzUtilsTest.tearDownN) Ú__name__Ú __module__Ú __qualname__r rr-r5r:r;r r r r rs' rÚ__main__)r'ÚunittestÚ unittest.mockrÚTestCaserr<Úmainr r r r Ús p ÿ