1 files changed, 127 insertions, 0 deletions
diff --git a/rdiff-backup/testing/rdifftest.py b/rdiff-backup/testing/rdifftest.py
new file mode 100644
index 0000000..471eab7
--- /dev/null
+++ b/rdiff-backup/testing/rdifftest.py
@@ -0,0 +1,127 @@
+import unittest, random
+
+execfile("commontest.py")
+rbexec("destructive_stepping.py")
+
+
+Log.setverbosity(6)
+
+def MakeRandomFile(path):
+	"""Writes a random file of length between 10000 and 100000"""
+	fp = open(path, "w")
+	randseq = []
+	for i in xrange(random.randrange(5000, 30000)):
+		randseq.append(chr(random.randrange(256)))
+	fp.write("".join(randseq))
+	fp.close()
+
+
+class RdiffTest(unittest.TestCase):
+	"""Test rdiff"""
+	lc = Globals.local_connection
+	basis = RPath(lc, "testfiles/basis")
+	new = RPath(lc, "testfiles/new")
+	output = RPath(lc, "testfiles/output")
+	delta = RPath(lc, "testfiles/delta")
+	signature = RPath(lc, "testfiles/signature")
+
+	def testRdiffSig(self):
+		"""Test making rdiff signatures"""
+		sig = RPath(self.lc, "testfiles/various_file_types/regular_file.sig")
+		sigfp = sig.open("r")
+		rfsig = Rdiff.get_signature(RPath(self.lc, "testfiles/various_file_types/regular_file"))
+		assert RPath.cmpfileobj(sigfp, rfsig)
+		sigfp.close()
+		rfsig.close()
+
+	def testRdiffDeltaPatch(self):
+		"""Test making deltas and patching files"""
+		rplist = [self.basis, self.new, self.delta,
+					self.signature, self.output]
+		for rp in rplist:
+			if rp.lstat(): rp.delete()
+			
+		for i in range(2):
+			MakeRandomFile(self.basis.path)
+			MakeRandomFile(self.new.path)
+			map(RPath.setdata, [self.basis, self.new])
+			assert self.basis.lstat() and self.new.lstat()
+			self.signature.write_from_fileobj(Rdiff.get_signature(self.basis))
+			assert self.signature.lstat()
+			self.delta.write_from_fileobj(Rdiff.get_delta(self.signature,
+														  self.new))
+			assert self.delta.lstat()
+			Rdiff.patch_action(self.basis, self.delta, self.output).execute()
+			assert RPath.cmp(self.new, self.output)
+			map(RPath.delete, rplist)
+
+	def testWriteDelta(self):
+		"""Test write delta feature of rdiff"""
+		rplist = [self.basis, self.new, self.delta, self.output]
+		MakeRandomFile(self.basis.path)
+		MakeRandomFile(self.new.path)
+		map(RPath.setdata, [self.basis, self.new])
+		assert self.basis.lstat() and self.new.lstat()
+
+		Rdiff.write_delta(self.basis, self.new, self.delta)
+		assert self.delta.lstat()
+		Rdiff.patch_action(self.basis, self.delta, self.output).execute()
+		assert RPath.cmp(self.new, self.output)
+		map(RPath.delete, rplist)		
+
+	def testRdiffRename(self):
+		"""Rdiff replacing original file with patch outfile"""
+		rplist = [self.basis, self.new, self.delta, self.signature]
+		for rp in rplist:
+			if rp.lstat(): rp.delete()
+
+		MakeRandomFile(self.basis.path)
+		MakeRandomFile(self.new.path)
+		map(RPath.setdata, [self.basis, self.new])
+		assert self.basis.lstat() and self.new.lstat()
+		self.signature.write_from_fileobj(Rdiff.get_signature(self.basis))
+		assert self.signature.lstat()
+		self.delta.write_from_fileobj(Rdiff.get_delta(self.signature,
+													  self.new))
+		assert self.delta.lstat()
+		Rdiff.patch_action(self.basis, self.delta).execute()
+		assert RPath.cmp(self.basis, self.new)
+		map(RPath.delete, rplist)
+
+	def testCopy(self):
+		"""Using rdiff to copy two files"""
+		rplist = [self.basis, self.new]
+		for rp in rplist:
+			if rp.lstat(): rp.delete()
+
+		MakeRandomFile(self.basis.path)
+		MakeRandomFile(self.new.path)
+		map(RPath.setdata, rplist)
+		Rdiff.copy_action(self.basis, self.new).execute()
+		assert RPath.cmp(self.basis, self.new)
+		map(RPath.delete, rplist)
+
+	def testPatchWithAttribs(self):
+		"""Using rdiff to copy two files with attributes"""
+		rplist = [self.basis, self.new, self.delta]
+		for rp in rplist:
+			if rp.lstat(): rp.delete()
+
+		MakeRandomFile(self.basis.path)
+		MakeRandomFile(self.new.path)
+		self.new.chmod(0401)
+		map(RPath.setdata, rplist)
+		Rdiff.write_delta(self.basis, self.new, self.delta)
+		RPath.copy_attribs(self.new, self.delta)
+		assert self.delta.getperms() == 0401
+
+		assert not self.basis == self.new
+		Rdiff.patch_with_attribs_action(self.basis, self.delta).execute()
+		if not self.basis == self.new:
+			print self.basis, self.new
+			assert 0
+		map(RPath.delete, rplist)
+
+
+if __name__ == '__main__':
+	unittest.main()