forked from TrueCloudLab/frostfs-testlib
46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import random
|
|
import re
|
|
import string
|
|
from datetime import datetime
|
|
|
|
ONLY_ASCII_LETTERS = string.ascii_letters
|
|
DIGITS_AND_ASCII_LETTERS = string.ascii_letters + string.digits
|
|
NON_DIGITS_AND_LETTERS = string.punctuation
|
|
|
|
|
|
def unique_name(prefix: str = ""):
|
|
"""
|
|
Generate unique short name of anything with prefix.
|
|
This should be unique in scope of multiple runs
|
|
|
|
Args:
|
|
prefix: prefix for unique name generation
|
|
Returns:
|
|
unique name string
|
|
"""
|
|
return f"{prefix}{hex(int(datetime.now().timestamp() * 1000000))}"
|
|
|
|
|
|
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
|