1234567891011121314151617181920212223 |
- import hashlib
- import io
- from os import path
- import pytest
- import slackviewer
- from slackviewer import archive
- from slackviewer.utils.six import to_bytes
- def test_SHA1_file():
- filepath = path.join("tests", "testarchive.zip")
- version = to_bytes(slackviewer.__version__)
- def SHA1_file(filepath, extra=b''):
- """The original unoptimized method (reads whole file instead of chunks)"""
- with io.open(filepath, 'rb') as f:
- return hashlib.sha1(f.read() + extra).hexdigest()
- expected = SHA1_file(filepath, version)
- actual = archive.SHA1_file(filepath, version)
- assert actual == expected
|