Updated Cryptographic() provider

mimesis/helpers.py

@@ -1,3 +1,4 @@
+import os
 import random
 
 
@@ -21,3 +22,8 @@ def randints(self, amount=None, a=1, b=100):
 
         return [self.randint(a, b)
                 for _ in range(amount)]
+
+    @staticmethod
+    def urandom(*args, **kwargs):
+        """Return a bytes object containing random bytes."""
+        return os.urandom(*args, **kwargs)

mimesis/providers/cryptographic.py

@@ -1,7 +1,6 @@
 from binascii import hexlify
 import hashlib
 import uuid
-import os
 
 from mimesis.exceptions import UnsupportedAlgorithm
 from mimesis.providers import BaseProvider
@@ -51,12 +50,13 @@ def bytes(self, entropy=None):
         if entropy is None:
             entropy = 32
 
-        return os.urandom(entropy)
+        return self.random.urandom(entropy)
 
     def token(self, entropy=None):
         """Return a random text string, in hexadecimal.
 
         :param entropy: Number of bytes.
         :return: Token.
         """
-        return hexlify(self.bytes(entropy)).decode('ascii')
+        token = hexlify(self.bytes(entropy))
+        return token.decode('ascii')

tests/test_helpers.py

@@ -19,3 +19,13 @@ def test_randints(random):
     assert len(result_custom) == 25
     # All elements in result_custom equals to 1.
     assert result_custom[0] == 1 and result_custom[-1] == 1
+
+
+@pytest.mark.parametrize(
+    'n', (8, 16, 32, 64),
+)
+def test_urandom(random, n):
+    result = random.urandom(n)
+
+    assert len(result) == n
+    assert isinstance(result, bytes)