diff options
author | Paul Kehrer <paul.l.kehrer@gmail.com> | 2017-11-20 22:25:18 +0800 |
---|---|---|
committer | Alex Gaynor <alex.gaynor@gmail.com> | 2017-11-20 09:25:18 -0500 |
commit | acbd662b62a2de0e7102d560f7e73d7feaf9e600 (patch) | |
tree | 0c3a8d3e58d8a5f0247f16409e2a0e8b7cd4aa2b /tests/test_rand.py | |
parent | 4aa52c33d3ee51c632e0e1e10cafb7745fd1028c (diff) | |
download | pyopenssl-git-acbd662b62a2de0e7102d560f7e73d7feaf9e600.tar.gz |
restore a subset of the rand module (#708)
* restore a subset of the rand module
* flake
* remove cleanup, go ahead and assume status will always be 1
* lighten and add power
Diffstat (limited to 'tests/test_rand.py')
-rw-r--r-- | tests/test_rand.py | 38 |
1 files changed, 38 insertions, 0 deletions
diff --git a/tests/test_rand.py b/tests/test_rand.py new file mode 100644 index 0000000..e04a24c --- /dev/null +++ b/tests/test_rand.py @@ -0,0 +1,38 @@ +# Copyright (c) Frederick Dean +# See LICENSE for details. + +""" +Unit tests for `OpenSSL.rand`. +""" + +import pytest + +from OpenSSL import rand + + +class TestRand(object): + + @pytest.mark.parametrize('args', [ + (b"foo", None), + (None, 3), + ]) + def test_add_wrong_args(self, args): + """ + `OpenSSL.rand.add` raises `TypeError` if called with arguments not of + type `str` and `int`. + """ + with pytest.raises(TypeError): + rand.add(*args) + + def test_add(self): + """ + `OpenSSL.rand.add` adds entropy to the PRNG. + """ + rand.add(b'hamburger', 3) + + def test_status(self): + """ + `OpenSSL.rand.status` returns `1` if the PRNG has sufficient entropy, + `0` otherwise. + """ + assert rand.status() == 1 |