1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
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()
|
