Initial version

git-svn-id: http://svn.savannah.nongnu.org/svn/rdiff-backup/trunk@8 2b77aa54-bcbc-44c9-a7ec-4f6cf2b41109

1 files changed, 145 insertions, 0 deletions

diff --git a/rdiff-backup/testing/hardlinktest.py b/rdiff-backup/testing/hardlinktest.py
new file mode 100644
index 0000000..38cdfbe
--- /dev/null
+++ b/rdiff-backup/testing/hardlinktest.py
@@ -0,0 +1,145 @@
+import os, unittest
+execfile("commontest.py")
+rbexec("main.py")
+
+
+Log.setverbosity(6)
+
+class HardlinkTest(unittest.TestCase):
+	"""Test cases for Hard links"""
+	outputrp = RPath(Globals.local_connection, "testfiles/output")
+	hardlink_dir1 = RPath(Globals.local_connection, "testfiles/hardlinks/dir1")
+	hardlink_dir1copy = \
+		RPath(Globals.local_connection, "testfiles/hardlinks/dir1copy")
+	hardlink_dir2 = RPath(Globals.local_connection, "testfiles/hardlinks/dir2")
+	hardlink_dir3 = RPath(Globals.local_connection, "testfiles/hardlinks/dir3")
+
+	def reset_output(self):
+		"""Erase and recreate testfiles/output directory"""
+		os.system(MiscDir+'/myrm testfiles/output')
+		self.outputrp.mkdir()
+
+	def testEquality(self):
+		"""Test rorp_eq function in conjunction with CompareRecursive"""
+		assert CompareRecursive(self.hardlink_dir1, self.hardlink_dir1copy)
+		assert CompareRecursive(self.hardlink_dir1, self.hardlink_dir2,
+								compare_hardlinks = None)
+		assert not CompareRecursive(self.hardlink_dir1, self.hardlink_dir2,
+								compare_hardlinks = 1)
+
+	def testCheckpointing(self):
+		"""Test saving and recovering of various dictionaries"""
+		d1 = {1:1}
+		d2 = {2:2}
+		d3 = {3:3}
+		d4 = {}
+
+		Hardlink._src_inode_indicies = d1
+		Hardlink._src_index_indicies = d2
+		Hardlink._dest_inode_indicies = d3
+		Hardlink._dest_index_indicies = d4
+
+		self.reset_output()
+		Time.setcurtime(12345)
+		Globals.isbackup_writer = 1
+		Hardlink.final_checkpoint(self.outputrp)
+
+		reset_hardlink_dicts()
+		assert Hardlink.retrieve_checkpoint(self.outputrp, 12345)
+		assert Hardlink._src_inode_indicies == d1, \
+			   Hardlink._src_inode_indicies
+		assert Hardlink._src_index_indicies == d2, \
+			   Hardlink._src_index_indicies
+		assert Hardlink._dest_inode_indicies == d3, \
+			   Hardlink._dest_inode_indicies
+		assert Hardlink._dest_index_indicies == d4, \
+			   Hardlink._dest_index_indicies
+
+	def testFinalwrite(self):
+		"""Test writing of the final database"""
+		Globals.isbackup_writer = 1
+		Time.setcurtime(123456)
+		Globals.rbdir = self.outputrp
+		finald = Hardlink._src_index_indicies = {'hello':'world'}
+		
+		self.reset_output()
+		Hardlink.final_writedata()
+
+		Hardlink._src_index_indicies = None
+		assert Hardlink.retrieve_final(123456)
+		assert Hardlink._src_index_indicies == finald
+
+	def testBuildingDict(self):
+		"""See if the partial inode dictionary is correct"""
+		Globals.preserve_hardlinks = 1
+		reset_hardlink_dicts()
+		for dsrp in DestructiveStepping.Iterate_from(self.hardlink_dir3, 1):
+			Hardlink.add_rorp(dsrp, 1)
+		
+		assert len(Hardlink._src_inode_indicies.keys()) == 3, \
+			   Hardlink._src_inode_indicies
+		assert len(Hardlink._src_index_indicies.keys()) == 3, \
+			   Hardlink._src_index_indicies
+		vals1 = Hardlink._src_inode_indicies.values()
+		vals2 = Hardlink._src_index_indicies.values()
+		vals1.sort()
+		vals2.sort()
+		assert vals1 == vals2
+
+	def testBuildingDict2(self):
+		"""Same as testBuildingDict but test destination building"""
+		Globals.preserve_hardlinks = 1
+		reset_hardlink_dicts()
+		for dsrp in DestructiveStepping.Iterate_from(self.hardlink_dir3, None):
+			Hardlink.add_rorp(dsrp, None)
+		
+		assert len(Hardlink._dest_inode_indicies.keys()) == 3, \
+			   Hardlink._dest_inode_indicies
+		assert len(Hardlink._dest_index_indicies.keys()) == 3, \
+			   Hardlink._dest_index_indicies
+		vals1 = Hardlink._dest_inode_indicies.values()
+		vals2 = Hardlink._dest_index_indicies.values()
+		vals1.sort()
+		vals2.sort()
+		assert vals1 == vals2
+
+	def testCompletedDict(self):
+		"""See if the hardlink dictionaries are built correctly"""
+		reset_hardlink_dicts()
+		for dsrp in DestructiveStepping.Iterate_from(self.hardlink_dir1, 1):
+			Hardlink.add_rorp(dsrp, 1)
+		assert Hardlink._src_inode_indicies == {}, \
+			   Hardlink._src_inode_indicies
+
+		hll1 = [('file1',), ('file2',), ('file3',)]
+		hll2 = [('file4',), ('file5',), ('file6',)]
+		dict = {}
+		for index in hll1: dict[index] = hll1
+		for index in hll2: dict[index] = hll2
+		assert Hardlink._src_index_indicies == dict
+
+		reset_hardlink_dicts()
+		for dsrp in DestructiveStepping.Iterate_from(self.hardlink_dir2, 1):
+			Hardlink.add_rorp(dsrp, 1)
+		assert Hardlink._src_inode_indicies == {}, \
+			   Hardlink._src_inode_indicies
+
+		hll1 = [('file1',), ('file3',), ('file4',)]
+		hll2 = [('file2',), ('file5',), ('file6',)]
+		dict = {}
+		for index in hll1: dict[index] = hll1
+		for index in hll2: dict[index] = hll2
+		assert Hardlink._src_index_indicies == dict
+
+	def testSeries(self):
+		"""Test hardlink system by backing up and restoring a few dirs"""
+		dirlist = ['testfiles/hardlinks/dir1',
+				   'testfiles/hardlinks/dir2',
+				   'testfiles/hardlinks/dir3',
+				   'testfiles/various_file_types']
+		BackupRestoreSeries(None, None, dirlist, compare_hardlinks=1)
+		BackupRestoreSeries(1, 1, dirlist, compare_hardlinks=1)
+
+
+
+if __name__ == "__main__": unittest.main()