ó î;,Uc@s–dZddlmZddlZddlZddlZddlmZmZm Z ddl m Z defd„ƒYZ e dkr’eƒndS( s( Unit tests for :py:obj:`OpenSSL.rand`. iÿÿÿÿ(tmainN(t NON_ASCIItTestCasetb(trandt RandTestscBs³eZd„Zd„Zd„Zd„Zd„Zd„Zd„Zd„Z d„Z d „Z d „Z d „Z d „Zd „Zd„Zd„Zd„Zd„Zd„ZRS(cCsF|jttjƒ|jttjdƒ|jttjddƒdS(s¤ :py:obj:`OpenSSL.rand.bytes` raises :py:obj:`TypeError` if called with the wrong number of arguments or with a non-:py:obj:`int` argument. iN(t assertRaisest TypeErrorRtbytestNone(tself((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_bytes_wrong_argsscCs|jttjtjƒdS(sˆ :py:obj:`OpenSSL.rand.bytes` raises :py:obj:`MemoryError` if more bytes are requested than will fit in memory. N(Rt MemoryErrorRRtsystmaxsize(R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_insufficientMemoryscCs¡tjdƒ}|jt|ƒdƒtjddƒ}|j||ƒtjddƒ}|jt|ƒdƒ|jttjdƒ}|jt|ƒdƒdS(s® Verify that we can obtain bytes from rand_bytes() and that they are different each time. Test the parameter of rand_bytes() for bad values. i2t num_bytesiiÿÿÿÿsnum_bytes must not be negativeN(RRt assertEqualtlentassertNotEqualRt ValueErrortstr(R tb1tb2tb3texc((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyt test_bytes$scCsq|jttjƒ|jttjtdƒdƒ|jttjddƒ|jttjtdƒddƒdS(s¿ When called with the wrong number of arguments, or with arguments not of type :py:obj:`str` and :py:obj:`int`, :py:obj:`OpenSSL.rand.add` raises :py:obj:`TypeError`. tfooiN(RRRtaddRR (R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_add_wrong_args4scCstjtdƒdƒdS(sF :py:obj:`OpenSSL.rand.add` adds entropy to the PRNG. t hamburgeriN(RRR(R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_add?scCsL|jttjƒ|jttjdƒ|jttjtdƒdƒdS(s§ When called with the wrong number of arguments, or with a non-:py:obj:`str` argument, :py:obj:`OpenSSL.rand.seed` raises :py:obj:`TypeError`. RN(RRRtseedR R(R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_seed_wrong_argsFscCstjtdƒƒdS(sG :py:obj:`OpenSSL.rand.seed` adds entropy to the PRNG. s milk shakeN(RR R(R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyt test_seedPscCs|jttjdƒdS(sr :py:obj:`OpenSSL.rand.status` raises :py:obj:`TypeError` when called with any arguments. N(RRRtstatusR (R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_status_wrong_argsWscCs|jtjƒdkƒdS(s :py:obj:`OpenSSL.rand.status` returns :py:obj:`True` if the PRNG has sufficient entropy, :py:obj:`False` otherwise. iiN(ii(t assertTrueRR#(R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyt test_status_scCs{|jttjƒ|jttjdƒ|jttjddƒ|jttjddƒ|jttjdddƒdS(s½ :py:obj:`OpenSSL.rand.egd` raises :py:obj:`TypeError` when called with the wrong number of arguments or with arguments not of type :py:obj:`str` and :py:obj:`int`. RiN(RRRtegdR (R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_egd_wrong_argsjs cCs?tj|jƒƒ}d}|j||kd||fƒdS(s‡ :py:obj:`OpenSSL.rand.egd` returns :py:obj:`0` or :py:obj:`-1` if the EGD socket passed to it does not exist. iÿÿÿÿis %r not in %rN(iÿÿÿÿi(RR'tmktempR%(R tresulttexpected((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_egd_missingvs  cCsBtj|jƒdƒ}d}|j||kd||fƒdS(s¼ :py:obj:`OpenSSL.rand.egd` returns :py:obj:`0` or :py:obj:`-1` if the EGD socket passed to it does not exist even if a size argument is explicitly passed. iiÿÿÿÿis %r not in %rN(iÿÿÿÿi(RR'R)R%(R R*R+((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_egd_missing_and_bytes‚s  cCs|jttjdƒdS(ss :py:obj:`OpenSSL.rand.cleanup` raises :py:obj:`TypeError` when called with any arguments. N(RRRtcleanupR (R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_cleanup_wrong_argsscCs|jtjƒdƒdS(sy :py:obj:`OpenSSL.rand.cleanup` releases the memory used by the PRNG and returns :py:obj:`None`. N(tassertIdenticalRR.R (R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyt test_cleanup—scCse|jttjƒ|jttjddƒ|jttjddƒ|jttjdddƒdS(s¹ :py:obj:`OpenSSL.rand.load_file` raises :py:obj:`TypeError` when called the wrong number of arguments or arguments not of type :py:obj:`str` and :py:obj:`int`. RiN(RRRt load_fileR (R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_load_file_wrong_argsŸscCsF|jttjƒ|jttjdƒ|jttjddƒdS(s¦ :py:obj:`OpenSSL.rand.write_file` raises :py:obj:`TypeError` when called with the wrong number of arguments or a non-:py:obj:`str` argument. RN(RRRt write_fileR (R ((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_write_file_wrong_argsªscCst|dƒWdQXzTtj|ƒtj|ƒtj}|jd|ƒtj|ƒtj|dƒWdtj|ƒXdS(sU Verify that ``rand.write_file`` and ``rand.load_file`` can be used. twNii( topenRR4toststattST_SIZERR2tunlink(R tpathtsize((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyt_read_write_test³s  cCs6|jƒ}|tjtjƒƒ7}|j|ƒdS(se Random data can be saved and loaded to files with paths specified as bytes. N(R)RtencodeR tgetfilesystemencodingR>(R R<((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_bytes_pathsËs cCs*|jƒjdƒt}|j|ƒdS(sg Random data can be saved and loaded to files with paths specified as unicode. sutf-8N(R)tdecodeRR>(R R<((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyttest_unicode_pathsÕs(t__name__t __module__R RRRRR!R"R$R&R(R,R-R/R1R3R5R>RARC(((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyRs&         t__main__(t__doc__tunittestRR8R9R tOpenSSL.test.utilRRRtOpenSSLRRRD(((s:/usr/lib/python2.7/dist-packages/OpenSSL/test/test_rand.pyts   Í