def test_compute_dir(self, mock_hashes):
"""
Test hashing a directory
"""
mock_hashes.return_value = self.increment_hash_multi
chksum = CheckSum('onedir')
chksum.count = 2
result = chksum.compute()
self.assertEquals(self.dir_compute, result)
def test_compute_file(self, mock_get_checksum):
"""
Test compute the checksum of a file
"""
mock_get_checksum.return_value = self.hello_world_sha256sum
chksum = CheckSum('onefile')
chksum.count = 1
result = chksum.compute()
self.assertEquals(self.file_compute, result)