diff options
author | bescoto <bescoto@2b77aa54-bcbc-44c9-a7ec-4f6cf2b41109> | 2005-10-27 05:59:02 +0000 |
---|---|---|
committer | bescoto <bescoto@2b77aa54-bcbc-44c9-a7ec-4f6cf2b41109> | 2005-10-27 05:59:02 +0000 |
commit | 80470345fa1998a033078314f77930a60ea14107 (patch) | |
tree | 319437e11b0b097362feb462a4891e5d6454db94 /rdiff-backup/testing | |
parent | 9276ba9789f0cd762a64c973683f53d8d15f1dc0 (diff) | |
download | rdiff-backup-80470345fa1998a033078314f77930a60ea14107.tar.gz |
Tests for new hash module
git-svn-id: http://svn.savannah.nongnu.org/svn/rdiff-backup/trunk@661 2b77aa54-bcbc-44c9-a7ec-4f6cf2b41109
Diffstat (limited to 'rdiff-backup/testing')
-rw-r--r-- | rdiff-backup/testing/hashtest.py | 103 |
1 files changed, 103 insertions, 0 deletions
diff --git a/rdiff-backup/testing/hashtest.py b/rdiff-backup/testing/hashtest.py new file mode 100644 index 0000000..653dda0 --- /dev/null +++ b/rdiff-backup/testing/hashtest.py @@ -0,0 +1,103 @@ +import unittest, StringIO +from rdiff_backup import hash, rpath, regress, restore, metadata +from commontest import * + +class HashTest(unittest.TestCase): + """Test the hash module""" + s1 = "Hello, world!" + s1_hash = "943a702d06f34599aee1f8da8ef9f7296031d699" + s2 = "The quick brown dog jumped over the lazy fox" + s2_hash = "eab21fb1a18b408909bae552b847f6b13f370f62" + s3 = "foobar" + s3_hash = "8843d7f92416211de9ebb963ff4ce28125932878" + + root_rp = rpath.RPath(Globals.local_connection, "testfiles") + + def test_basic(self): + """Compare sha1sum of a few strings""" + sfile = StringIO.StringIO(self.s1) + fw = hash.FileWrapper(sfile) + assert fw.read() == self.s1 + report = fw.close() + assert report.sha1_digest == self.s1_hash, report.sha1_digest + + sfile2 = StringIO.StringIO(self.s1) + fw2 = hash.FileWrapper(sfile2) + assert fw2.read(5) == self.s1[:5] + assert fw2.read() == self.s1[5:] + report2 = fw2.close() + assert report2.sha1_digest == self.s1_hash, report2.sha1_digest + + def make_dirs(self): + """Make two input directories""" + d1 = self.root_rp.append("hashtest1") + re_init_dir(d1) + d2 = self.root_rp.append("hashtest2") + re_init_dir(d2) + + d1f1 = d1.append("file1") + d1f1.write_string(self.s1) + d1f1l = d1.append("file1_linked") + d1f1l.hardlink(d1f1.path) + + d1f2 = d1.append("file2") + d1f2.write_string(self.s2) + d1f2l = d1.append("file2_linked") + d1f2l.hardlink(d1f2.path) + + d1_hashlist = [None, self.s1_hash, self.s1_hash, + self.s2_hash, self.s2_hash] + + d2f1 = d2.append("file1") + rpath.copy_with_attribs(d1f1, d2f1) + d2f1l = d2.append("file1_linked") + d2f1l.write_string(self.s3) + + d1f2 = d2.append("file2") + d1f2.mkdir() + + d2_hashlist = [None, self.s1_hash, self.s3_hash, None] + + return (d1, d1_hashlist, d2, d2_hashlist) + + def extract_hashs(self, metadata_rp): + """Return list of hashes in the metadata_rp""" + result = [] + comp = metadata_rp.isinccompressed() + extractor = metadata.RorpExtractor(metadata_rp.open("r", comp)) + for rorp in extractor.iterate(): + if rorp.has_sha1(): result.append(rorp.get_sha1()) + else: result.append(None) + return result + + def test_session(self): + """Run actual sessions and make sure proper hashes recorded + + There are a few code paths here we need to test: creating + ordinary files, updating ordinary files with diffs, hard + linking, and keeping files the same. + + """ + in_rp1, hashlist1, in_rp2, hashlist2 = self.make_dirs() + Myrm("testfiles/output") + + rdiff_backup(1, 1, in_rp1.path, "testfiles/output", 10000, "-v5") + meta_prefix = rpath.RPath(Globals.local_connection, + "testfiles/output/rdiff-backup-data/mirror_metadata") + incs = restore.get_inclist(meta_prefix) + assert len(incs) == 1 + metadata_rp = incs[0] + hashlist = self.extract_hashs(metadata_rp) + assert hashlist == hashlist1, (hashlist1, hashlist) + + rdiff_backup(1, 1, in_rp2.path, "testfiles/output", 20000, "-v5") + incs = restore.get_inclist(meta_prefix) + assert len(incs) == 2 + metadata_rp.delete() # easy way to find the other one + incs = restore.get_inclist(meta_prefix) + assert len(incs) == 1 + hashlist = self.extract_hashs(incs[0]) + assert hashlist == hashlist2, (hashlist2, hashlist) + + +if __name__ == "__main__": unittest.main() |