anikeev-yadro/frostfs-testcases
1
0
Fork 0
forked from TrueCloudLab/frostfs-testcases
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add get arbitrary range from file

Browse source 
Signed-off-by: anikeev-yadro <a.anikeev@yadro.com>
This commit is contained in:
anikeev-yadro 2022-10-24 21:40:30 +03:00 committed by anikeev-yadro
parent 8a48402f53
commit 0e86d55806
1 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
pytest_tests/helpers/file_helper.py
View file

@ -60,19 +60,23 @@ def generate_file_with_content(
@allure.step("Get File Hash") @allure.step("Get File Hash")
def get_file_hash(file_path: str, len: Optional[int] = None) -> str: def get_file_hash(file_path: str, len: Optional[int] = None, offset: Optional[int] = None) -> str:
"""Generates hash for the specified file. """Generates hash for the specified file.
Args: Args:
file_path: Path to the file to generate hash for. file_path: Path to the file to generate hash for.
len: How many bytes to read. len: How many bytes to read.
offset: Which position to start reading
Returns: Returns:
Hash of the file as hex-encoded string. Hash of the file as hex-encoded string.
""" """
file_hash = hashlib.sha256() file_hash = hashlib.sha256()
with open(file_path, "rb") as out: with open(file_path, "rb") as out:
if len: if len and offset is None:
file_hash.update(out.read(len))
elif len and offset:
out.seek(offset, 1)
file_hash.update(out.read(len)) file_hash.update(out.read(len))
else: else:
file_hash.update(out.read()) file_hash.update(out.read())