class TestMd5sums(unittest.TestCase):
"""Test computing and verifying MD5 sums via files
"""
def setUp(self):
# Create and populate test directory
self.dir = ExampleDirScooby()
self.dir.create_directory()
# Make temporary area for input/ouput checksum files
self.md5sum_dir = tempfile.mkdtemp()
self.checksum_file = os.path.join(self.md5sum_dir,"checksums")
# Reference checksums for test directory
self.reference_checksums = []
for f in self.dir.filelist(full_path=False):
self.reference_checksums.append("%s %s" % (self.dir.checksum_for_file(f),f))
self.reference_checksums.append('')
self.reference_checksums = '\n'.join(self.reference_checksums)
# Store current dir and move to top level of test directory
self.pwd = os.getcwd()
os.chdir(self.dir.dirn)
def tearDown(self):
# Move back to original directory
os.chdir(self.pwd)
# Clean up test directory and checksum file
self.dir.delete_directory()
shutil.rmtree(self.md5sum_dir)
def test_compute_md5sums(self):
"""compute_md5sums make md5sum file for test directory
"""
compute_md5sums('.',output_file=self.checksum_file,relative=True)
checksums = open(self.checksum_file,'r').read()
reference_checksums = self.reference_checksums.split('\n')
reference_checksums.sort()
checksums = checksums.split('\n')
checksums.sort()
print str(checksums)
for l1,l2 in zip(reference_checksums,checksums):
self.assertEqual(l1,l2)
def test_verify_md5sums(self):
# Verify md5sums for test directory
fp = open(self.checksum_file,'w')
fp.write(self.reference_checksums)
fp.close()
self.assertEqual(verify_md5sums(self.checksum_file),0)
def test_compute_md5sum_for_file(self):
# Compute md5sum for a single file
compute_md5sum_for_file('test.txt',output_file=self.checksum_file)
checksum = open(self.checksum_file,'r').read()
self.assertEqual("0b26e313ed4a7ca6904b0e9369e5b957 test.txt\n",checksum)
def test_broken_links(self):
"""compute_md5sums make md5sum file for test directory with broken links
"""
# Add broken link to test dir
self.dir.add_link("broken","missing.txt")
compute_md5sums('.',output_file=self.checksum_file,relative=True)
