Implement a simple checksum test

This commit is contained in:
Kristóf Tóth 2018-03-26 00:01:28 +02:00
parent a951c9e7bc
commit 31825b9501
1 changed files with 47 additions and 0 deletions

47
test.py Normal file
View File

@ -0,0 +1,47 @@
import unittest
from random import choice
from os import chdir
from os.path import join
from tempfile import TemporaryDirectory
from copy import deepcopy
from hashlib import sha256
from coub_dl import coub_dl, File
class TestCoubDl(unittest.TestCase):
def setUp(self):
self.tmpdir = TemporaryDirectory()
self.directory = self.tmpdir.name
print(self.directory)
chdir(self.directory)
self.files = deepcopy(coub_dl.default_files)
for key in {key: self.files[key] for key in self.files if key not in coub_dl.output_files}:
self.files[key] = join(self.directory, self.files[key])
self.files[File.OUTPUT] = 'coub.mp4'
def tearDown(self):
self.tmpdir.cleanup()
def test_checksums(self):
urls = {'https://coub.com/view/dshfp': 'b5e1a4b3376a08d5e0bd833c0f5a0e82fbc09100b829b43c176396d4f26e08e4',
'https://coub.com/view/suj0d': '82895d364290f88a1ccca1ca03d1a96344501a5556d23c7193ce6107823ddbe9',
'https://coub.com/view/ucshn': 'f70b98cf28837d58e083d523933bcca20c2de344b7f13a322c771f0a284113d7'}
test_url = choice(list(urls.keys()))
coub_dl(test_url, self.files)()
self.assertEqual(TestCoubDl.hash_file(self.files[File.OUTPUT], sha256), urls[test_url])
@staticmethod
def hash_file(file, hasher, read_buffer=65536):
with open(file, 'rb') as ifile:
hasher = hasher()
while True:
chunk = ifile.read(read_buffer)
if not chunk: break
hasher.update(chunk)
return hasher.hexdigest()
if __name__ == '__main__':
unittest.main()