frostfs-testlib/src/frostfs_testlib/utils/string_utils.py
Aleksei Chetaev 8ed26cef7d Fix issue with generate random string
Signed-off-by: Aleksei Chetaev <alex.chetaev@gmail.com>
2023-03-06 20:04:02 +01:00

32 lines
882 B
Python

import random
import re
import string
ONLY_ASCII_LETTERS = string.ascii_letters
DIGITS_AND_ASCII_LETTERS = string.ascii_letters + string.digits
NON_DIGITS_AND_LETTERS = string.punctuation
def random_string(length: int = 5, source: str = ONLY_ASCII_LETTERS):
"""
Generate random string from source letters list
Args:
length: length for generated string
source: source string with letters for generate random string
Returns:
(str): random string with len == length
"""
return "".join(random.choice(source) for i in range(length))
def is_str_match_pattern(error: Exception, status_pattern: str) -> bool:
"""
Determines whether exception matches specified status pattern.
We use re.search() to be consistent with pytest.raises.
"""
match = re.search(status_pattern, str(error))
return match is not None